diff --git a/src/main/java/com/alibaba/druid/sql/parser/SQLExprParser.java b/src/main/java/com/alibaba/druid/sql/parser/SQLExprParser.java index 1423971f0c..27605810e4 100644 --- a/src/main/java/com/alibaba/druid/sql/parser/SQLExprParser.java +++ b/src/main/java/com/alibaba/druid/sql/parser/SQLExprParser.java @@ -1516,7 +1516,7 @@ public class SQLExprParser extends SQLParser { if (lexer.token() == Token.KEY) { lexer.nextToken(); } - column.getConstraints().add(new SQLColumnPrimaryKey()); + column.getConstraints().add(new SQLColumnUniqueKey()); return parseColumnRest(column); } diff --git a/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest49.java b/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest49.java index a9e238989d..6edd5b7409 100644 --- a/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest49.java +++ b/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest49.java @@ -56,7 +56,7 @@ public class MySqlCreateTableTest49 extends MysqlTest { String output = SQLUtils.toMySqlString(stmt); Assert.assertEquals("CREATE TABLE tauth (" // - + "\n\tcid varchar(36) NOT NULL PRIMARY KEY, "// + + "\n\tcid varchar(36) NOT NULL UNIQUE, "// + "\n\tcdesc varchar(200), "// + "\n\tcname varchar(100) NOT NULL, "// + "\n\tcseq decimal(22, 0), "// diff --git a/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest58.java b/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest58.java new file mode 100644 index 0000000000..26a89ac9dd --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/sql/mysql/MySqlCreateTableTest58.java @@ -0,0 +1,63 @@ +/* + * Copyright 1999-2011 Alibaba Group Holding Ltd. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.alibaba.druid.bvt.sql.mysql; + +import org.junit.Assert; +import org.junit.Test; + +import com.alibaba.druid.sql.MysqlTest; +import com.alibaba.druid.sql.SQLUtils; +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 MySqlCreateTableTest58 extends MysqlTest { + + @Test + public void test_one() throws Exception { + String sql = "CREATE TABLE `appservice_account` (" + "`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY," + + "`password` varchar(128) NOT NULL," + "`last_login` datetime NOT NULL," + + "`username` varchar(40) NOT NULL UNIQUE," + "`date_of_birth` date NOT NULL," + + "`head` varchar(100) NOT NULL," + "`headThumb` varchar(100) NOT NULL," + + "`name` varchar(50) NOT NULL," + "`gender` integer," + "`uploadVideoCount` integer NOT NULL," + + "`fansCount` integer NOT NULL," + "`balance` numeric(19, 2) NOT NULL," + + "`brick` integer NOT NULL," + "`reward` integer NOT NULL," + "`token` varchar(500) NOT NULL," + + "`weiboUserid` varchar(50) NOT NULL," + "`weiboAccesstoken` varchar(500) NOT NULL," + + "`qqUserid` varchar(50) NOT NULL," + "`qqAccesstoken` varchar(500) NOT NULL," + + "`wechatUserid` varchar(50) NOT NULL," + "`wechatAccesstoken` varchar(500) NOT NULL," + + "`is_active` bool NOT NULL," + "`is_admin` bool NOT NULL" + ")"; + + MySqlStatementParser parser = new MySqlStatementParser(sql); + SQLStatement stmt = parser.parseCreateTable(); + + MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor(); + stmt.accept(visitor); + + String output = SQLUtils.toMySqlString(stmt); + Assert.assertEquals("CREATE TABLE `appservice_account` (" + "\n\t`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, " + + "\n\t`password` varchar(128) NOT NULL, " + "\n\t`last_login` datetime NOT NULL, " + + "\n\t`username` varchar(40) NOT NULL UNIQUE, " + "\n\t`date_of_birth` date NOT NULL, " + + "\n\t`head` varchar(100) NOT NULL, " + "\n\t`headThumb` varchar(100) NOT NULL, " + + "\n\t`name` varchar(50) NOT NULL, " + "\n\t`gender` integer, " + "\n\t`uploadVideoCount` integer NOT NULL, " + + "\n\t`fansCount` integer NOT NULL, " + "\n\t`balance` numeric(19, 2) NOT NULL, " + + "\n\t`brick` integer NOT NULL, " + "\n\t`reward` integer NOT NULL, " + "\n\t`token` varchar(500) NOT NULL, " + + "\n\t`weiboUserid` varchar(50) NOT NULL, " + "\n\t`weiboAccesstoken` varchar(500) NOT NULL, " + + "\n\t`qqUserid` varchar(50) NOT NULL, " + "\n\t`qqAccesstoken` varchar(500) NOT NULL, " + + "\n\t`wechatUserid` varchar(50) NOT NULL, " + "\n\t`wechatAccesstoken` varchar(500) NOT NULL, " + + "\n\t`is_active` bool NOT NULL, " + "\n\t`is_admin` bool NOT NULL" + "\n)",output); + + } +}