Function SerializeUpdateOneParams

pg_documentdb/src/commands/update.c:2727–2792

Serializes the update one params and shardkey bson as a pgbson to send down to the worker. */

/* Serializes the update one params and shardkey bson as a pgbson to send down to the worker. */
static pgbson *
SerializeUpdateOneParams(UpdateOneParams *params, pgbson *shardKeyBson)
{
	pgbson_writer commandWriter;
	PgbsonWriterInit(&commandWriter);

	pgbson_writer writer;
	PgbsonWriterStartDocument(&commandWriter, "updateOne", -1, &writer);

	if (params->query != NULL)
	{
		PgbsonWriterAppendValue(&writer, "query", -1, params->query);
	}

	if (shardKeyBson != NULL)
	{
		PgbsonWriterAppendDocument(&writer, "shardKeyBson", -1, shardKeyBson);
	}

	/* Back-compat (to handle new coordinator, old worker)
	 * write it as a document that is { "": value }
	 * TODO: See if this can be elided in the future
	 */
	pgbson_writer valueWriter;
	PgbsonWriterStartDocument(&writer, "update", -1, &valueWriter);
	PgbsonWriterAppendValue(&valueWriter, "", 0, params->update);
	PgbsonWriterEndDocument(&writer, &valueWriter);

	PgbsonWriterAppendBool(&writer, "isUpsert", -1, params->isUpsert != 0);

	PgbsonWriterAppendBool(&writer, "bypassDocumentValidation", -1,
						   params->bypassDocumentValidation);

	if (params->sort != NULL)
	{
		PgbsonWriterAppendValue(&writer, "sort", 4, params->sort);
	}

	PgbsonWriterAppendInt32(&writer, "returnDocument", -1, (int) params->returnDocument);

	if (params->returnFields != NULL)
	{
		PgbsonWriterAppendValue(&writer, "returnFields", -1, params->returnFields);
	}

	if (params->arrayFilters != NULL)
	{
		/*
		 * back compat: To handle new coordinator old worker scenario, write it in the
		 * prior format of { "": value }
		 * TODO: See if this can be elided in the future
		 */
		PgbsonWriterStartDocument(&writer, "arrayFilters", -1, &valueWriter);
		PgbsonWriterAppendValue(&valueWriter, "", 0, params->arrayFilters);
		PgbsonWriterEndDocument(&writer, &valueWriter);
	}

	if (params->variableSpec != NULL &&