添加SQL统计功能，统计SQL中所使用的表和字段

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@271 b9813039-fb51-4c41-a8b9-e21c2acb5095

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest1.java

@@ -0,0 +1,70 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest1 extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "select a.name, b.name FROM users a, usergroups b on a.groupId = b.id";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(2, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+        Assert.assertEquals(true, visitor.getTables().containsKey("usergroups"));
+
+        Assert.assertEquals(4, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "groupId")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("usergroups", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("usergroups", "name")));
+
+    }
+    
+    public void test_1() throws Exception {
+        String sql = "select a.name, b.name FROM users a, usergroups b on a.groupId = b.id where a.groupID = ?";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(2, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+        Assert.assertEquals(true, visitor.getTables().containsKey("usergroups"));
+
+        Assert.assertEquals(4, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "groupId")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("usergroups", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("usergroups", "name")));
+
+    }
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest2.java

@@ -0,0 +1,89 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest2 extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "select id, name FROM users";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+
+    }
+    
+    public void test_2() throws Exception {
+        String sql = "select id, name FROM users a";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+
+    }
+    
+    public void test_3() throws Exception {
+        String sql = "select id, a.name FROM users a";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+
+    }
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest3.java

@@ -0,0 +1,45 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest3 extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "insert into users2 (id2, name2) select id, name FROM users where loginCount > 1";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(2, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+        Assert.assertEquals(true, visitor.getTables().containsKey("users2"));
+
+        Assert.assertEquals(5, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "loginCount")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users2", "name2")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users2", "id2")));
+
+    }
+    
+   
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest4.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest4 extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "insert into users (id, name) values(?, ?)";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+
+    }
+
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest_Delete.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleSchemaStatVisitor;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest_Delete extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "delete from users where id = ? or name = ?";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+
+    }
+
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest_Subquery.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleSchemaStatVisitor;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest_Subquery extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT id2, name2 FROM (select id, name from users limit 10) a";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+
+    }
+
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest_Subquery2.java

@@ -0,0 +1,44 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleSchemaStatVisitor;
+import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest_Subquery2 extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT a.id, a.name, b.name groupName FROM (select id, name, groupId from users WHERE ROWNUM < 10) a inner join groups b on a.groupId = b.id";
+
+        OracleStatementParser parser = new OracleStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(2, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+        Assert.assertEquals(true, visitor.getTables().containsKey("groups"));
+
+        Assert.assertEquals(5, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "groupId")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "name")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("groups", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("groups", "name")));
+
+    }
+
+}

src/test/java/com/alibaba/druid/sql/oracle/bvt/visitor/OracleSchemaStatVisitorTest_Update.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.oracle.bvt.visitor;
+
+import java.util.List;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat.Column;
+
+public class OracleSchemaStatVisitorTest_Update extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "update users set loginCount = loginCount + 1 where id = ?";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println(sql);
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertEquals(1, visitor.getTables().size());
+        Assert.assertEquals(true, visitor.getTables().containsKey("users"));
+
+        Assert.assertEquals(2, visitor.getFields().size());
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "id")));
+        Assert.assertEquals(true, visitor.getFields().contains(new Column("users", "loginCount")));
+
+    }
+
+}