enhance for schema stat

2023-03-04 19:17:16 +00:00 · 2012-02-24 14:19:00 +08:00
parent b46c327160
commit 5fc9acdb4c
9 changed files with 691 additions and 525 deletions
@@ -0,0 +1,40 @@
+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.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;
+
+public class OracleSelectTest extends OracleTest {
+
+    public void test_0() throws Exception {
+        String sql = "INSERT INTO films VALUES ('UA502', 'Bananas', 105, '1971-07-13', 'Comedy', '82 minutes');";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(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());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
+
+        Assert.assertEquals(0, visitor.getColumns().size());
+        
+//        Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "producer_id")));
+    }
+
+    
+}
@@ -0,0 +1,54 @@
+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.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;
+
+public class OracleSelectTest2 extends OracleTest {
+
+    public void test_0() throws Exception {
+        String sql = "WITH " + //
+                     "   dept_costs AS (" + //
+                     "      SELECT department_name, SUM(salary) dept_total" + //
+                     "         FROM employees e, departments d" + //
+                     "         WHERE e.department_id = d.department_id" + //
+                     "      GROUP BY department_name)," + //
+                     "   avg_cost AS (" + //
+                     "      SELECT SUM(dept_total)/COUNT(*) avg" + //
+                     "      FROM dept_costs)" + //
+                     "SELECT * FROM dept_costs" + //
+                     "   WHERE dept_total >" + //
+                     "      (SELECT avg FROM avg_cost)" + //
+                     "      ORDER BY department_name;";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(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());
+
+        Assert.assertEquals(2, visitor.getTables().size());
+        
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("employees")));
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("departments")));
+
+        Assert.assertEquals(0, visitor.getColumns().size());
+
+        // Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "producer_id")));
+    }
+
+}
@@ -0,0 +1,21 @@
+package com.alibaba.druid.sql;
+
+import java.util.List;
+
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.visitor.OracleOutputVisitor;
+
+public class OracleTest extends TestCase {
+	protected String output(List<SQLStatement> stmtList) {
+		StringBuilder out = new StringBuilder();
+		OracleOutputVisitor visitor = new OracleOutputVisitor(out);
+
+		for (SQLStatement stmt : stmtList) {
+			stmt.accept(visitor);
+		}
+
+		return out.toString();
+	}
+}