Function ExtensionExecuteQueryViaSPI

pg_documentdb/src/utils/query_utils.c:55–82

* ExtensionExecuteQueryViaSPI executes given query via SPI and returns first * attribute of the first tuple returned. * * Throws an error if command fails, or connection establishment to SPI fails * for some reason. * * Note that returning NULL doesn't mean failure. It either means returned * attribute is NULL or query returned nothing at all. In that case, isNull * is set to true. */

 * is set to true.
 */
Datum
ExtensionExecuteQueryViaSPI(const char *query, bool readOnly, int expectedSPIOK,
							bool *isNull)
{
	if (SPI_connect() != SPI_OK_CONNECT)
	{
		ereport(ERROR, (errmsg("could not connect to SPI manager")));
	}

	ereport(DEBUG1, (errmsg("executing \"%s\" via SPI", query), errhidestmt(true),
					 errhidecontext(true)));

	int tupleCountLimit = 1;
	int spiErrorCode = SPI_execute(query, readOnly, tupleCountLimit);
	if (spiErrorCode != expectedSPIOK)
	{
		ereport(ERROR, (errmsg("could not run SPI query %d", spiErrorCode)));
	}

	Datum retDatum = SPIReturnDatum(isNull, 1);

	if (SPI_finish() != SPI_OK_FINISH)
	{
		ereport(ERROR, (errmsg("could not finish SPI connection")));
	}

	return retDatum;
}


/*