Function InsertRetryRecord

pg_documentdb/src/commands/retryable_writes.c:312–456

* InsertRetryRecord inserts a retryable write record into the retry * table of a collection. */

 * table of a collection.
 */
void
InsertRetryRecord(uint64 collectionId, int64 shardKeyValue, text *transactionId,
				  pgbson *objectId, bool rowsAffected, pgbson *resultDocument)
{
	StringInfoData query;
	int spiStatus PG_USED_FOR_ASSERTS_ONLY = 0;

	SPI_connect();

	if (UseLocalRetryTable())
	{
		const int argCount = 6;
		Oid argTypes[6];
		Datum argValues[6];
		char argNulls[] = { ' ', ' ', ' ', ' ', ' ', ' ' };

		/*
		 * With a single shared retry table, all collections' retry records
		 * share the same index pages, which may cause buffer contention under
		 * heavy concurrent write workloads compared to per-collection retry
		 * tables. In practice this is expected to be low severity since retry
		 * records are short-lived.
		 */
		initStringInfo(&query);
		appendStringInfo(&query,
						 "INSERT INTO %s.retryable_writes"
						 " (collection_id, shard_key_value, transaction_id, object_id, "
						 "  rows_affected, result_document) "
						 " VALUES ($1, $2, $3, $4::%s, $5, $6::%s)",
						 ApiDataSchemaName, FullBsonTypeName, FullBsonTypeName);

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

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

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

		argTypes[3] = BYTEAOID;

		if (objectId != NULL)
		{
			argValues[3] = PointerGetDatum(objectId);
			argNulls[3] = ' ';
		}
		else
		{
			argNulls[3] = 'n';
		}

		argTypes[4] = BOOLOID;
		argValues[4] = BoolGetDatum(rowsAffected);

		argTypes[5] = BYTEAOID;

		if (resultDocument != NULL)