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

Allow LOCK TABLE to be run in a procedure / function

pull/5899/head
gledis69 2022-04-19 16:59:38 +03:00
parent 3debcd4f38
commit 1f926db4fa
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -472,9 +472,9 @@ ProcessUtilityInternal(PlannedStmt *pstmt,
{ {
LockStmt *stmt = (LockStmt *) parsetree; LockStmt *stmt = (LockStmt *) parsetree;
ereport(NOTICE, errmsg("Processing LOCK command.")); bool isTopLevel = context == PROCESS_UTILITY_TOPLEVEL;
ErrorIfUnsupportedLockStmt(stmt, isTopLevel);
ErrorIfUnsupportedLockStmt(stmt);
uint32 nowaitFlag = stmt->nowait ? DIST_LOCK_NOWAIT : 0; uint32 nowaitFlag = stmt->nowait ? DIST_LOCK_NOWAIT : 0;
AcquireDistributedLockOnRelations(stmt->relations, stmt->mode, AcquireDistributedLockOnRelations(stmt->relations, stmt->mode,
DIST_LOCK_VIEWS_RECUR | nowaitFlag); DIST_LOCK_VIEWS_RECUR | nowaitFlag);

4
src/backend/distributed/utils/resource_lock.c
View File

@ -1253,9 +1253,9 @@ AcquireDistributedLockOnRelations(List *relationList, LOCKMODE lockMode, uint32
* - Locking shard, but citus.enable_manual_changes_to_shards is false * - Locking shard, but citus.enable_manual_changes_to_shards is false
*/ */
void void
ErrorIfUnsupportedLockStmt(LockStmt *stmt) ErrorIfUnsupportedLockStmt(LockStmt *stmt, bool isTopLevel)
{ {
RequireTransactionBlock(true, "LOCK TABLE"); RequireTransactionBlock(isTopLevel, "LOCK TABLE");
RangeVar *rangeVar = NULL; RangeVar *rangeVar = NULL;
foreach_ptr(rangeVar, stmt->relations) foreach_ptr(rangeVar, stmt->relations)

2
src/include/distributed/resource_lock.h
View File

@ -163,5 +163,5 @@ extern LOCKMODE LockModeCStringToLockMode(const char *lockModeName);
extern const char * LockModeToLockModeCString(LOCKMODE lockMode); extern const char * LockModeToLockModeCString(LOCKMODE lockMode);
extern void AcquireDistributedLockOnRelations(List *relationList, LOCKMODE lockMode, extern void AcquireDistributedLockOnRelations(List *relationList, LOCKMODE lockMode,
uint32 configs); uint32 configs);
extern void ErrorIfUnsupportedLockStmt(LockStmt *stmt); extern void ErrorIfUnsupportedLockStmt(LockStmt *stmt, bool isTopLevel);
#endif /* RESOURCE_LOCK_H */ #endif /* RESOURCE_LOCK_H */