bug fixed for mysql parser

2023-03-04 19:17:16 +00:00 · 2012-03-12 11:19:49 +08:00
parent efacfa93b7 786859fc71
commit 8efca053d6
10 changed files with 302 additions and 4 deletions
@@ -0,0 +1,39 @@
+package com.alibaba.druid.bvt.sql.mysql;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.MysqlTest;
+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;
+
+public class MySqlSelectTest_3 extends MysqlTest {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT 'Monty!' REGEXP '.*'";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("orderBy : " + visitor.getOrderByColumns());
+        
+        Assert.assertEquals(0, visitor.getTables().size());
+        Assert.assertEquals(0, visitor.getColumns().size());
+        Assert.assertEquals(0, visitor.getConditions().size());
+
+//        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("mytable")));
+    }
+}
@@ -0,0 +1,39 @@
+package com.alibaba.druid.bvt.sql.mysql;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.MysqlTest;
+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;
+
+public class MySqlSelectTest_4 extends MysqlTest {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT 'Monty!' NOT REGEXP '.*'";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("orderBy : " + visitor.getOrderByColumns());
+        
+        Assert.assertEquals(0, visitor.getTables().size());
+        Assert.assertEquals(0, visitor.getColumns().size());
+        Assert.assertEquals(0, visitor.getConditions().size());
+
+//        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("mytable")));
+    }
+}
@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.mysql;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.MysqlTest;
+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 MySqlSelectTest_5 extends MysqlTest {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT 'Monty!' RLIKE '.*'";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("orderBy : " + visitor.getOrderByColumns());
+        
+        Assert.assertEquals(0, visitor.getTables().size());
+        Assert.assertEquals(0, visitor.getColumns().size());
+        Assert.assertEquals(0, visitor.getConditions().size());
+
+//        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("mytable")));
+    }
+}
@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.mysql;
+
+import java.util.List;
+
+import junit.framework.Assert;
+
+import com.alibaba.druid.sql.MysqlTest;
+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 MySqlSelectTest_6 extends MysqlTest {
+
+    public void test_0() throws Exception {
+        String sql = "SELECT 'Monty!' NOT RLIKE '.*'";
+
+        MySqlStatementParser parser = new MySqlStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+        System.out.println("coditions : " + visitor.getConditions());
+        System.out.println("orderBy : " + visitor.getOrderByColumns());
+        
+        Assert.assertEquals(0, visitor.getTables().size());
+        Assert.assertEquals(0, visitor.getColumns().size());
+        Assert.assertEquals(0, visitor.getConditions().size());
+
+//        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("mytable")));
+    }
+}
@@ -0,0 +1,113 @@
+package com.alibaba.druid.mysql;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+import java.sql.Statement;
+
+import junit.framework.TestCase;
+
+import com.alibaba.druid.pool.DruidDataSource;
+
+public class MySqlBigTableTest extends TestCase {
+    final int COUNT = 800;
+    
+    private String          jdbcUrl;
+    private String          user;
+    private String          password;
+    private String          driverClass;
+
+    private DruidDataSource dataSource;
+
+    protected void setUp() throws Exception {
+        jdbcUrl = "jdbc:mysql://10.20.144.27:3306/dragoon_v25_masterdb";
+        user = "dragoon_test";
+        password = "dragoon_test";
+        driverClass = "com.mysql.jdbc.Driver";
+
+        dataSource = new DruidDataSource();
+        dataSource.setDriverClassName(driverClass);
+        dataSource.setUrl(jdbcUrl);
+        dataSource.setPoolPreparedStatements(true);
+        dataSource.setUsername(user);
+        dataSource.setPassword(password);
+
+        createTable();
+    }
+
+    protected void tearDown() throws Exception {
+        try {
+            dropTable();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+
+        if (dataSource != null) {
+            dataSource.close();
+        }
+    }
+
+    public void test_0() throws Exception {
+        
+        StringBuffer ddl = new StringBuffer();
+        ddl.append("INSERT INTO t_big (");
+        for (int i = 0; i < COUNT; ++i) {
+            if (i != 0) {
+                ddl.append(", ");
+            }
+            ddl.append("F" + i);
+        }
+        ddl.append(") VALUES (");
+        for (int i = 0; i < COUNT; ++i) {
+            if (i != 0) {
+                ddl.append(", ");
+            }
+            ddl.append("?");
+        }
+        ddl.append(")");
+
+        Connection conn = dataSource.getConnection();
+
+        System.out.println(ddl.toString());
+
+        PreparedStatement stmt = conn.prepareStatement(ddl.toString());
+
+        for (int i = 0; i < COUNT; ++i) {
+            stmt.setInt(i + 1, i);
+        }
+        stmt.execute();
+        stmt.close();
+
+        conn.close();
+    }
+
+    private void dropTable() throws SQLException {
+
+        Connection conn = dataSource.getConnection();
+
+        Statement stmt = conn.createStatement();
+        stmt.execute("DROP TABLE t_big");
+        stmt.close();
+
+        conn.close();
+    }
+
+    private void createTable() throws SQLException {
+        StringBuffer ddl = new StringBuffer();
+        ddl.append("CREATE TABLE t_big (FID INT AUTO_INCREMENT PRIMARY KEY ");
+        for (int i = 0; i < COUNT; ++i) {
+            ddl.append(", ");
+            ddl.append("F" + i);
+            ddl.append(" BIGINT NULL");
+        }
+        ddl.append(")");
+
+        Connection conn = dataSource.getConnection();
+
+        Statement stmt = conn.createStatement();
+        stmt.execute(ddl.toString());
+        stmt.close();
+
+        conn.close();
+    }
+}