citus/src/backend/distributed/deparser/citus_deparseutils.c

69 lines
1.9 KiB
C

#include "postgres.h"
#include "utils/builtins.h"
#include "commands/defrem.h"
#include "utils/elog.h"
#include "utils/rel.h"
#include "utils/relcache.h"
#include "utils/syscache.h"
#include "utils/typcache.h"
#include "distributed/deparser.h"
#include "distributed/pg_version_constants.h"
/**
* Convert a DefElem option to a SQL statement and append it to the given StringInfo buffer.
*
* @param buf The StringInfo buffer to append the SQL statement to.
* @param option The DefElem option to convert to a SQL statement.
* @param opt_formats The option format specification to use for the conversion.
* @param num_opt_formats The number of option formats in the opt_formats array.
*/
void
optionToStatement(StringInfo buf, DefElem *option, const struct
option_format *opt_formats, int
opt_formats_len)
{
const char *name = option->defname;
int i;
for (i = 0; i < opt_formats_len; i++)
{
if (strcmp(name, opt_formats[i].name) == 0)
{
if (opt_formats[i].type == OPTION_FORMAT_STRING)
{
char *value = defGetString(option);
appendStringInfo(buf, opt_formats[i].format, quote_identifier(value));
}
else if (opt_formats[i].type == OPTION_FORMAT_INTEGER)
{
int32 value = defGetInt32(option);
appendStringInfo(buf, opt_formats[i].format, value);
}
else if (opt_formats[i].type == OPTION_FORMAT_BOOLEAN)
{
bool value = defGetBoolean(option);
appendStringInfo(buf, opt_formats[i].format, value ? "true" : "false");
}
#if PG_VERSION_NUM >= PG_VERSION_15
else if (opt_formats[i].type == OPTION_FORMAT_OBJECT_ID)
{
Oid value = defGetObjectId(option);
appendStringInfo(buf, opt_formats[i].format, value);
}
#endif
else if (opt_formats[i].type == OPTION_FORMAT_LITERAL_CSTR)
{
char *value = defGetString(option);
appendStringInfo(buf, opt_formats[i].format, quote_literal_cstr(value));
}
else
{
elog(ERROR, "unrecognized option type: %d", opt_formats[i].type);
}
break;
}
}
}