960930b645
there are views or functions depending on columns after dropped ones. The issue was reported by depesz, and original patch by Denish Patel. Improved documentation how to build binaries from source. COPYRIGHT updated.
125 lines
3.0 KiB
C
Executable File
125 lines
3.0 KiB
C
Executable File
/*-------------------------------------------------------------------------
|
|
* pgut-spi.c
|
|
*
|
|
* Portions Copyright (c) 2008-2011, NIPPON TELEGRAPH AND TELEPHONE CORPORATION
|
|
* Portions Copyright (c) 2011, Itagaki Takahiro
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include "postgres.h"
|
|
#include "pgut-spi.h"
|
|
|
|
#define EXEC_FAILED(ret, expected) \
|
|
(((expected) > 0 && (ret) != (expected)) || (ret) < 0)
|
|
|
|
static void
|
|
termStringInfo(StringInfo str)
|
|
{
|
|
if (str && str->data)
|
|
pfree(str->data);
|
|
}
|
|
|
|
/* appendStringInfoVA + automatic buffer extension */
|
|
static void
|
|
appendStringInfoVA_s(StringInfo str, const char *fmt, va_list args)
|
|
{
|
|
while (!appendStringInfoVA(str, fmt, args))
|
|
{
|
|
/* Double the buffer size and try again. */
|
|
enlargeStringInfo(str, str->maxlen);
|
|
}
|
|
}
|
|
|
|
/* simple execute */
|
|
void
|
|
execute(int expected, const char *sql)
|
|
{
|
|
int ret = SPI_execute(sql, false, 0);
|
|
if EXEC_FAILED(ret, expected)
|
|
elog(ERROR, "query failed: (sql=%s, code=%d, expected=%d)", sql, ret, expected);
|
|
}
|
|
|
|
/* execute prepared plan */
|
|
void
|
|
execute_plan(int expected, SPIPlanPtr plan, Datum *values, const char *nulls)
|
|
{
|
|
int ret = SPI_execute_plan(plan, values, nulls, false, 0);
|
|
if EXEC_FAILED(ret, expected)
|
|
elog(ERROR, "query failed: (code=%d, expected=%d)", ret, expected);
|
|
}
|
|
|
|
/* execute sql with format */
|
|
void
|
|
execute_with_format(int expected, const char *format, ...)
|
|
{
|
|
va_list ap;
|
|
StringInfoData sql;
|
|
int ret;
|
|
|
|
initStringInfo(&sql);
|
|
va_start(ap, format);
|
|
appendStringInfoVA_s(&sql, format, ap);
|
|
va_end(ap);
|
|
|
|
if (strlen(sql.data) == 0)
|
|
elog(WARNING, "execute_with_format(%s)", format);
|
|
ret = SPI_exec(sql.data, 0);
|
|
if EXEC_FAILED(ret, expected)
|
|
elog(ERROR, "query failed: (sql=%s, code=%d, expected=%d)", sql.data, ret, expected);
|
|
|
|
termStringInfo(&sql);
|
|
}
|
|
|
|
void
|
|
execute_with_args(int expected, const char *src, int nargs, Oid argtypes[], Datum values[], const bool nulls[])
|
|
{
|
|
int ret;
|
|
int i;
|
|
char c_nulls[FUNC_MAX_ARGS];
|
|
|
|
for (i = 0; i < nargs; i++)
|
|
c_nulls[i] = (nulls[i] ? 'n' : ' ');
|
|
|
|
ret = SPI_execute_with_args(src, nargs, argtypes, values, c_nulls, false, 0);
|
|
if EXEC_FAILED(ret, expected)
|
|
elog(ERROR, "query failed: (sql=%s, code=%d, expected=%d)", src, ret, expected);
|
|
}
|
|
|
|
void
|
|
execute_with_format_args(int expected, const char *format, int nargs, Oid argtypes[], Datum values[], const bool nulls[], ...)
|
|
{
|
|
va_list ap;
|
|
StringInfoData sql;
|
|
|
|
initStringInfo(&sql);
|
|
va_start(ap, nulls);
|
|
appendStringInfoVA_s(&sql, format, ap);
|
|
va_end(ap);
|
|
|
|
execute_with_args(expected, sql.data, nargs, argtypes, values, nulls);
|
|
|
|
termStringInfo(&sql);
|
|
}
|
|
|
|
|
|
#if PG_VERSION_NUM < 80400
|
|
|
|
int
|
|
SPI_execute_with_args(const char *src,
|
|
int nargs, Oid *argtypes,
|
|
Datum *values, const char *nulls,
|
|
bool read_only, long tcount)
|
|
{
|
|
SPIPlanPtr plan;
|
|
int ret;
|
|
|
|
plan = SPI_prepare(src, nargs, argtypes);
|
|
if (plan == NULL)
|
|
return SPI_result;
|
|
ret = SPI_execute_plan(plan, values, nulls, read_only, tcount);
|
|
SPI_freeplan(plan);
|
|
return ret;
|
|
}
|
|
|
|
#endif
|