Function ProcessUpdate

pg_documentdb/src/commands/update.c:1644–1789

* ProcessUpdate processes a single update operation defined in * updateSpec on the given collection. */

 * updateSpec on the given collection.
 */
static void
ProcessUpdate(MongoCollection *collection, UpdateSpec *updateSpec,
			  text *transactionId, UpdateResult *result, bool forceInlineWrites,
			  ExprEvalState *stateForSchemaValidation)
{
	const bson_value_t *query = updateSpec->updateOneParams.query;
	const bson_value_t *update = updateSpec->updateOneParams.update;
	const bson_value_t *arrayFilters = updateSpec->updateOneParams.arrayFilters;
	bool isUpsert = updateSpec->updateOneParams.isUpsert;
	bool isMulti = updateSpec->isMulti;

	/* sort cannot be performed when multi is set to true  */
	if (isMulti && updateSpec->updateOneParams.sort != NULL)
	{
		ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_FAILEDTOPARSE),
						errmsg("sort option can not be set when multi=true")));
	}

	/* determine whether query filters by a single shard key value */
	int64 shardKeyHash = 0;
	bool isShardKeyValueCollationAware = false;
	bool hasShardKeyValueFilter =
		ComputeShardKeyHashForQueryValue(collection->shardKey, collection->collectionId,
										 query,
										 &shardKeyHash, &isShardKeyValueCollationAware);

	result->rowsMatched = 0;
	result->rowsModified = 0;
	result->performedUpsert = false;
	result->upsertedObjectId = NULL;

	if (isMulti)
	{
		if (DetermineUpdateType(update) == UpdateType_ReplaceDocument)
		{
			ereport(ERROR, (errcode(ERRCODE_DOCUMENTDB_FAILEDTOPARSE),
							errmsg(
								"multi=true and replace-style updates cannot be used together.")));
		}


		/*
		 * Update as many document as match the query. This is not a retryable
		 * operation, so we ignore transactionId.
		 */
		bool hasOnlyObjectIdFilter = false;

		UpdateAllMatchingDocsResult updateAllResult = UpdateAllMatchingDocuments(
			collection, &updateSpec->updateOneParams,
			hasShardKeyValueFilter,
			shardKeyHash, stateForSchemaValidation,
			&hasOnlyObjectIdFilter);
		result->rowsMatched = updateAllResult.matchedDocs;
		result->rowsModified = updateAllResult.rowsUpdated;

		/*
		 * In case of an upsert,
		 */