From db2c52cd65bac27e895f7b14fff9a88b9f33b9dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=83=AD=E7=A6=8F?= <guofug@yonyou.com>
Date: Thu, 11 Aug 2022 20:43:32 +0800
Subject: [PATCH] =?UTF-8?q?fix:SQLStatement=20clone=E6=96=B9=E6=B3=95=20or?=
 =?UTF-8?q?acle=E4=BD=BF=E7=94=A8=E8=A1=8C=E5=88=97=E8=BD=AC=E7=BD=AEpiovt?=
 =?UTF-8?q?=20=E4=B8=A2=E5=A4=B1=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ast/statement/SQLSubqueryTableSource.java |  7 +++
 .../sql/parser/OraclePivotCloneTest.java      | 54 +++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 src/test/java/com/alibaba/druid/sql/parser/OraclePivotCloneTest.java

diff --git a/src/main/java/com/alibaba/druid/sql/ast/statement/SQLSubqueryTableSource.java b/src/main/java/com/alibaba/druid/sql/ast/statement/SQLSubqueryTableSource.java
index 67c19df139..f51bddceb8 100644
--- a/src/main/java/com/alibaba/druid/sql/ast/statement/SQLSubqueryTableSource.java
+++ b/src/main/java/com/alibaba/druid/sql/ast/statement/SQLSubqueryTableSource.java
@@ -16,6 +16,7 @@
 package com.alibaba.druid.sql.ast.statement;
 
 import com.alibaba.druid.sql.ast.SQLName;
+import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectSubqueryTableSource;
 import com.alibaba.druid.sql.visitor.SQLASTVisitor;
 
 import java.util.ArrayList;
@@ -73,6 +74,12 @@ public class SQLSubqueryTableSource extends SQLTableSourceImpl {
     public void cloneTo(SQLSubqueryTableSource x) {
         x.alias = alias;
 
+        if (this instanceof OracleSelectSubqueryTableSource) {
+            if (((OracleSelectSubqueryTableSource)this).getPivot() != null) {
+                ((OracleSelectSubqueryTableSource)x).setPivot(((OracleSelectSubqueryTableSource)this).getPivot());
+            }
+        }
+
         if (select != null) {
             x.select = select.clone();
             x.select.setParent(x);
diff --git a/src/test/java/com/alibaba/druid/sql/parser/OraclePivotCloneTest.java b/src/test/java/com/alibaba/druid/sql/parser/OraclePivotCloneTest.java
new file mode 100644
index 0000000000..148ae59592
--- /dev/null
+++ b/src/test/java/com/alibaba/druid/sql/parser/OraclePivotCloneTest.java
@@ -0,0 +1,54 @@
+package com.alibaba.druid.sql.parser;
+
+import com.alibaba.druid.DbType;
+import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import junit.framework.TestCase;
+
+/**
+ * @author gfChris
+ * @version 1.0
+ * @Description
+ * @date 2022/8/11 下午4:46
+ */
+public class OraclePivotCloneTest extends TestCase {
+
+    public void testCreateCharset() {
+
+        String sql = "Select \n" +
+                "DEPT_ID,\n" +
+                "M01,M02,M03,M04,M05,M06,M07,M08,M09,M10,M11,M12\n" +
+                "from \n" +
+                "(\n" +
+                "Select \n" +
+                "DEPT_ID,CMONTH,\n" +
+                "SUM(SO_TAXMONEY) AS SO_TAXMONEY\n" +
+                "FROM DW_SCM_DIM_SALEDATA\n" +
+                "GROUP BY \n" +
+                "DEPT_ID,CMONTH\n" +
+                ")  \n" +
+                "pivot(\n" +
+                "    sum(SO_TAXMONEY) for CMONTH in (    \n" +
+                "        '01' as M01,                \n" +
+                "        '02' as M02,                \n" +
+                "        '03' as M03,\n" +
+                "        '04' as M04,\n" +
+                "        '05' as M05,\n" +
+                "        '06' as M06,\n" +
+                "        '07' as M07,\n" +
+                "        '08' as M08,\n" +
+                "        '09' as M09,\n" +
+                "        '10' as M10,\n" +
+                "        '11' as M11,\n" +
+                "        '12' as M12\n" +
+                "    )\n" +
+                "  );";
+
+        SQLStatement sqlStatement = SQLUtils.parseSingleStatement(sql, DbType.oracle);
+
+        SQLStatement sqlStatement1 = sqlStatement.clone();
+
+        System.out.println(sqlStatement1.toString());
+    }
+
+}