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

Fixes indentation

pull/7404/head
gurkanindibay 2024-01-29 17:50:52 +03:00
parent 754d5899f2
commit 71a733cdec
1 changed files with 28 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -112,8 +112,8 @@ typedef struct NonMainDbDistributedStatementInfo
typedef struct ObjectInfo typedef struct ObjectInfo
{ {
char *name; char *name;
Oid id; Oid id;
} ObjectInfo; } ObjectInfo;
/* /*
@ -158,7 +158,7 @@ static void RunPostprocessMainDBCommand(Node *parsetree);
static bool IsStatementSupportedInNonMainDb(Node *parsetree); static bool IsStatementSupportedInNonMainDb(Node *parsetree);
static bool StatementRequiresMarkDistributedFromNonMainDb(Node *parsetree); static bool StatementRequiresMarkDistributedFromNonMainDb(Node *parsetree);
static ObjectInfo GetObjectInfo(Node *parsetree); static ObjectInfo GetObjectInfo(Node *parsetree);
static void MarkObjectDistributedInNonMainDb( Node *parsetree, ObjectInfo objectInfo); static void MarkObjectDistributedInNonMainDb(Node *parsetree, ObjectInfo objectInfo);
/* /*
* ProcessUtilityParseTree is a convenience method to create a PlannedStmt out of * ProcessUtilityParseTree is a convenience method to create a PlannedStmt out of
@ -1674,42 +1674,45 @@ RunPostprocessMainDBCommand(Node *parsetree)
MarkObjectDistributedInNonMainDb(parsetree, objectInfo); MarkObjectDistributedInNonMainDb(parsetree, objectInfo);
} }
/* /*
* GetObjectInfo returns the name and oid of the object in the given parsetree. * GetObjectInfo returns the name and oid of the object in the given parsetree.
*/ */
static ObjectInfo GetObjectInfo(Node *parsetree) static ObjectInfo
GetObjectInfo(Node *parsetree)
{ {
ObjectInfo info; ObjectInfo info;
if (IsA(parsetree, CreateRoleStmt)) if (IsA(parsetree, CreateRoleStmt))
{ {
CreateRoleStmt *stmt = castNode(CreateRoleStmt, parsetree); CreateRoleStmt *stmt = castNode(CreateRoleStmt, parsetree);
info.name = stmt->role; info.name = stmt->role;
info.id = get_role_oid(stmt->role, false); info.id = get_role_oid(stmt->role, false);
} }
// Add else if branches for other statement types /* Add else if branches for other statement types */
return info; return info;
} }
/* /*
* MarkObjectDistributedInNonMainDb marks the given object as distributed on the * MarkObjectDistributedInNonMainDb marks the given object as distributed on the
* non-main database. * non-main database.
*/ */
static void MarkObjectDistributedInNonMainDb( Node *parsetree, ObjectInfo objectInfo) static void
MarkObjectDistributedInNonMainDb(Node *parsetree, ObjectInfo objectInfo)
{ {
StringInfo mainDBQuery = makeStringInfo(); StringInfo mainDBQuery = makeStringInfo();
appendStringInfo(mainDBQuery, appendStringInfo(mainDBQuery,
MARK_OBJECT_DISTRIBUTED, MARK_OBJECT_DISTRIBUTED,
AuthIdRelationId, AuthIdRelationId,
quote_literal_cstr(objectInfo.name), quote_literal_cstr(objectInfo.name),
objectInfo.id, objectInfo.id,
quote_literal_cstr(CurrentUserName())); quote_literal_cstr(CurrentUserName()));
RunCitusMainDBQuery(mainDBQuery->data); RunCitusMainDBQuery(mainDBQuery->data);
} }
/* /*
* IsStatementSupportedIn2Pc returns true if the statement is supported from a * IsStatementSupportedIn2Pc returns true if the statement is supported from a
* non-main database. * non-main database.