Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

improved sql parser SchemaStatVisitor.

Browse Source
This commit is contained in:
wenshao
2016-09-27 11:11:32 +08:00
parent f1eb24a1de
commit 4833b71e83
11 changed files with 153 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableAddIndex_0.java
+11
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import junit.framework.TestCase;
import org.junit.Assert;
@@ -37,6 +40,14 @@ public class MySqlAlterTableAddIndex_0 extends TestCase {
Assert.assertEquals("alter table `test`.`tb1`" + //
"\n\tadd index `ix` (`f2` asc)", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("test.tb1");
assertNotNull(tableStat);
assertEquals(1, tableStat.getCreateIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableAddIndex_1.java
+11
View File
@@ -15,6 +15,10 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import org.junit.Assert;
import junit.framework.TestCase;
@@ -36,6 +40,13 @@ public class MySqlAlterTableAddIndex_1 extends TestCase {
Assert.assertEquals("alter table `test`.`tb1`" + //
"\n\tadd unique index `ix2` (`fid` asc)", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = new SQLUtils().createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("test.tb1");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getCreateIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableAddPrimaryKey.java
+10
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import org.junit.Assert;
import junit.framework.TestCase;
@@ -38,6 +41,13 @@ public class MySqlAlterTableAddPrimaryKey extends TestCase {
Assert.assertEquals("alter table `test`.`tb1`" + //
"\n\tchange column `fid` `fid` INT(11) not null default null,\n\t" + //
"add primary key (`fid`)", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = new SQLUtils().createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("test.tb1");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getCreateIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableAddPrimaryKey_1.java
+10
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import junit.framework.TestCase;
import org.junit.Assert;
@@ -37,6 +40,13 @@ public class MySqlAlterTableAddPrimaryKey_1 extends TestCase {
Assert.assertEquals("alter table tabelname" + //
"\n\tadd constraint mYconstraint primary key (id)", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = new SQLUtils().createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("tabelname");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getCreateIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableAddUniqueTest.java
+10
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import junit.framework.TestCase;
import org.junit.Assert;
@@ -37,6 +40,13 @@ public class MySqlAlterTableAddUniqueTest extends TestCase {
Assert.assertEquals("alter table icp.wx_msg"
+ "\n\tadd unique (msgId, msgType, event, eventKey)", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = new SQLUtils().createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("icp.wx_msg");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getCreateIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableDropIndex_0.java
+10
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import org.junit.Assert;
import junit.framework.TestCase;
@@ -36,6 +39,13 @@ public class MySqlAlterTableDropIndex_0 extends TestCase {
Assert.assertEquals("alter table `test`.`tb1`" + //
"\n\tdrop index `ix`", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = new SQLUtils().createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("test.tb1");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getDropIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableDropKey.java
+10
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import junit.framework.TestCase;
import org.junit.Assert;
@@ -37,6 +40,13 @@ public class MySqlAlterTableDropKey extends TestCase {
Assert.assertEquals("alter table t6" + //
"\n\tdrop key v", SQLUtils.toMySqlString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
SchemaStatVisitor visitor = new SQLUtils().createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
TableStat tableStat = visitor.getTableStat("t6");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getDropIndexCount());
}
}
src/test/java/com/alibaba/druid/bvt/sql/mysql/alter/MySqlAlterTableTest_addFk.java
+9 -1
View File
@@ -15,6 +15,9 @@
*/
package com.alibaba.druid.bvt.sql.mysql.alter;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import junit.framework.TestCase;
import org.junit.Assert;
@@ -34,7 +37,7 @@ public class MySqlAlterTableTest_addFk extends TestCase {
SQLStatement stmt = parser.parseStatementList().get(0);
parser.match(Token.EOF);
MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
// System.out.println("Tables : " + visitor.getTables());
@@ -54,5 +57,10 @@ public class MySqlAlterTableTest_addFk extends TestCase {
Assert.assertEquals(2, visitor.getTables().size());
Assert.assertEquals(2, visitor.getColumns().size());
TableStat tableStat = visitor.getTableStat("Test2");
assertNotNull(tableStat);
assertEquals(1, tableStat.getAlterCount());
assertEquals(1, tableStat.getCreateIndexCount());
}
}