bug fixed for sql parser. for issue #1970

2023-03-04 19:17:16 +00:00 · 2017-09-14 11:07:47 +08:00
parent 98b3263363
commit ff9364cf12
3 changed files with 100 additions and 5 deletions
@@ -1745,6 +1745,13 @@ class SchemaResolveVisitorFactory {
                            x.setResolvedOwnerObject(parameter);
                            return;
                        }
+                    } else if (parentCtx.object instanceof SQLMergeStatement) {
+                        SQLMergeStatement mergeStatement = (SQLMergeStatement) parentCtx.object;
+                        SQLTableSource into = mergeStatement.getInto();
+                        if (into instanceof SQLSubqueryTableSource
+                                && into.aliasHashCode64() == owner_hash) {
+                            x.setResolvedOwnerObject(into);
+                        }
                    }

                    SQLDeclareItem declareItem = parentCtx.findDeclare(owner_hash);
@@ -2159,7 +2166,11 @@ class SchemaResolveVisitorFactory {
        SchemaResolveVisitor.Context ctx = visitor.createContext(x);

        SQLTableSource into = x.getInto();
-        ctx.setTableSource(into);
+        if (into instanceof SQLExprTableSource) {
+            ctx.setTableSource(into);
+        } else {
+            into.accept(visitor);
+        }

        SQLTableSource using = x.getUsing();
        if (using != null) {
@@ -2206,10 +2206,14 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {

        setMode(x, Mode.Merge);

-        String ident = x.getInto().toString();
-
-        TableStat stat = getTableStat(ident);
-        stat.incrementMergeCount();
+        SQLTableSource into = x.getInto();
+        if (into instanceof SQLExprTableSource) {
+            String ident = ((SQLExprTableSource) into).getExpr().toString();
+            TableStat stat = getTableStat(ident);
+            stat.incrementMergeCount();
+        } else {
+            into.accept(this);
+        }

        x.getOn().accept(this);

@@ -0,0 +1,80 @@
+/*
+ * Copyright 1999-2017 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.bvt.sql.oracle.visitor;
+
+import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.sql.dialect.oracle.visitor.OracleSchemaStatVisitor;
+import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
+import com.alibaba.druid.util.JdbcConstants;
+import junit.framework.TestCase;
+import org.junit.Assert;
+
+import java.util.List;
+
+public class OracleSchemaStatVisitorTest8_merge_into extends TestCase {
+    private final static String dbType = JdbcConstants.ORACLE;
+
+    public void test_0() throws Exception {
+        String sql = "MERGE INTO (\n" +
+                "SELECT *\n" +
+                "FROM KPI_M_CW_INCOME_FACT_BAK\n" +
+                "WHERE THE_DATE = '{THISMONTH}'\n" +
+                "AND AREA_LEVEL <= 1\n" +
+                "AND TYPE_ID = '2'\n" +
+                ") A\n" +
+                "USING (\n" +
+                "SELECT *\n" +
+                "FROM M_HEALTH_APPRAISE_LOAD\n" +
+                "WHERE THE_DATE = TRUNC(SYSDATE)\n" +
+                "AND AREA_LEVEL <= 1\n" +
+                ") B ON (A.AREA_ID = B.AREA_ID\n" +
+                "AND A.AREA_LEVEL = B.AREA_LEVEL)\n" +
+                "WHEN MATCHED THEN UPDATE SET A.SUM_CHRG_YS = ROUND(B.TOTAL_CHARGE * 1.00 / 10000, 2), A.CHARGE = B.THIS_CHARGE;";
+
+        List<SQLStatement> statementList = SQLUtils.toStatementList(sql, dbType);
+        SQLStatement stmt = statementList.get(0);
+
+        assertEquals(1, statementList.size());
+
+        SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(dbType);
+        stmt.accept(visitor);
+
+        System.out.println(stmt.toString());
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("relationShip : " + visitor.getRelationships());
+        System.out.println("where : " + visitor.getConditions());
+        System.out.println("groupBy : " + visitor.getGroupByColumns());
+
+        assertEquals(2, visitor.getTables().size());
+        assertTrue(visitor.containsTable("KPI_M_CW_INCOME_FACT_BAK"));
+        assertTrue(visitor.containsTable("M_HEALTH_APPRAISE_LOAD"));
+//
+        assertEquals(7, visitor.getColumns().size());
+        assertTrue(visitor.containsColumn("M_HEALTH_APPRAISE_LOAD", "*"));
+        assertTrue(visitor.containsColumn("M_HEALTH_APPRAISE_LOAD", "THE_DATE"));
+        assertTrue(visitor.containsColumn("M_HEALTH_APPRAISE_LOAD", "AREA_LEVEL"));
+        assertTrue(visitor.containsColumn("KPI_M_CW_INCOME_FACT_BAK", "*"));
+        assertTrue(visitor.containsColumn("KPI_M_CW_INCOME_FACT_BAK", "THE_DATE"));
+        assertTrue(visitor.containsColumn("KPI_M_CW_INCOME_FACT_BAK", "AREA_LEVEL"));
+        assertTrue(visitor.containsColumn("KPI_M_CW_INCOME_FACT_BAK", "TYPE_ID"));
+
+    }
+
+}