bug fixed OracleSQL Parser, support with subQuery

2023-03-04 19:17:16 +00:00 · 2012-08-30 05:24:06 +08:00
parent f60f4dbdad
commit f1449462c7
2 changed files with 96 additions and 1 deletions
@@ -686,7 +686,7 @@ public class OracleSelectParser extends SQLSelectParser {
        if (lexer.token() == (Token.LPAREN)) {
            lexer.nextToken();
            OracleSelectSubqueryTableSource tableSource;
-            if (lexer.token() == Token.SELECT) {
+            if (lexer.token() == Token.SELECT || lexer.token() == Token.WITH) {
                tableSource = new OracleSelectSubqueryTableSource(select());
            } else if (lexer.token() == (Token.LPAREN)) {
                tableSource = new OracleSelectSubqueryTableSource(select());
@@ -0,0 +1,95 @@
+/*
+ * Copyright 1999-2011 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;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.OracleTest;
+import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
+import com.alibaba.druid.sql.ast.statement.SQLSelect;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
+import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.sql.dialect.oracle.visitor.OracleSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+public class OracleSelectTest38 extends OracleTest {
+
+    public void test_0() throws Exception {
+        String sql = //
+        "select * from " +
+        "(with vw_kreis_statics_t as" +
+        "  (select substr(xzqh,1,6) xzqh,swrslx,sum(swrs_count) acd_totle from" +
+        "    (select xzqh,sglx,case when (swrs7 <  3) then '1'" +
+        "       when (swrs7 <  5) then '2' when (swrs7 <=  9) then '3' else '4' end  swrslx,1 swrs_count" +
+        "       from acduser.vw_acd_info where sglx='1' " +
+        "       " +
+        "                    and sgfssj  >=   ?" +
+        "                 " +
+        "                 " +
+        "        )" +
+        "   group by substr(xzqh,1,6),swrslx)" +
+        "" +
+        "   select e.\"XZQH\",e.\"LESS3\",e.\"F3TO5\",e.\"F5TO9\",e.\"MORE9\",kreis_code, kreis_name,px1,py1,px2,py2 from" +
+        		"    ( select" +
+        		"     xzqh," +
+        		"     nvl(max(decode(swrslx,'1',acd_totle)),0)  less3," +
+        		"     nvl(max(decode(swrslx,'2',acd_totle)),0)  f3to5," +
+        		"     nvl(max(decode(swrslx,'3',acd_totle)),0)  f5to9," +
+        		"     nvl(max(decode(swrslx,'4',acd_totle)),0)  more9" +
+        		"     from( select * from acduser.vw_kreis_statics_t) group by xzqh  " +
+        		"     ) e" +
+        		"" +
+        		"  left join" +
+        		" acduser.vw_sc_kreis_code_lv2 f on e.xzqh = f.short_kreis_code) " +
+        		"   where kreis_code in" +
+        		"(select * from " +
+        		"  (select tbek_code from acduser.vw_kreis_code start with tbek_code = ? connect by prior tbek_pk=tbek_parent ) " +
+        		"where  tbek_code != ?)"; //
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("relationships : " + visitor.getRelationships());
+        System.out.println("orderBy : " + visitor.getOrderByColumns());
+
+        Assert.assertEquals(4, visitor.getTables().size());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("acduser.vw_acd_info")));
+
+        Assert.assertEquals(15, visitor.getColumns().size());
+
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("acduser.vw_acd_info", "xzqh")));
+        Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("acduser.vw_acd_info", "sglx")));
+
+        // Assert.assertTrue(visitor.getOrderByColumns().contains(new TableStat.Column("employees", "last_name")));
+    }
+}