mirror of https://github.com/citusdata/citus.git
346 lines
8.6 KiB
C
346 lines
8.6 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);
|
|
|
|
const struct option_format create_database_option_formats[] = {
|
|
{ "owner", " OWNER %s", OPTION_FORMAT_STRING },
|
|
{ "template", " TEMPLATE %s", OPTION_FORMAT_STRING },
|
|
{ "encoding", " ENCODING %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "strategy", " STRATEGY %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "locale", " LOCALE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "lc_collate", " LC_COLLATE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "lc_ctype", " LC_CTYPE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "icu_locale", " ICU_LOCALE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "icu_rules", " ICU_RULES %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "locale_provider", " LOCALE_PROVIDER %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "collation_version", " COLLATION_VERSION %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "tablespace", " TABLESPACE %s", OPTION_FORMAT_STRING },
|
|
{ "allow_connections", " ALLOW_CONNECTIONS %s", OPTION_FORMAT_BOOLEAN },
|
|
{ "connection_limit", " CONNECTION LIMIT %d", OPTION_FORMAT_INTEGER },
|
|
{ "is_template", " IS_TEMPLATE %s", OPTION_FORMAT_BOOLEAN },
|
|
{ "oid", " OID %d", OPTION_FORMAT_OBJECT_ID }
|
|
};
|
|
|
|
|
|
const struct option_format alter_database_option_formats[] = {
|
|
{ "is_template", " IS_TEMPLATE %s", OPTION_FORMAT_BOOLEAN },
|
|
{ "allow_connections", " ALLOW_CONNECTIONS %s", OPTION_FORMAT_BOOLEAN },
|
|
{ "connection_limit", " CONNECTION_LIMIT %d", OPTION_FORMAT_INTEGER },
|
|
};
|
|
|
|
const struct option_format create_database_option_formats[] = {
|
|
{ "owner", " OWNER %s", OPTION_FORMAT_STRING },
|
|
{ "template", " TEMPLATE %s", OPTION_FORMAT_STRING },
|
|
{ "encoding", " ENCODING %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "strategy", " STRATEGY %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "locale", " LOCALE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "lc_collate", " LC_COLLATE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "lc_ctype", " LC_CTYPE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "icu_locale", " ICU_LOCALE %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "icu_rules", " ICU_RULES %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "locale_provider", " LOCALE_PROVIDER %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "collation_version", " COLLATION_VERSION %s", OPTION_FORMAT_LITERAL_CSTR },
|
|
{ "tablespace", " TABLESPACE %s", OPTION_FORMAT_STRING },
|
|
{ "allow_connections", " ALLOW_CONNECTIONS %s", OPTION_FORMAT_BOOLEAN },
|
|
{ "connection_limit", " CONNECTION LIMIT %d", OPTION_FORMAT_INTEGER },
|
|
{ "is_template", " IS_TEMPLATE %s", OPTION_FORMAT_BOOLEAN },
|
|
{ "oid", " OID %d", OPTION_FORMAT_OBJECT_ID }
|
|
};
|
|
|
|
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 bool
|
|
AppendBasicAlterDatabaseOptions(StringInfo buf, DefElem *def, bool
|
|
prefix_appended_for_basic_options, char *dbname)
|
|
{
|
|
if (!prefix_appended_for_basic_options)
|
|
{
|
|
appendStringInfo(buf, "ALTER DATABASE %s WITH ", quote_identifier(dbname));
|
|
prefix_appended_for_basic_options = true;
|
|
}
|
|
optionToStatement(buf, def, alter_database_option_formats, lengthof(
|
|
alter_database_option_formats));
|
|
return prefix_appended_for_basic_options;
|
|
}
|
|
|
|
|
|
static void
|
|
AppendAlterDatabaseSetTablespace(StringInfo buf, DefElem *def, char *dbname)
|
|
{
|
|
appendStringInfo(buf,
|
|
"ALTER DATABASE %s SET TABLESPACE %s",
|
|
quote_identifier(dbname), quote_identifier(defGetString(def)));
|
|
}
|
|
|
|
|
|
static void
|
|
AppendAlterDatabaseStmt(StringInfo buf, AlterDatabaseStmt *stmt)
|
|
{
|
|
if (stmt->options)
|
|
{
|
|
ListCell *cell = NULL;
|
|
bool prefix_appended_for_basic_options = false;
|
|
foreach(cell, stmt->options)
|
|
{
|
|
DefElem *def = castNode(DefElem, lfirst(cell));
|
|
if (strcmp(def->defname, "tablespace") == 0)
|
|
{
|
|
AppendAlterDatabaseSetTablespace(buf, def, stmt->dbname);
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
prefix_appended_for_basic_options = AppendBasicAlterDatabaseOptions(buf,
|
|
def,
|
|
prefix_appended_for_basic_options,
|
|
stmt->
|
|
dbname);
|
|
}
|
|
}
|
|
}
|
|
|
|
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
|
|
|
|
char *
|
|
DeparseAlterDatabaseRenameStmt(Node *node)
|
|
{
|
|
RenameStmt *stmt = (RenameStmt *) node;
|
|
|
|
StringInfoData str;
|
|
initStringInfo(&str);
|
|
|
|
appendStringInfo(&str, "ALTER DATABASE %s RENAME TO %s",
|
|
quote_identifier(stmt->subname),
|
|
quote_identifier(stmt->newname));
|
|
|
|
return str.data;
|
|
}
|
|
|
|
|
|
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;
|
|
}
|
|
|
|
|
|
static void
|
|
AppendCreateDatabaseStmt(StringInfo buf, CreatedbStmt *stmt)
|
|
{
|
|
appendStringInfo(buf,
|
|
"CREATE DATABASE %s",
|
|
quote_identifier(stmt->dbname));
|
|
|
|
DefElem *option = NULL;
|
|
|
|
foreach_ptr(option, stmt->options)
|
|
{
|
|
optionToStatement(buf, option, create_database_option_formats, lengthof(
|
|
create_database_option_formats));
|
|
}
|
|
}
|
|
|
|
|
|
char *
|
|
DeparseCreateDatabaseStmt(Node *node)
|
|
{
|
|
CreatedbStmt *stmt = castNode(CreatedbStmt, node);
|
|
StringInfoData str = { 0 };
|
|
initStringInfo(&str);
|
|
|
|
AppendCreateDatabaseStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|
|
|
|
|
|
static void
|
|
AppendDropDatabaseStmt(StringInfo buf, DropdbStmt *stmt)
|
|
{
|
|
char *if_exists_statement = stmt->missing_ok ? "IF EXISTS" : "";
|
|
appendStringInfo(buf,
|
|
"DROP DATABASE %s %s",
|
|
if_exists_statement,
|
|
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)));
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
char *
|
|
DeparseDropDatabaseStmt(Node *node)
|
|
{
|
|
DropdbStmt *stmt = castNode(DropdbStmt, node);
|
|
StringInfoData str = { 0 };
|
|
initStringInfo(&str);
|
|
|
|
AppendDropDatabaseStmt(&str, stmt);
|
|
|
|
return str.data;
|
|
}
|