Fix gaussdb vacuum clause transfer issue

alibaba · Dec 13, 2024 · 2648a17 · 2648a17
1 parent 4087fa9
commit 2648a17
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 8 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/gaussdb/visitor/GaussDbOutputVisitor.java b/core/src/main/java/com/alibaba/druid/sql/dialect/gaussdb/visitor/GaussDbOutputVisitor.java
@@ -12,6 +12,7 @@
 import com.alibaba.druid.sql.dialect.gaussdb.ast.stmt.GaussDbCreateTableStatement;
 import com.alibaba.druid.sql.dialect.gaussdb.ast.stmt.GaussDbInsertStatement;
 import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGInsertStatement;
+import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGVacuumStatement;
 import com.alibaba.druid.sql.dialect.postgresql.visitor.PGOutputVisitor;
 
 import java.util.List;
@@ -366,4 +367,14 @@ public boolean visit(SQLIntervalExpr x) {
         }
         return false;
     }
+
+    @Override
+    protected void printVacuumRest(PGVacuumStatement x) {
+        if (x.isDeltaMerge()) {
+            print0(ucase ? "DELTAMERGE " : "deltamerge ");
+        }
+        if (x.isHdfsDirectory()) {
+            print0(ucase ? "HDFSDIRECTORY " : "hdfsdirectory ");
+        }
+    }
 }
diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/postgresql/ast/stmt/PGVacuumStatement.java b/core/src/main/java/com/alibaba/druid/sql/dialect/postgresql/ast/stmt/PGVacuumStatement.java
@@ -122,8 +122,6 @@ public void setTruncate(boolean truncate) {
     protected void accept0(SQLASTVisitor visitor) {
         if (visitor instanceof PGASTVisitor) {
             accept0((PGASTVisitor) visitor);
-        } else {
-            super.accept0(visitor);
         }
     }
 

diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/postgresql/visitor/PGOutputVisitor.java b/core/src/main/java/com/alibaba/druid/sql/dialect/postgresql/visitor/PGOutputVisitor.java
@@ -2857,16 +2857,14 @@ public boolean visit(PGVacuumStatement x) {
         if (x.isTruncate()) {
             print0(ucase ? "TRUNCATE " : "truncate ");
         }
-        if (x.isDeltaMerge()) {
-            print0(ucase ? "DELTAMERGE " : "deltamerge ");
-        }
-        if (x.isHdfsDirectory()) {
-            print0(ucase ? "HDFSDIRECTORY " : "hdfsdirectory ");
-        }
+        printVacuumRest(x);
         printAndAccept(x.getTableSources(), ", ");
         return false;
     }
 
+    protected void printVacuumRest(PGVacuumStatement x){
+    }
+
     @Override
     public boolean visit(PGAlterDatabaseStatement x) {
         print0(ucase ? "ALTER DATABASE " : "alter database ");