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

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

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

@@ -0,0 +1,90 @@
+package com.alibaba.druid.sql.mysql.bvt.visitor;
+
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.Reader;
+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.MySqlOutputVisitor;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlParameterizedOutputVisitor;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.util.JdbcUtils;
+
+public class MySqlResourceTest extends TestCase {
+
+    public void test_0() throws Exception {
+        exec_test("bvt/parser/mysql-0.txt");
+        exec_test("bvt/parser/mysql-1.txt");
+        exec_test("bvt/parser/mysql-2.txt");
+        exec_test("bvt/parser/mysql-3.txt");
+        exec_test("bvt/parser/mysql-4.txt");
+        exec_test("bvt/parser/mysql-5.txt");
+        exec_test("bvt/parser/mysql-6.txt");
+        exec_test("bvt/parser/mysql-7.txt");
+        exec_test("bvt/parser/mysql-8.txt");
+        exec_test("bvt/parser/mysql-9.txt");
+        exec_test("bvt/parser/mysql-10.txt");
+    }
+
+    public void exec_test(String resource) throws Exception {
+        System.out.println(resource);
+        InputStream is = null;
+
+        is = Thread.currentThread().getContextClassLoader().getResourceAsStream(resource);
+        Reader reader = new InputStreamReader(is, "UTF-8");
+        String input = JdbcUtils.read(reader);
+        JdbcUtils.close(reader);
+        String[] items = input.split("---------------------------");
+        String sql = items[0].trim();
+        String expect = items[1].trim();
+        
+        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());
+        
+        System.out.println();
+        System.out.println();
+    }
+
+    void mergValidate(String sql, String expect) {
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+
+        Assert.assertEquals(1, statementList.size());
+
+        StringBuilder out = new StringBuilder();
+        MySqlParameterizedOutputVisitor visitor = new MySqlParameterizedOutputVisitor(out);
+        statemen.accept(visitor);
+
+        System.out.println(out.toString());
+
+        Assert.assertEquals(expect, out.toString());
+    }
+
+    private String output(List<SQLStatement> stmtList) {
+        StringBuilder out = new StringBuilder();
+        MySqlOutputVisitor visitor = new MySqlOutputVisitor(out);
+
+        for (SQLStatement stmt : stmtList) {
+            stmt.accept(visitor);
+        }
+
+        return out.toString();
+    }
+}

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

@@ -39,4 +39,32 @@ public class MySqlSchemaStatVisitorTest1 extends TestCase {
         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")));
+
+    }
 }