Added parsing logic

apache · Jan 21, 2025 · e5684a0 · e5684a0
1 parent 221bc1b
commit e5684a0
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/...tdb/confignode/manager/pipe/extractor/PipeConfigPhysicalPlanTablePatternParseVisitor.java b/...tdb/confignode/manager/pipe/extractor/PipeConfigPhysicalPlanTablePatternParseVisitor.java
@@ -31,6 +31,7 @@
 import org.apache.iotdb.confignode.consensus.request.write.table.CommitDeleteTablePlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.RenameTableColumnPlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.SetTableColumnCommentPlan;
+import org.apache.iotdb.confignode.consensus.request.write.table.SetTableCommentPlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.SetTablePropertiesPlan;
 
 import java.util.Optional;
@@ -128,6 +129,15 @@ public Optional<ConfigPhysicalPlan> visitCommitDeleteTable(
         : Optional.empty();
   }
 
+  @Override
+  public Optional<ConfigPhysicalPlan> visitSetTableComment(
+      final SetTableCommentPlan setTableCommentPlan, final TablePattern pattern) {
+    return matchDatabaseAndTableName(
+            setTableCommentPlan.getDatabase(), setTableCommentPlan.getTableName(), pattern)
+        ? Optional.of(setTableCommentPlan)
+        : Optional.empty();
+  }
+
   @Override
   public Optional<ConfigPhysicalPlan> visitSetTableColumnComment(
       final SetTableColumnCommentPlan setTableColumnCommentPlan, final TablePattern pattern) {

diff --git a/...confignode/manager/pipe/extractor/PipeConfigPhysicalPlanTablePatternParseVisitorTest.java b/...confignode/manager/pipe/extractor/PipeConfigPhysicalPlanTablePatternParseVisitorTest.java
@@ -31,6 +31,7 @@
 import org.apache.iotdb.confignode.consensus.request.write.table.CommitDeleteTablePlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.RenameTableColumnPlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.SetTableColumnCommentPlan;
+import org.apache.iotdb.confignode.consensus.request.write.table.SetTableCommentPlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.SetTablePropertiesPlan;
 import org.apache.iotdb.confignode.rpc.thrift.TDatabaseSchema;
 
@@ -113,6 +114,14 @@ public void testCommitDeleteTable() {
         new CommitDeleteTablePlan("da", "ac"));
   }
 
+  @Test
+  public void testSetTableComment() {
+    testInput(
+        new SetTableCommentPlan("db1", "ab", "a"),
+        new SetTableCommentPlan("db1", "ac", "a"),
+        new SetTableCommentPlan("da", "ac", "a"));
+  }
+
   @Test
   public void testSetTableColumnComment() {
     testInput(