improved sql parser.

src/test/java/com/alibaba/druid/bvt/filter/wall/pg/PGWallTest1.java

@@ -0,0 +1,34 @@
+/*
+ * Copyright 1999-2101 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.filter.wall.pg;
+
+import com.alibaba.druid.wall.WallUtils;
+import junit.framework.TestCase;
+import org.junit.Assert;
+
+/**
+ * SQLServerWallTest
+ * 
+ * @author RaymondXiu
+ * @version 1.0, 2012-3-18
+ * @see
+ */
+public class PGWallTest1 extends TestCase {
+    public void test_wall() throws Exception {
+        Assert.assertTrue(WallUtils.isValidatePostgres(//
+        "select pg_encoding_to_char(encoding) from pg_database where datname = current_database()"));
+    }
+}

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

@@ -0,0 +1,107 @@
+/*
+ * Copyright 1999-2101 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.visitor;
+
+import com.alibaba.druid.sql.ast.SQLName;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
+import com.alibaba.druid.sql.ast.statement.SQLTableElement;
+import com.alibaba.druid.sql.ast.statement.SQLUnique;
+import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+import junit.framework.TestCase;
+import org.junit.Assert;
+
+import java.util.List;
+
+public class MySqlSchemaStatVisitorTest_pk_0 extends TestCase {
+
+	public void test_0() throws Exception {
+		String sql ="CREATE TABLE `m_dt` (\n" +
+				"  `id` int(10) unsigned NOT NULL AUTO_INCREMENT COMMENT '主键',\n" +
+				"  `gmt_create` datetime NOT NULL COMMENT '创建时间',\n" +
+				"  `gmt_modified` datetime NOT NULL COMMENT '修改时间',\n" +
+				"  `instance_id` int(11) NOT NULL COMMENT '实例id',\n" +
+				"  `schema_name` varchar(128) NOT NULL COMMENT '数据库schema名称',\n" +
+				"  `state` tinyint(4) NOT NULL DEFAULT '0' COMMENT ' x dd ',\n" +
+				"  `busi_user` varchar(64) DEFAULT NULL COMMENT 'JDBC业务用户',\n" +
+				"  `bzp` varchar(128) DEFAULT NULL COMMENT 'bcp',\n" +
+				"  `ecd` varchar(32) DEFAULT NULL COMMENT 'xxx',\n" +
+				"  `last_sync_time` datetime DEFAULT NULL COMMENT '最后同步时间',\n" +
+				"  `catalog_name` varchar(128) NOT NULL COMMENT '物理库名称',\n" +
+				"  `search_name` varchar(256) NOT NULL COMMENT '用于搜索，区分不同数据库的不同字段信息',\n" +
+				"  `db_type` tinyint(4) NOT NULL COMMENT '数据库类型，和meta_instance表一致',\n" +
+				"  `et` varchar(32) NOT NULL DEFAULT '' COMMENT 'et',\n" +
+				"  `ae` varchar(32) DEFAULT NULL COMMENT 'ae',\n" +
+				"  PRIMARY KEY (`id`),\n" +
+				"  UNIQUE KEY `uk_instanceid_schemaname` (`instance_id`,`catalog_name`,`schema_name`),\n" +
+				"  KEY `idx_schema_name` (`schema_name`),\n" +
+				"  KEY `idx_instance_id_state` (`instance_id`,`id`,`state`),\n" +
+				"  KEY `idx_search_name` (`search_name`(255))\n" +
+				") ENGINE=InnoDB AUTO_INCREMENT=408695 DEFAULT CHARSET=utf8 COMMENT='数据库表信息'";
+
+//		sql = "select columnName from table1 where id in (select id from table3 where name = ?)";
+		MySqlStatementParser parser = new MySqlStatementParser(sql);
+		List<SQLStatement> statementList = parser.parseStatementList();
+		SQLCreateTableStatement stmt = (SQLCreateTableStatement) statementList.get(0);
+
+		Assert.assertEquals(1, statementList.size());
+
+		MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
+		stmt.accept(visitor);
+
+		System.out.println(sql);
+		System.out.println("Tables : " + visitor.getTables());
+		System.out.println("fields : " + visitor.getColumns());
+
+		Assert.assertEquals(1, visitor.getTables().size());
+		Assert.assertEquals(true, visitor.containsTable("m_dt"));
+
+		Assert.assertEquals(15, visitor.getColumns().size());
+
+		{
+			TableStat.Column column = visitor.getColumn("m_dt", "id");
+			assertNotNull(column);
+			assertTrue(column.isPrimaryKey());
+		}
+		{
+			TableStat.Column column = visitor.getColumn("m_dt", "schema_name");
+			assertNotNull(column);
+			assertTrue(column.isUnique());
+		}
+		{
+			TableStat.Column column = visitor.getColumn("m_dt", "instance_id");
+			assertNotNull(column);
+			assertTrue(column.isUnique());
+		}
+
+		for (SQLTableElement element : stmt.getTableElementList()) {
+			if (element instanceof SQLUnique) {
+				SQLName name = ((SQLUnique) element).getName();
+				if (name != null) {
+					String uniqueName = name.toString();
+				}
+			}
+		}
+		// Assert.assertEquals(true, visitor.getFields().contains(new
+		// Column("users", "id")));
+		// Assert.assertEquals(true, visitor.getFields().contains(new
+		// Column("users", "name")));
+
+	}
+
+}