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
{
char *name;
Oid id;
char *name;
Oid id;
} ObjectInfo;
/*
@ -158,7 +158,7 @@ static void RunPostprocessMainDBCommand(Node *parsetree);
static bool IsStatementSupportedInNonMainDb(Node *parsetree);
static bool StatementRequiresMarkDistributedFromNonMainDb(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
@ -1674,42 +1674,45 @@ RunPostprocessMainDBCommand(Node *parsetree)
MarkObjectDistributedInNonMainDb(parsetree, objectInfo);
}
/*
* 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))
{
CreateRoleStmt *stmt = castNode(CreateRoleStmt, parsetree);
info.name = stmt->role;
info.id = get_role_oid(stmt->role, false);
}
// Add else if branches for other statement types
if (IsA(parsetree, CreateRoleStmt))
{
CreateRoleStmt *stmt = castNode(CreateRoleStmt, parsetree);
info.name = stmt->role;
info.id = get_role_oid(stmt->role, false);
}
/* Add else if branches for other statement types */
return info;
return info;
}
/*
* MarkObjectDistributedInNonMainDb marks the given object as distributed on the
* non-main database.
*/
static void MarkObjectDistributedInNonMainDb( Node *parsetree, ObjectInfo objectInfo)
*/
static void
MarkObjectDistributedInNonMainDb(Node *parsetree, ObjectInfo objectInfo)
{
StringInfo mainDBQuery = makeStringInfo();
appendStringInfo(mainDBQuery,
MARK_OBJECT_DISTRIBUTED,
AuthIdRelationId,
quote_literal_cstr(objectInfo.name),
objectInfo.id,
quote_literal_cstr(CurrentUserName()));
RunCitusMainDBQuery(mainDBQuery->data);
StringInfo mainDBQuery = makeStringInfo();
appendStringInfo(mainDBQuery,
MARK_OBJECT_DISTRIBUTED,
AuthIdRelationId,
quote_literal_cstr(objectInfo.name),
objectInfo.id,
quote_literal_cstr(CurrentUserName()));
RunCitusMainDBQuery(mainDBQuery->data);
}
/*
* IsStatementSupportedIn2Pc returns true if the statement is supported from a
* non-main database.