external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

a bit of re-ardering

pull/7439/head
Onur Tirtir 2024-02-21 11:21:37 +03:00
parent 0ab2ee8ed4
commit e9b67f158c
1 changed files with 25 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src/backend/distributed/commands/utility_hook.c
View File

@ -162,6 +162,7 @@ static bool ShouldCheckUndistributeCitusLocalTables(void);
* Functions to support commands used to manage node-wide objects from non-main * Functions to support commands used to manage node-wide objects from non-main
* databases. * databases.
*/ */
static bool IsCommandToCreateOrDropMainDB(Node *parsetree);
static void RunPreprocessMainDBCommand(Node *parsetree); static void RunPreprocessMainDBCommand(Node *parsetree);
static void RunPostprocessMainDBCommand(Node *parsetree); static void RunPostprocessMainDBCommand(Node *parsetree);
static bool IsStatementSupportedFromNonMainDb(Node *parsetree); static bool IsStatementSupportedFromNonMainDb(Node *parsetree);
@ -187,7 +188,6 @@ static const NonMainDbDistributedStatementInfo NonMainDbSupportedStatements[] =
{ T_GrantStmt, false, NonMainDbCheckSupportedObjectTypeForGrant } { T_GrantStmt, false, NonMainDbCheckSupportedObjectTypeForGrant }
}; };
static bool IsCommandToCreateOrDropMainDB(Node *parsetree);
/* /*
* ProcessUtilityParseTree is a convenience method to create a PlannedStmt out of * ProcessUtilityParseTree is a convenience method to create a PlannedStmt out of
@ -331,6 +331,7 @@ citus_ProcessUtility(PlannedStmt *pstmt,
!IsCommandToCreateOrDropMainDB(parsetree)) !IsCommandToCreateOrDropMainDB(parsetree))
{ {
RunPreprocessMainDBCommand(parsetree); RunPreprocessMainDBCommand(parsetree);
if (IsA(parsetree, CreatedbStmt) || if (IsA(parsetree, CreatedbStmt) ||
IsA(parsetree, DropdbStmt)) IsA(parsetree, DropdbStmt))
{ {
@ -1681,6 +1682,29 @@ DropSchemaOrDBInProgress(void)
} }
/*
* IsCommandToCreateOrDropMainDB checks if this query creates or drops the
* main database, so we can make an exception and not send this query to
* the main database.
*/
static bool
IsCommandToCreateOrDropMainDB(Node *parsetree)
{
if (IsA(parsetree, CreatedbStmt))
{
CreatedbStmt *createdbStmt = castNode(CreatedbStmt, parsetree);
return strcmp(createdbStmt->dbname, MainDb) == 0;
}
else if (IsA(parsetree, DropdbStmt))
{
DropdbStmt *dropdbStmt = castNode(DropdbStmt, parsetree);
return strcmp(dropdbStmt->dbname, MainDb) == 0;
}
return false;
}
/* /*
* RunPreprocessMainDBCommand runs the necessary commands for a query, in main * RunPreprocessMainDBCommand runs the necessary commands for a query, in main
* database before query is run on the local node with PrevProcessUtility * database before query is run on the local node with PrevProcessUtility
@ -1836,25 +1860,3 @@ NonMainDbCheckSupportedObjectTypeForGrant(Node *node)
GrantStmt *stmt = castNode(GrantStmt, node); GrantStmt *stmt = castNode(GrantStmt, node);
return stmt->objtype == OBJECT_DATABASE; return stmt->objtype == OBJECT_DATABASE;
} }
/*
* IsCommandToCreateOrDropMainDB checks if this query creates or drops the
* main database, so we can make an exception and not send this query to
* the main database.
*/
static bool
IsCommandToCreateOrDropMainDB(Node *parsetree)
{
if (IsA(parsetree, CreatedbStmt))
{
CreatedbStmt *createdbStmt = castNode(CreatedbStmt, parsetree);
return strcmp(createdbStmt->dbname, MainDb) == 0;
}
else if (IsA(parsetree, DropdbStmt))
{
DropdbStmt *dropdbStmt = castNode(DropdbStmt, parsetree);
return strcmp(dropdbStmt->dbname, MainDb) == 0;
}
return false;
}