mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
improved sql parser.
This commit is contained in:
wenshao
+204
-16
@@ -20,11 +20,14 @@ import com.alibaba.druid.sql.SQLUtils;
|
||||
import com.alibaba.druid.sql.ast.SQLStatement;
|
||||
import com.alibaba.druid.sql.ast.statement.SQLAlterTableStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlCreateTableStatement;
|
||||
import com.alibaba.druid.sql.repository.SchemaRepository;
|
||||
import com.alibaba.druid.util.JdbcConstants;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
private SchemaRepository repository = new SchemaRepository(JdbcConstants.MYSQL);
|
||||
|
||||
|
||||
public void test_0() throws Exception {
|
||||
String sql = "CREATE TABLE `test1` (\n" +
|
||||
@@ -52,16 +55,13 @@ public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
") ENGINE=InnoDB AUTO_INCREMENT=1769503 DEFAULT CHARSET=utf8mb4 COMMENT='10000000'";
|
||||
|
||||
|
||||
MySqlCreateTableStatement createTableStmt;
|
||||
{
|
||||
List<SQLStatement> statementList = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL, true);
|
||||
assertEquals(1, statementList.size());
|
||||
createTableStmt = (MySqlCreateTableStatement) statementList.get(0);
|
||||
}
|
||||
repository.acceptDDL(sql);
|
||||
|
||||
MySqlCreateTableStatement createTableStmt = (MySqlCreateTableStatement) repository.findTable("test1").getStatement();
|
||||
assertEquals(21, createTableStmt.getTableElementList().size());
|
||||
|
||||
StringBuilder buf = new StringBuilder();
|
||||
createTableStmt.showCoumns(buf);
|
||||
//StringBuilder buf = new StringBuilder();
|
||||
//createTableStmt.showCoumns(buf);
|
||||
|
||||
assertEquals("+--------------+---------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
@@ -83,7 +83,7 @@ public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
"| c_text | text | NO | | NULL | |\n" +
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"+--------------+---------------+------+-----+---------------------+-----------------------------+\n", buf.toString());
|
||||
"+--------------+---------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
|
||||
SQLAlterTableStatement stmtDropColumn;
|
||||
{
|
||||
@@ -94,8 +94,6 @@ public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
createTableStmt.apply(stmtDropColumn);
|
||||
assertEquals(20, createTableStmt.getTableElementList().size());
|
||||
|
||||
buf.delete(0, buf.length());
|
||||
createTableStmt.showCoumns(buf);
|
||||
assertEquals("+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
@@ -115,7 +113,7 @@ public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
"| c_text | text | NO | | NULL | |\n" +
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n", buf.toString());
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
|
||||
SQLAlterTableStatement stmtAddColumn;
|
||||
{
|
||||
@@ -126,8 +124,6 @@ public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
createTableStmt.apply(stmtAddColumn);
|
||||
assertEquals(21, createTableStmt.getTableElementList().size());
|
||||
|
||||
buf.delete(0, buf.length());
|
||||
createTableStmt.showCoumns(buf);
|
||||
assertEquals("+--------------+---------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
"+--------------+---------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
@@ -148,8 +144,200 @@ public class MySqlCreateTable_showColumns_repository_test extends MysqlTest {
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"| c_decimal | decimal(10,3) | NO | | NULL | |\n" +
|
||||
"+--------------+---------------+------+-----+---------------------+-----------------------------+\n", buf.toString());
|
||||
"+--------------+---------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
|
||||
//System.out.println(buf.toString());
|
||||
SQLAlterTableStatement stmtChangeColumn;
|
||||
{
|
||||
List<SQLStatement> statementList = SQLUtils.parseStatements("ALTER TABLE test1 CHANGE COLUMN c_decimal c_decimal_1 INT(11) NOT NULL DEFAULT NULL FIRST id;", JdbcConstants.MYSQL, true);
|
||||
assertEquals(1, statementList.size());
|
||||
stmtChangeColumn = (SQLAlterTableStatement) statementList.get(0);
|
||||
}
|
||||
createTableStmt.apply(stmtChangeColumn);
|
||||
assertEquals(21, createTableStmt.getTableElementList().size());
|
||||
//String sql = "ALTER TABLE `test`.`tb1` CHANGE COLUMN `fid` `fid` INT(11) NOT NULL DEFAULT NULL, ADD PRIMARY KEY (`fid`) ;";
|
||||
|
||||
assertEquals("+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| c_decimal_1 | INT(11) | YES | | NULL | |\n" +
|
||||
"| id | bigint(20) | YES | PRI | NULL | auto_increment |\n" +
|
||||
"| c_tinyint | tinyint(4) | NO | PRI | 1 | |\n" +
|
||||
"| c_smallint | smallint(6) | NO | | 0 | |\n" +
|
||||
"| c_mediumint | mediumint(9) | NO | | NULL | |\n" +
|
||||
"| c_int | int(11) | NO | | NULL | |\n" +
|
||||
"| c_bigint | bigint(20) | NO | | NULL | |\n" +
|
||||
"| c_date | date | NO | | 0000-00-00 | |\n" +
|
||||
"| c_datetime | datetime | NO | | 0000-00-00 00:00:00 | |\n" +
|
||||
"| c_timestamp | timestamp | NO | | NULL | on update CURRENT_TIMESTAMP |\n" +
|
||||
"| c_time | time | NO | | NULL | |\n" +
|
||||
"| c_char | char(10) | NO | MUL | NULL | |\n" +
|
||||
"| c_varchar | varchar(10) | NO | MUL | hello | |\n" +
|
||||
"| c_blob | blob | NO | | NULL | |\n" +
|
||||
"| c_text | text | NO | | NULL | |\n" +
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
|
||||
|
||||
SQLAlterTableStatement stmtDropPk;
|
||||
{
|
||||
List<SQLStatement> statementList = SQLUtils.parseStatements("ALTER TABLE test1 DROP PRIMARY KEY;", JdbcConstants.MYSQL, true);
|
||||
assertEquals(1, statementList.size());
|
||||
stmtDropPk = (SQLAlterTableStatement) statementList.get(0);
|
||||
}
|
||||
createTableStmt.apply(stmtDropPk);
|
||||
assertEquals(20, createTableStmt.getTableElementList().size());
|
||||
|
||||
assertEquals("CREATE TABLE `test1` (\n" +
|
||||
"\tc_decimal_1 INT(11) NOT NULL DEFAULT NULL,\n" +
|
||||
"\t`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'id',\n" +
|
||||
"\t`c_tinyint` tinyint(4) DEFAULT '1' COMMENT 'tinyint',\n" +
|
||||
"\t`c_smallint` smallint(6) DEFAULT 0 COMMENT 'smallint',\n" +
|
||||
"\t`c_mediumint` mediumint(9) DEFAULT NULL COMMENT 'mediumint',\n" +
|
||||
"\t`c_int` int(11) DEFAULT NULL COMMENT 'int',\n" +
|
||||
"\t`c_bigint` bigint(20) DEFAULT NULL COMMENT 'bigint',\n" +
|
||||
"\t`c_date` date DEFAULT '0000-00-00' COMMENT 'date',\n" +
|
||||
"\t`c_datetime` datetime DEFAULT '0000-00-00 00:00:00' COMMENT 'datetime',\n" +
|
||||
"\t`c_timestamp` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP COMMENT 'timestamp',\n" +
|
||||
"\t`c_time` time DEFAULT NULL COMMENT 'time',\n" +
|
||||
"\t`c_char` char(10) DEFAULT NULL COMMENT 'char',\n" +
|
||||
"\t`c_varchar` varchar(10) DEFAULT 'hello' COMMENT 'varchar',\n" +
|
||||
"\t`c_blob` blob COMMENT 'blob',\n" +
|
||||
"\t`c_text` text COMMENT 'text',\n" +
|
||||
"\t`c_mediumtext` mediumtext COMMENT 'mediumtext',\n" +
|
||||
"\t`c_longblob` longblob COMMENT 'longblob',\n" +
|
||||
"\tUNIQUE `uk_a` (`c_varchar`, `c_mediumint`),\n" +
|
||||
"\tKEY `k_c` (`c_tinyint`, `c_int`),\n" +
|
||||
"\tKEY `k_d` (`c_char`, `c_bigint`)\n" +
|
||||
") ENGINE = InnoDB AUTO_INCREMENT = 1769503 CHARSET = utf8mb4 COMMENT = '10000000'", repository.console("show create table test1"));
|
||||
|
||||
|
||||
assertEquals("+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| c_decimal_1 | INT(11) | YES | | NULL | |\n" +
|
||||
"| id | bigint(20) | YES | | NULL | auto_increment |\n" +
|
||||
"| c_tinyint | tinyint(4) | NO | MUL | 1 | |\n" +
|
||||
"| c_smallint | smallint(6) | NO | | 0 | |\n" +
|
||||
"| c_mediumint | mediumint(9) | NO | | NULL | |\n" +
|
||||
"| c_int | int(11) | NO | | NULL | |\n" +
|
||||
"| c_bigint | bigint(20) | NO | | NULL | |\n" +
|
||||
"| c_date | date | NO | | 0000-00-00 | |\n" +
|
||||
"| c_datetime | datetime | NO | | 0000-00-00 00:00:00 | |\n" +
|
||||
"| c_timestamp | timestamp | NO | | NULL | on update CURRENT_TIMESTAMP |\n" +
|
||||
"| c_time | time | NO | | NULL | |\n" +
|
||||
"| c_char | char(10) | NO | MUL | NULL | |\n" +
|
||||
"| c_varchar | varchar(10) | NO | MUL | hello | |\n" +
|
||||
"| c_blob | blob | NO | | NULL | |\n" +
|
||||
"| c_text | text | NO | | NULL | |\n" +
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
|
||||
SQLAlterTableStatement stmtDropIdx;
|
||||
{
|
||||
List<SQLStatement> statementList = SQLUtils.parseStatements("ALTER TABLE test1 DROP INDEX k_d;", JdbcConstants.MYSQL, true);
|
||||
assertEquals(1, statementList.size());
|
||||
stmtDropIdx = (SQLAlterTableStatement) statementList.get(0);
|
||||
}
|
||||
createTableStmt.apply(stmtDropIdx);
|
||||
assertEquals(19, createTableStmt.getTableElementList().size());
|
||||
|
||||
assertEquals("CREATE TABLE `test1` (\n" +
|
||||
"\tc_decimal_1 INT(11) NOT NULL DEFAULT NULL,\n" +
|
||||
"\t`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'id',\n" +
|
||||
"\t`c_tinyint` tinyint(4) DEFAULT '1' COMMENT 'tinyint',\n" +
|
||||
"\t`c_smallint` smallint(6) DEFAULT 0 COMMENT 'smallint',\n" +
|
||||
"\t`c_mediumint` mediumint(9) DEFAULT NULL COMMENT 'mediumint',\n" +
|
||||
"\t`c_int` int(11) DEFAULT NULL COMMENT 'int',\n" +
|
||||
"\t`c_bigint` bigint(20) DEFAULT NULL COMMENT 'bigint',\n" +
|
||||
"\t`c_date` date DEFAULT '0000-00-00' COMMENT 'date',\n" +
|
||||
"\t`c_datetime` datetime DEFAULT '0000-00-00 00:00:00' COMMENT 'datetime',\n" +
|
||||
"\t`c_timestamp` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP COMMENT 'timestamp',\n" +
|
||||
"\t`c_time` time DEFAULT NULL COMMENT 'time',\n" +
|
||||
"\t`c_char` char(10) DEFAULT NULL COMMENT 'char',\n" +
|
||||
"\t`c_varchar` varchar(10) DEFAULT 'hello' COMMENT 'varchar',\n" +
|
||||
"\t`c_blob` blob COMMENT 'blob',\n" +
|
||||
"\t`c_text` text COMMENT 'text',\n" +
|
||||
"\t`c_mediumtext` mediumtext COMMENT 'mediumtext',\n" +
|
||||
"\t`c_longblob` longblob COMMENT 'longblob',\n" +
|
||||
"\tUNIQUE `uk_a` (`c_varchar`, `c_mediumint`),\n" +
|
||||
"\tKEY `k_c` (`c_tinyint`, `c_int`)\n" +
|
||||
") ENGINE = InnoDB AUTO_INCREMENT = 1769503 CHARSET = utf8mb4 COMMENT = '10000000'", repository.console("show create table test1"));
|
||||
|
||||
assertEquals("+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| c_decimal_1 | INT(11) | YES | | NULL | |\n" +
|
||||
"| id | bigint(20) | YES | | NULL | auto_increment |\n" +
|
||||
"| c_tinyint | tinyint(4) | NO | MUL | 1 | |\n" +
|
||||
"| c_smallint | smallint(6) | NO | | 0 | |\n" +
|
||||
"| c_mediumint | mediumint(9) | NO | | NULL | |\n" +
|
||||
"| c_int | int(11) | NO | | NULL | |\n" +
|
||||
"| c_bigint | bigint(20) | NO | | NULL | |\n" +
|
||||
"| c_date | date | NO | | 0000-00-00 | |\n" +
|
||||
"| c_datetime | datetime | NO | | 0000-00-00 00:00:00 | |\n" +
|
||||
"| c_timestamp | timestamp | NO | | NULL | on update CURRENT_TIMESTAMP |\n" +
|
||||
"| c_time | time | NO | | NULL | |\n" +
|
||||
"| c_char | char(10) | NO | | NULL | |\n" +
|
||||
"| c_varchar | varchar(10) | NO | MUL | hello | |\n" +
|
||||
"| c_blob | blob | NO | | NULL | |\n" +
|
||||
"| c_text | text | NO | | NULL | |\n" +
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
|
||||
|
||||
SQLAlterTableStatement stmtDropUk;
|
||||
{
|
||||
List<SQLStatement> statementList = SQLUtils.parseStatements("ALTER TABLE test1 DROP INDEX uk_a;", JdbcConstants.MYSQL, true);
|
||||
assertEquals(1, statementList.size());
|
||||
stmtDropUk = (SQLAlterTableStatement) statementList.get(0);
|
||||
}
|
||||
createTableStmt.apply(stmtDropUk);
|
||||
assertEquals(18, createTableStmt.getTableElementList().size());
|
||||
|
||||
assertEquals("CREATE TABLE `test1` (\n" +
|
||||
"\tc_decimal_1 INT(11) NOT NULL DEFAULT NULL,\n" +
|
||||
"\t`id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'id',\n" +
|
||||
"\t`c_tinyint` tinyint(4) DEFAULT '1' COMMENT 'tinyint',\n" +
|
||||
"\t`c_smallint` smallint(6) DEFAULT 0 COMMENT 'smallint',\n" +
|
||||
"\t`c_mediumint` mediumint(9) DEFAULT NULL COMMENT 'mediumint',\n" +
|
||||
"\t`c_int` int(11) DEFAULT NULL COMMENT 'int',\n" +
|
||||
"\t`c_bigint` bigint(20) DEFAULT NULL COMMENT 'bigint',\n" +
|
||||
"\t`c_date` date DEFAULT '0000-00-00' COMMENT 'date',\n" +
|
||||
"\t`c_datetime` datetime DEFAULT '0000-00-00 00:00:00' COMMENT 'datetime',\n" +
|
||||
"\t`c_timestamp` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP COMMENT 'timestamp',\n" +
|
||||
"\t`c_time` time DEFAULT NULL COMMENT 'time',\n" +
|
||||
"\t`c_char` char(10) DEFAULT NULL COMMENT 'char',\n" +
|
||||
"\t`c_varchar` varchar(10) DEFAULT 'hello' COMMENT 'varchar',\n" +
|
||||
"\t`c_blob` blob COMMENT 'blob',\n" +
|
||||
"\t`c_text` text COMMENT 'text',\n" +
|
||||
"\t`c_mediumtext` mediumtext COMMENT 'mediumtext',\n" +
|
||||
"\t`c_longblob` longblob COMMENT 'longblob',\n" +
|
||||
"\tKEY `k_c` (`c_tinyint`, `c_int`)\n" +
|
||||
") ENGINE = InnoDB AUTO_INCREMENT = 1769503 CHARSET = utf8mb4 COMMENT = '10000000'", createTableStmt.toString());
|
||||
|
||||
assertEquals("+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| Field | Type | Null | Key | Default | Extra |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n" +
|
||||
"| c_decimal_1 | INT(11) | YES | | NULL | |\n" +
|
||||
"| id | bigint(20) | YES | | NULL | auto_increment |\n" +
|
||||
"| c_tinyint | tinyint(4) | NO | MUL | 1 | |\n" +
|
||||
"| c_smallint | smallint(6) | NO | | 0 | |\n" +
|
||||
"| c_mediumint | mediumint(9) | NO | | NULL | |\n" +
|
||||
"| c_int | int(11) | NO | | NULL | |\n" +
|
||||
"| c_bigint | bigint(20) | NO | | NULL | |\n" +
|
||||
"| c_date | date | NO | | 0000-00-00 | |\n" +
|
||||
"| c_datetime | datetime | NO | | 0000-00-00 00:00:00 | |\n" +
|
||||
"| c_timestamp | timestamp | NO | | NULL | on update CURRENT_TIMESTAMP |\n" +
|
||||
"| c_time | time | NO | | NULL | |\n" +
|
||||
"| c_char | char(10) | NO | | NULL | |\n" +
|
||||
"| c_varchar | varchar(10) | NO | | hello | |\n" +
|
||||
"| c_blob | blob | NO | | NULL | |\n" +
|
||||
"| c_text | text | NO | | NULL | |\n" +
|
||||
"| c_mediumtext | mediumtext | NO | | NULL | |\n" +
|
||||
"| c_longblob | longblob | NO | | NULL | |\n" +
|
||||
"+--------------+--------------+------+-----+---------------------+-----------------------------+\n", repository.console("show columns from test1"));
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user