improved sql parser for postgresql

src/test/java/com/alibaba/druid/bvt/sql/postgresql/datatypes/BitStringTest.java

@@ -0,0 +1,20 @@
+package com.alibaba.druid.bvt.sql.postgresql.datatypes;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryExpr;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGExprParser;
+import com.alibaba.druid.util.JdbcConstants;
+
+public class BitStringTest extends PGTest {
+
+    public void test_timestamp() throws Exception {
+        String sql = "B'101'";
+        PGExprParser parser = new PGExprParser(sql);
+        SQLBinaryExpr expr = (SQLBinaryExpr) parser.expr();
+        Assert.assertEquals("B'101'", SQLUtils.toSQLString(expr, JdbcConstants.POSTGRESQL));
+    }
+
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableAddCheck0.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableAddCheck0 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ADD CHECK (name <> '');";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableAddColumnTest0.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableAddColumnTest0 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ADD COLUMN description text;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableAddColumnTest1.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableAddColumnTest1 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ADD COLUMN description text CHECK (description <> '');";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableAddFk0.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableAddFk0 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ADD FOREIGN KEY (product_group_id) REFERENCES product_groups;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableAddUnique0.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableAddUnique0 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ADD CONSTRAINT some_name UNIQUE (product_no);";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableDropColumnTest0.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableDropColumnTest0 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products DROP COLUMN description;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableDropColumnTest1.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableDropColumnTest1 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products DROP COLUMN description CASCADE;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableDropConstraint.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableDropConstraint extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products DROP CONSTRAINT some_name;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 0);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableDropDefault.java

@@ -0,0 +1,41 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableDropDefault extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ALTER COLUMN price DROP DEFAULT;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement stmt = statementList.get(0);
+        print(statementList);
+        
+        Assert.assertEquals("ALTER TABLE products"
+                + "\n\tALTER COLUMN price DROP DEFAULT", stmt.toString());
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        stmt.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableDropNotNull.java

@@ -0,0 +1,41 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableDropNotNull extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ALTER COLUMN product_no DROP NOT NULL;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement stmt = statementList.get(0);
+        print(statementList);
+        
+        Assert.assertEquals("ALTER TABLE products"
+                + "\n\tALTER COLUMN product_no DROP NOT NULL", stmt.toString());
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        stmt.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableRenameTest.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableRenameTest extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products RENAME TO items;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 0);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableRenameTest1.java

@@ -0,0 +1,38 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableRenameTest1 extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products RENAME COLUMN product_no TO product_number;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        print(statementList);
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 2);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableSetDefault.java

@@ -0,0 +1,41 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableSetDefault extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ALTER COLUMN price SET DEFAULT 7.77;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement stmt = statementList.get(0);
+        print(statementList);
+        
+        Assert.assertEquals("ALTER TABLE products"
+                + "\n\tALTER COLUMN price SET DEFAULT 7.77", stmt.toString());
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        stmt.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}

src/test/java/com/alibaba/druid/bvt/sql/postgresql/ddl/PGAlterTableSetNotNull.java

@@ -0,0 +1,41 @@
+package com.alibaba.druid.bvt.sql.postgresql.ddl;
+
+import java.util.List;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PGTest;
+import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser;
+import com.alibaba.druid.sql.dialect.postgresql.visitor.PGSchemaStatVisitor;
+import com.alibaba.druid.stat.TableStat;
+
+
+public class PGAlterTableSetNotNull extends PGTest {
+    public void test_0 () throws Exception {
+        String sql = "ALTER TABLE products ALTER COLUMN product_no SET NOT NULL;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement stmt = statementList.get(0);
+        print(statementList);
+        
+        Assert.assertEquals("ALTER TABLE products"
+                + "\n\tALTER COLUMN product_no SET NOT NULL", stmt.toString());
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        stmt.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getColumns());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("products")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getDropCount() == 0);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("products")).getAlterCount() == 1);
+
+        Assert.assertTrue(visitor.getColumns().size() == 1);
+    }
+}