Optimize CTE in parser. (#31447)

apache · May 30, 2024 · d44e4f8 · d44e4f8
1 parent 448a433
commit d44e4f8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/...a/org/apache/shardingsphere/sql/parser/mysql/visitor/statement/MySQLStatementVisitor.java b/...a/org/apache/shardingsphere/sql/parser/mysql/visitor/statement/MySQLStatementVisitor.java
@@ -745,7 +745,7 @@ public ASTNode visitWithClause(final WithClauseContext ctx) {
         for (CteClauseContext each : ctx.cteClause()) {
             commonTableExpressions.add((CommonTableExpressionSegment) visit(each));
         }
-        return new WithSegment(ctx.start.getStartIndex(), ctx.stop.getStopIndex(), commonTableExpressions);
+        return new WithSegment(ctx.start.getStartIndex(), ctx.stop.getStopIndex(), commonTableExpressions, null != ctx.RECURSIVE());
     }
 
     @SuppressWarnings("unchecked")

diff --git a/...ain/java/org/apache/shardingsphere/sql/parser/sql/common/segment/generic/WithSegment.java b/...ain/java/org/apache/shardingsphere/sql/parser/sql/common/segment/generic/WithSegment.java
@@ -17,6 +17,7 @@
 
 package org.apache.shardingsphere.sql.parser.sql.common.segment.generic;
 
+import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
 import org.apache.shardingsphere.sql.parser.sql.common.segment.SQLSegment;
@@ -27,6 +28,7 @@
 /**
  * With segment.
  */
+@AllArgsConstructor
 @RequiredArgsConstructor
 @Getter
 public final class WithSegment implements SQLSegment {
@@ -36,4 +38,6 @@ public final class WithSegment implements SQLSegment {
     private final int stopIndex;
 
     private final Collection<CommonTableExpressionSegment> commonTableExpressions;
+
+    private boolean recursive;
 }