From 7f7fd482fcd1e40c07d5c1a63b02d4aacb0ace55 Mon Sep 17 00:00:00 2001 From: Burak Velioglu Date: Wed, 26 Jan 2022 02:23:45 +0300 Subject: [PATCH] Add ensure coordinator check to new UDF --- src/backend/distributed/worker/worker_drop_protocol.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/backend/distributed/worker/worker_drop_protocol.c b/src/backend/distributed/worker/worker_drop_protocol.c index fad84b308..f2ae3597f 100644 --- 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);