Add ensure coordinator check to new UDF

2022-01-26 02:23:45 +03:00 · 2022-01-26 02:23:45 +03:00 · 7f7fd482fc
parent cf5f554fc9
commit 7f7fd482fc
1 changed files with 6 additions and 0 deletions
--- a/src/backend/distributed/worker/worker_drop_protocol.c
+++ b/src/backend/distributed/worker/worker_drop_protocol.c
@ -168,6 +168,12 @@ worker_drop_shell_table(PG_FUNCTION_ARGS)

 	EnsureTableOwner(relationId);

+	if (GetLocalGroupId() == COORDINATOR_GROUP_ID)
+	{
+		ereport(ERROR, (errmsg("worker_drop_shell_table is only allowed to run"
+							   " on worker nodes")));
+	}
+
 	/* first check the relation type */
 	Relation distributedRelation = relation_open(relationId, AccessShareLock);
 	EnsureRelationKindSupported(relationId);