Function DeleteRetryRecord

pg_documentdb/src/commands/retryable_writes.c:174–305

* DeleteRetryRecordGetObjectId deletes a retry record and returns the object ID * that was stored in the record. */

 * that was stored in the record.
 */
bool
DeleteRetryRecord(uint64 collectionId, int64 shardKeyValue,
				  text *transactionId, RetryableWriteResult *writeResult)
{
	StringInfoData query;
	int spiStatus PG_USED_FOR_ASSERTS_ONLY = 0;
	bool foundRetryRecord = false;

	MemoryContext outerContext = CurrentMemoryContext;

	SPI_connect();

	if (UseLocalRetryTable())
	{
		const int argCount = 3;
		Oid argTypes[3];
		Datum argValues[3];

		initStringInfo(&query);
		appendStringInfo(&query,
						 "DELETE FROM %s.retryable_writes"
						 " WHERE collection_id = $1 AND shard_key_value = $2"
						 " AND transaction_id = $3"
						 " RETURNING object_id, rows_affected, result_document",
						 ApiDataSchemaName);

		argTypes[0] = INT8OID;
		argValues[0] = Int64GetDatum((int64) collectionId);

		argTypes[1] = INT8OID;
		argValues[1] = Int64GetDatum(shardKeyValue);

		argTypes[2] = TEXTOID;
		argValues[2] = PointerGetDatum(transactionId);

		char *argNulls = NULL;
		bool readOnly = false;
		long maxTupleCount = 0;

		SPIPlanPtr plan = GetSPIQueryPlan(collectionId,
										  QUERY_ID_GLOBAL_RETRY_RECORD_DELETE,
										  query.data, argTypes, argCount);

		spiStatus = SPI_execute_plan(plan, argValues, argNulls, readOnly,
									 maxTupleCount);
		Assert(spiStatus == SPI_OK_DELETE_RETURNING);
	}
	else
	{
		const int argCount = 2;
		Oid argTypes[2];
		Datum argValues[2];

		initStringInfo(&query);
		appendStringInfo(&query,
						 "DELETE FROM %s.retry_" UINT64_FORMAT
						 " WHERE shard_key_value = $1 AND transaction_id = $2"
						 " RETURNING object_id, rows_affected, result_document",