From 6513ff0aae7098ddd3beb25f92df00871f5426ff Mon Sep 17 00:00:00 2001
From: chen <chenbobo3@huawei.com>
Date: Mon, 12 Dec 2022 12:50:29 +0000
Subject: [PATCH]  forbid drop partition delta/cudesc table

Signed-off-by: chen <chenbobo3@huawei.com>
---
 .../optimizer/commands/tablecmds.cpp          | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/src/gausskernel/optimizer/commands/tablecmds.cpp b/src/gausskernel/optimizer/commands/tablecmds.cpp
index f428341c9b..ee0b2f6461 100644
--- a/src/gausskernel/optimizer/commands/tablecmds.cpp
+++ b/src/gausskernel/optimizer/commands/tablecmds.cpp
@@ -3803,6 +3803,47 @@ static void DropRelationPermissionCheck(char relkind, Oid relOid, Oid nspOid, co
         aclcheck_error(aclresult, ACL_KIND_CLASS, relname);
     }
 }
+
+static bool IsPartitionDeltaCudesc(Oid relOid)
+{
+#define PARTITION_DELTA_NAME "pg_delta_part_"
+#define PARTITION_CUDESC_NAME "pg_cudesc_part_"
+
+    int attnum;
+    bool found = false;
+    ScanKeyData scanKey[1];
+    TableScanDesc scan;
+    Relation pgpartition = NULL;
+    Relation rel = NULL;
+
+    rel = try_relation_open(relOid, AccessShareLock);
+    if (!RelationIsValid(rel)) {
+        return false;
+    }
+
+    const char *relname = RelationGetRelationName(rel);
+    if (strncmp(relname, PARTITION_DELTA_NAME, strlen(PARTITION_DELTA_NAME)) == 0) {
+        attnum = Anum_pg_partition_reltoastrelid;
+    } else if (strncmp(relname, PARTITION_CUDESC_NAME, strlen(PARTITION_CUDESC_NAME)) == 0) {
+        attnum = Anum_pg_partition_relcudescrelid;
+    } else {
+        heap_close(rel, AccessShareLock);
+        return false;
+    }
+
+    ScanKeyInit(&scanKey[0], attnum, BTEqualStrategyNumber, F_OIDEQ, ObjectIdGetDatum(rel->rd_id));
+    pgpartition = heap_open(PartitionRelationId, AccessShareLock);
+    scan = tableam_scan_begin(pgpartition, SnapshotNow, 1, scanKey);
+    if (tableam_scan_getnexttuple(scan, ForwardScanDirection)) {
+        found = true;
+    }
+    tableam_scan_end(scan);
+    heap_close(pgpartition, AccessShareLock);
+    heap_close(rel, AccessShareLock);
+
+    return found;
+}
+
 /*
  * Before acquiring a table lock, check whether we have sufficient rights.
  * In the case of DROP INDEX, also try to lock the table before the index.
@@ -3881,6 +3922,11 @@ static void RangeVarCallbackForDropRelation(
             invalid_system_index = true;
     }
 
+    if (IsPartitionDeltaCudesc(relOid)) {
+        ereport(ERROR,
+            (errcode(ERRCODE_WRONG_OBJECT_TYPE),
+                errmsg("cannot drop relation \"%s\", it is a partition delta/cudesc table", rel->relname)));
+    }
 
     /* Permission Check */
     DropRelationPermissionCheck(relkind, relOid, classform->relnamespace, rel->relname);
-- 
Gitee