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

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

src/test/java/com/alibaba/druid/sql/OnlineSQLTest2.java

@@ -0,0 +1,87 @@
+package com.alibaba.druid.sql;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.ResultSet;
+import java.sql.Statement;
+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;
+
+public class OnlineSQLTest2 extends TestCase {
+
+    private String url      = "jdbc:mysql://10.20.36.26/dragoon_v25_monitordb_test";
+    private String user     = "dragoon";
+    private String password = "dragoon";
+
+    public void test_list_sql() throws Exception {
+
+        Connection conn = DriverManager.getConnection(url, user, password);
+
+        int count = 0;
+        String sql = "SELECT id, value FROM m_sql_const LIMIT 100";
+        Statement stmt = conn.createStatement();
+        ResultSet rs = stmt.executeQuery(sql);
+        while (rs.next()) {
+            int id = rs.getInt(1);
+            String value = rs.getString(2);
+
+            if (value.indexOf('（') != -1) {
+                continue;
+            }
+
+            boolean sqlFlag = false;
+            String lowerSql = value.toLowerCase();
+            if (lowerSql.startsWith("insert") || lowerSql.startsWith("select") || lowerSql.startsWith("upate")
+                || lowerSql.startsWith("delete") || lowerSql.startsWith("create") || lowerSql.startsWith("drop")) {
+                sqlFlag = true;
+            }
+
+            if (!sqlFlag) {
+                continue;
+            }
+
+            System.out.println(value);
+            mysqlStat(id, lowerSql);
+            System.out.println();
+            count++;
+        }
+        rs.close();
+        stmt.close();
+
+        System.out.println("COUNT : " + count);
+
+        conn.close();
+    }
+
+    void mysqlStat(int id, String sql) throws Exception {
+        sql = sql.trim();
+        boolean sqlFlag = false;
+        String lowerSql = sql.toLowerCase();
+        if (lowerSql.startsWith("insert") || lowerSql.startsWith("select") || lowerSql.startsWith("upate") || lowerSql.startsWith("delete")
+            || lowerSql.startsWith("create") || lowerSql.startsWith("drop")) {
+            sqlFlag = true;
+        }
+
+        if (!sqlFlag) {
+            return;
+        }
+
+        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("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+    }
+}

src/test/java/com/alibaba/druid/sql/mysql/bvt/visitor/MySqlSchemaStatVisitorTest1.java

@@ -0,0 +1,42 @@
+package com.alibaba.druid.sql.mysql.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.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor.Column;
+
+public class MySqlSchemaStatVisitorTest1 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")));
+
+    }
+}

src/test/java/com/alibaba/druid/sql/mysql/bvt/visitor/MySqlSchemaStatVisitorTest2.java

@@ -0,0 +1,89 @@
+package com.alibaba.druid.sql.mysql.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.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor.Column;
+
+public class MySqlSchemaStatVisitorTest2 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/mysql/bvt/visitor/MySqlSchemaStatVisitorTest3.java

@@ -0,0 +1,45 @@
+package com.alibaba.druid.sql.mysql.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.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor.Column;
+
+public class MySqlSchemaStatVisitorTest3 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/mysql/bvt/visitor/MySqlSchemaStatVisitorTest4.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.mysql.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.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor.Column;
+
+public class MySqlSchemaStatVisitorTest4 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/mysql/bvt/visitor/MySqlSchemaStatVisitorTest_Delete.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.mysql.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.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor.Column;
+
+public class MySqlSchemaStatVisitorTest_Delete extends TestCase {
+
+    public void test_0() throws Exception {
+        String sql = "delete from users where id = ? or name = ?";
+
+        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/mysql/bvt/visitor/MySqlSchemaStatVisitorTest_Update.java

@@ -0,0 +1,40 @@
+package com.alibaba.druid.sql.mysql.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.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor.Column;
+
+public class MySqlSchemaStatVisitorTest_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")));
+
+    }
+
+}