mirror of https://github.com/citusdata/citus.git
376 lines
8.5 KiB
C
376 lines
8.5 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* deparse_database_stmts.c
|
|
*
|
|
* All routines to deparse database statements.
|
|
*
|
|
* Copyright (c), Citus Data, Inc.
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include "postgres.h"
|
|
|
|
#include "pg_version_compat.h"
|
|
|
|
#include "catalog/namespace.h"
|
|
#include "lib/stringinfo.h"
|
|
#include "nodes/parsenodes.h"
|
|
#include "utils/builtins.h"
|
|
|
|
#include "distributed/deparser.h"
|
|
#include "distributed/citus_ruleutils.h"
|
|
#include "commands/defrem.h"
|
|
#include "distributed/deparser.h"
|
|
#include "distributed/log_utils.h"
|
|
#include "parser/parse_type.h"
|
|
#include "distributed/listutils.h"
|
|
|
|
static void AppendAlterDatabaseOwnerStmt(StringInfo buf, AlterOwnerStmt *stmt);
|
|
static void AppendAlterDatabaseStmt(StringInfo buf, AlterDatabaseStmt *stmt);
|
|
static void AppendDefElemConnLimit(StringInfo buf, DefElem *def);
|
|
|
|
char *
|
|
DeparseAlterDatabaseOwnerStmt(Node *node)
|
|
{
|
|
AlterOwnerStmt *stmt = castNode(AlterOwnerStmt, node);
|
|
StringInfoData str = {0};
|
|
initStringInfo(&str);
|
|
|
|
Assert(stmt->objectType == OBJECT_DATABASE);
|
|
|
|
AppendAlterDatabaseOwnerStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
static void
|
|
AppendAlterDatabaseOwnerStmt(StringInfo buf, AlterOwnerStmt *stmt)
|
|
{
|
|
Assert(stmt->objectType == OBJECT_DATABASE);
|
|
|
|
appendStringInfo(buf,
|
|
"ALTER DATABASE %s OWNER TO %s;",
|
|
quote_identifier(strVal((String *)stmt->object)),
|
|
RoleSpecString(stmt->newowner, true));
|
|
}
|
|
|
|
static void
|
|
AppendGrantDatabases(StringInfo buf, GrantStmt *stmt)
|
|
{
|
|
ListCell *cell = NULL;
|
|
appendStringInfo(buf, " ON DATABASE ");
|
|
|
|
foreach (cell, stmt->objects)
|
|
{
|
|
char *database = strVal(lfirst(cell));
|
|
appendStringInfoString(buf, quote_identifier(database));
|
|
if (cell != list_tail(stmt->objects))
|
|
{
|
|
appendStringInfo(buf, ", ");
|
|
}
|
|
}
|
|
}
|
|
|
|
static void
|
|
AppendGrantOnDatabaseStmt(StringInfo buf, GrantStmt *stmt)
|
|
{
|
|
Assert(stmt->objtype == OBJECT_DATABASE);
|
|
|
|
AppendGrantSharedPrefix(buf, stmt);
|
|
|
|
AppendGrantDatabases(buf, stmt);
|
|
|
|
AppendGrantSharedSuffix(buf, stmt);
|
|
}
|
|
|
|
static void
|
|
AppendDefElemConnLimit(StringInfo buf, DefElem *def)
|
|
{
|
|
appendStringInfo(buf, " CONNECTION LIMIT %ld", (long int)defGetNumeric(def));
|
|
}
|
|
|
|
static void
|
|
AppendAlterDatabaseStmt(StringInfo buf, AlterDatabaseStmt *stmt)
|
|
{
|
|
appendStringInfo(buf, "ALTER DATABASE %s ", quote_identifier(stmt->dbname));
|
|
|
|
if (stmt->options)
|
|
{
|
|
ListCell *cell = NULL;
|
|
appendStringInfo(buf, "WITH ");
|
|
foreach (cell, stmt->options)
|
|
{
|
|
DefElem *def = castNode(DefElem, lfirst(cell));
|
|
if (strcmp(def->defname, "is_template") == 0)
|
|
{
|
|
appendStringInfo(buf, "IS_TEMPLATE %s",
|
|
quote_literal_cstr(strVal(def->arg)));
|
|
}
|
|
else if (strcmp(def->defname, "connection_limit") == 0)
|
|
{
|
|
AppendDefElemConnLimit(buf, def);
|
|
}
|
|
else if (strcmp(def->defname, "allow_connections") == 0)
|
|
{
|
|
ereport(ERROR,
|
|
errmsg("ALLOW_CONNECTIONS is not supported"));
|
|
}
|
|
else
|
|
{
|
|
ereport(ERROR,
|
|
errmsg("unrecognized ALTER DATABASE option: %s",
|
|
def->defname));
|
|
}
|
|
}
|
|
}
|
|
|
|
appendStringInfo(buf, ";");
|
|
}
|
|
|
|
char *
|
|
DeparseGrantOnDatabaseStmt(Node *node)
|
|
{
|
|
GrantStmt *stmt = castNode(GrantStmt, node);
|
|
Assert(stmt->objtype == OBJECT_DATABASE);
|
|
|
|
StringInfoData str = {0};
|
|
initStringInfo(&str);
|
|
|
|
AppendGrantOnDatabaseStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
char *
|
|
DeparseAlterDatabaseStmt(Node *node)
|
|
{
|
|
AlterDatabaseStmt *stmt = castNode(AlterDatabaseStmt, node);
|
|
|
|
StringInfoData str = {0};
|
|
initStringInfo(&str);
|
|
|
|
AppendAlterDatabaseStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
#if PG_VERSION_NUM >= PG_VERSION_15
|
|
char *
|
|
DeparseAlterDatabaseRefreshCollStmt(Node *node)
|
|
{
|
|
AlterDatabaseRefreshCollStmt *stmt = (AlterDatabaseRefreshCollStmt *)node;
|
|
|
|
StringInfoData str;
|
|
initStringInfo(&str);
|
|
|
|
appendStringInfo(&str, "ALTER DATABASE %s REFRESH COLLATION VERSION;",
|
|
quote_identifier(
|
|
stmt->dbname));
|
|
|
|
return str.data;
|
|
}
|
|
|
|
#endif
|
|
|
|
static void
|
|
AppendAlterDatabaseSetStmt(StringInfo buf, AlterDatabaseSetStmt *stmt)
|
|
{
|
|
appendStringInfo(buf, "ALTER DATABASE %s", quote_identifier(stmt->dbname));
|
|
|
|
VariableSetStmt *varSetStmt = castNode(VariableSetStmt, stmt->setstmt);
|
|
|
|
AppendVariableSet(buf, varSetStmt);
|
|
}
|
|
|
|
char *
|
|
DeparseAlterDatabaseSetStmt(Node *node)
|
|
{
|
|
AlterDatabaseSetStmt *stmt = castNode(AlterDatabaseSetStmt, node);
|
|
|
|
StringInfoData str = {0};
|
|
initStringInfo(&str);
|
|
|
|
AppendAlterDatabaseSetStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
char *
|
|
DeparseCreateDatabaseSetStmt(Node *node)
|
|
{
|
|
CreatedbStmt *stmt = castNode(CreatedbStmt, node);
|
|
StringInfoData str = {0};
|
|
initStringInfo(&str);
|
|
|
|
AppendCreatedbStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
static void
|
|
AppendCreatedbStmt(StringInfo buf, CreatedbStmt *stmt)
|
|
{
|
|
appendStringInfo(buf,
|
|
"CREATE DATABASE %s",
|
|
quote_identifier(stmt->dbname));
|
|
|
|
DefElem *option = NULL;
|
|
|
|
foreach_ptr(option, stmt->options)
|
|
{
|
|
if (strcmp(option->defname, "tablespace") == 0)
|
|
{
|
|
char *tablespaceName = defGetString(option);
|
|
|
|
appendStringInfo(buf, " TABLESPACE %s",
|
|
quote_identifier(tablespaceName));
|
|
}
|
|
else if (strcmp(option->defname, "owner") == 0)
|
|
{
|
|
char *owner = defGetString(option);
|
|
|
|
appendStringInfo(buf, " OWNER %s",
|
|
quote_identifier(owner));
|
|
}
|
|
else if (strcmp(option->defname, "template") == 0)
|
|
{
|
|
char *template = defGetString(option);
|
|
|
|
appendStringInfo(buf, " TEMPLATE %s",
|
|
quote_identifier(template));
|
|
}
|
|
else if (strcmp(option->defname, "encoding") == 0)
|
|
{
|
|
char *encoding = defGetString(option);
|
|
|
|
appendStringInfo(buf, " ENCODING %s",
|
|
quote_literal_cstr(encoding));
|
|
}
|
|
else if (strcmp(option->defname, "locale") == 0)
|
|
{
|
|
char *locale = defGetString(option);
|
|
|
|
appendStringInfo(buf, " LOCALE %s",
|
|
quote_literal_cstr(locale));
|
|
}
|
|
else if (strcmp(option->defname, "lc_collate") == 0)
|
|
{
|
|
char *lc_collate = defGetString(option);
|
|
|
|
appendStringInfo(buf, " LC_COLLATE %s",
|
|
quote_literal_cstr(lc_collate));
|
|
}
|
|
else if (strcmp(option->defname, "lc_ctype") == 0)
|
|
{
|
|
char *lc_ctype = defGetString(option);
|
|
|
|
appendStringInfo(buf, " LC_CTYPE %s",
|
|
quote_literal_cstr(lc_ctype));
|
|
}
|
|
else if (strcmp(option->defname, "icu_locale") == 0)
|
|
{
|
|
char *icuLocale = defGetString(option);
|
|
|
|
appendStringInfo(buf, " ICU_LOCALE %s",
|
|
quote_literal_cstr(icuLocale));
|
|
}
|
|
else if (strcmp(option->defname, "locale_provider") == 0)
|
|
{
|
|
char *localeProvider = defGetString(option);
|
|
|
|
appendStringInfo(buf, " LOCALE_PROVIDER %s",
|
|
quote_literal_cstr(localeProvider));
|
|
}
|
|
else if (strcmp(option->defname, "is_template") == 0)
|
|
{
|
|
bool isTemplate = defGetBoolean(option);
|
|
|
|
appendStringInfo(buf, " IS_TEMPLATE %s",
|
|
isTemplate ? "true" : "false");
|
|
}
|
|
else if (strcmp(option->defname, "allow_connections") == 0)
|
|
{
|
|
bool allowConnections = defGetBoolean(option);
|
|
|
|
appendStringInfo(buf, " ALLOW_CONNECTIONS %s",
|
|
allowConnections ? "true" : "false");
|
|
}
|
|
else if (strcmp(option->defname, "connection_limit") == 0)
|
|
{
|
|
int connectionLimit = defGetInt32(option);
|
|
|
|
appendStringInfo(buf, " CONNECTION_LIMIT %d", connectionLimit);
|
|
}
|
|
#if PG_VERSION_NUM >= PG_VERSION_15
|
|
else if (strcmp(option->defname, "collation_version") == 0)
|
|
{
|
|
char *collationVersion = defGetString(option);
|
|
|
|
appendStringInfo(buf, " COLLATION_VERSION %s",
|
|
quote_literal_cstr(collationVersion));
|
|
}
|
|
else if (strcmp(option->defname, "oid") == 0)
|
|
{
|
|
Oid objectId = defGetObjectId(option);
|
|
|
|
appendStringInfo(buf, " OID %d", objectId);
|
|
}
|
|
else if (strcmp(option->defname, "strategy") == 0)
|
|
{
|
|
char *strategy = defGetString(option);
|
|
|
|
appendStringInfo(buf, " STRATEGY %s",
|
|
quote_literal_cstr(strategy));
|
|
}
|
|
#endif
|
|
else if (strcmp(option->defname, "location") == 0)
|
|
{
|
|
/* deprecated option */
|
|
}
|
|
else
|
|
{
|
|
ereport(ERROR, (errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("unrecognized CREATE DATABASE option \"%s\"",
|
|
option->defname)));
|
|
}
|
|
}
|
|
}
|
|
|
|
char *
|
|
DeparseDropDatabaseStmt(Node *node)
|
|
{
|
|
DropdbStmt *stmt = castNode(DropdbStmt, node);
|
|
StringInfoData str = { 0 };
|
|
initStringInfo(&str);
|
|
|
|
AppendDropDatabaseStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
|
|
static void
|
|
AppendDropDatabaseStmt(StringInfo buf, DropdbStmt *stmt)
|
|
{
|
|
appendStringInfo(buf,
|
|
"DROP DATABASE %s",
|
|
quote_identifier(stmt->dbname));
|
|
|
|
DefElem *option = NULL;
|
|
|
|
foreach_ptr(option, stmt->options)
|
|
{
|
|
if (strcmp(option->defname, "force") == 0)
|
|
{
|
|
appendStringInfo(buf, " FORCE");
|
|
}
|
|
else
|
|
{
|
|
ereport(ERROR, (errcode(ERRCODE_SYNTAX_ERROR),
|
|
errmsg("unrecognized DROP DATABASE option \"%s\"",
|
|
option->defname)));
|
|
}
|
|
}
|
|
}
|