Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

improved sql parser. support create materialized view

Browse Source
This commit is contained in:
wenshao
2017-06-30 11:25:11 +08:00
parent 2c52661fe9
commit 6cfba1a0b9
9 changed files with 667 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/bvt/sql/oracle/create/OracleCreateMaterializedViewTest0.java
+77
View File
@@ -0,0 +1,77 @@
/*
* 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.create;
import com.alibaba.druid.sql.OracleTest;
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.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import java.util.List;
public class OracleCreateMaterializedViewTest0 extends OracleTest {
public void test_types() throws Exception {
String sql = //
"CREATE MATERIALIZED VIEW sales_summary AS\n" +
" SELECT\n" +
" seller_no,\n" +
" invoice_date,\n" +
" sum(invoice_amt) as sales_amt\n" +
" FROM invoice\n" +
" WHERE invoice_date < CURRENT_DATE\n" +
" GROUP BY\n" +
" seller_no,\n" +
" invoice_date\n" +
" ORDER BY\n" +
" seller_no,\n" +
" invoice_date;\n";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement stmt = statementList.get(0);
print(statementList);
assertEquals(1, statementList.size());
assertEquals("CREATE MATERIALIZED VIEW sales_summary\n" +
"AS\n" +
"SELECT seller_no, invoice_date, SUM(invoice_amt) AS sales_amt\n" +
"FROM invoice\n" +
"WHERE invoice_date < CURRENT_DATE\n" +
"GROUP BY seller_no, invoice_date\n" +
"ORDER BY seller_no, invoice_date;",//
SQLUtils.toSQLString(stmt, JdbcConstants.ORACLE));
OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
stmt.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());
assertEquals(1, visitor.getTables().size());
assertEquals(5, visitor.getColumns().size());
assertTrue(visitor.getColumns().contains(new TableStat.Column("invoice", "seller_no")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/oracle/create/OracleCreateMaterializedViewTest1.java
+74
View File
@@ -0,0 +1,74 @@
/*
* 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.create;
import com.alibaba.druid.sql.OracleTest;
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.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import java.util.List;
public class OracleCreateMaterializedViewTest1 extends OracleTest {
public void test_types() throws Exception {
String sql = //
"CREATE MATERIALIZED VIEW sales_mv\n" +
" BUILD IMMEDIATE\n" +
" REFRESH FAST ON COMMIT\n" +
" AS SELECT t.calendar_year, p.prod_id, \n" +
" SUM(s.amount_sold) AS sum_sales\n" +
" FROM times t, products p, sales s\n" +
" WHERE t.time_id = s.time_id AND p.prod_id = s.prod_id\n" +
" GROUP BY t.calendar_year, p.prod_id;";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement stmt = statementList.get(0);
print(statementList);
assertEquals(1, statementList.size());
assertEquals("CREATE MATERIALIZED VIEW sales_mv\n" +
"BUILD IMMEDIATE\n" +
"REFRESH FAST ON COMMIT\n" +
"AS\n" +
"SELECT t.calendar_year, p.prod_id, SUM(s.amount_sold) AS sum_sales\n" +
"FROM times t, products p, sales s\n" +
"WHERE t.time_id = s.time_id\n" +
"\tAND p.prod_id = s.prod_id\n" +
"GROUP BY t.calendar_year, p.prod_id;",//
SQLUtils.toSQLString(stmt, JdbcConstants.ORACLE));
OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
stmt.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());
assertEquals(3, visitor.getTables().size());
assertEquals(7, visitor.getColumns().size());
assertTrue(visitor.getColumns().contains(new TableStat.Column("times", "calendar_year")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/oracle/create/OracleCreateMaterializedViewTest2.java
+80
View File
@@ -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.create;
import com.alibaba.druid.sql.OracleTest;
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.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import java.util.List;
public class OracleCreateMaterializedViewTest2 extends OracleTest {
public void test_types() throws Exception {
String sql = //
"CREATE MATERIALIZED VIEW sales_by_month_by_state\n" +
" TABLESPACE example\n" +
" PARALLEL 4\n" +
" BUILD IMMEDIATE\n" +
" REFRESH COMPLETE\n" +
" ENABLE QUERY REWRITE\n" +
" AS SELECT t.calendar_month_desc, c.cust_state_province,\n" +
" SUM(s.amount_sold) AS sum_sales\n" +
" FROM times t, sales s, customers c\n" +
" WHERE s.time_id = t.time_id AND s.cust_id = c.cust_id\n" +
" GROUP BY t.calendar_month_desc, c.cust_state_province;";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement stmt = statementList.get(0);
print(statementList);
assertEquals(1, statementList.size());
assertEquals("CREATE MATERIALIZED VIEW sales_by_month_by_state\n" +
"TABLESPACE example\n" +
"PARALLEL 4\n" +
"BUILD IMMEDIATE\n" +
"REFRESH COMPLETE\n" +
"ENABLE QUERY REWRITE\n" +
"AS\n" +
"SELECT t.calendar_month_desc, c.cust_state_province, SUM(s.amount_sold) AS sum_sales\n" +
"FROM times t, sales s, customers c\n" +
"WHERE s.time_id = t.time_id\n" +
"\tAND s.cust_id = c.cust_id\n" +
"GROUP BY t.calendar_month_desc, c.cust_state_province;",//
SQLUtils.toSQLString(stmt, JdbcConstants.ORACLE));
OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
stmt.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());
assertEquals(3, visitor.getTables().size());
assertEquals(8, visitor.getColumns().size());
assertTrue(visitor.getColumns().contains(new TableStat.Column("times", "calendar_month_desc")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/oracle/create/OracleCreateViewTest10_with_read_only.java
+1 -1
View File
@@ -65,6 +65,6 @@ public class OracleCreateViewTest10_with_read_only extends OracleTest {
assertEquals(3, visitor.getColumns().size());
assertTrue(visitor.getColumns().contains(new TableStat.Column("employees", "last_name")));
assertTrue(visitor.getColumns().contains(new TableStat.Column("customers", "cust_last_name")));
}
}