(shardingsphere) branch master updated: Minor refactor for SubqueryExtractUtils (#33222)

[chengzhang]

This is an automated email from the ASF dual-hosted git repository.

chengzhang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 26c0cadfe6c Minor refactor for SubqueryExtractUtils (#33222)
26c0cadfe6c is described below

commit 26c0cadfe6ccf9f83be6496c1ee0c7185b903c13
Author: Zhengqiang Duan <duanzhengqi...@apache.org>
AuthorDate: Sat Oct 12 17:52:04 2024 +0800

    Minor refactor for SubqueryExtractUtils (#33222)
---
 .../shardingsphere/sql/parser/statement/core/enums/SubqueryType.java    | 2 +-
 .../sql/parser/statement/core/util/SubqueryExtractUtils.java            | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/enums/SubqueryType.java
 
b/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/enums/SubqueryType.java
index 87fe5163983..a0751b0a805 100644
--- 
a/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/enums/SubqueryType.java
+++ 
b/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/enums/SubqueryType.java
@@ -22,5 +22,5 @@ package 
org.apache.shardingsphere.sql.parser.statement.core.enums;
  */
 public enum SubqueryType {
     
-    PROJECTION, TABLE, JOIN, PREDICATE, INSERT_SELECT, EXISTS
+    PROJECTION, TABLE, JOIN, PREDICATE, INSERT_SELECT, EXISTS, WITH
 }
diff --git 
a/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/util/SubqueryExtractUtils.java
 
b/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/util/SubqueryExtractUtils.java
index 5dba65e27c5..ef6a46260f3 100644
--- 
a/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/util/SubqueryExtractUtils.java
+++ 
b/parser/sql/statement/core/src/main/java/org/apache/shardingsphere/sql/parser/statement/core/util/SubqueryExtractUtils.java
@@ -84,6 +84,7 @@ public final class SubqueryExtractUtils {
     
     private static void extractSubquerySegmentsFromCTEs(final 
List<SubquerySegment> result, final Collection<CommonTableExpressionSegment> 
withSegment) {
         for (CommonTableExpressionSegment each : withSegment) {
+            each.getSubquery().setSubqueryType(SubqueryType.WITH);
             result.add(each.getSubquery());
             extractSubquerySegments(result, each.getSubquery().getSelect());
         }