sql parser support drop table

2023-03-04 19:17:16 +00:00 · 2012-02-22 20:25:36 +08:00
parent 7473d3f8a0
commit 5b6b8b8b76
4 changed files with 109 additions and 9 deletions
@@ -0,0 +1,41 @@
+package com.alibaba.druid.bvt.sql.postgresql;
+
+import java.util.List;
+
+import junit.framework.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 DropTableTest extends PGTest {
+
+    public void test_0() throws Exception {
+        String sql = "DROP TABLE films, distributors;";
+
+        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
+        List<SQLStatement> statementList = parser.parseStatementList();
+        SQLStatement statemen = statementList.get(0);
+        System.out.println(output(statementList));
+
+        Assert.assertEquals(1, statementList.size());
+
+        PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
+        statemen.accept(visitor);
+
+        System.out.println("Tables : " + visitor.getTables());
+        System.out.println("fields : " + visitor.getFields());
+
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("films")));
+        Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("distributors")));
+        
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("films")).getDropCount() == 1);
+        Assert.assertTrue(visitor.getTables().get(new TableStat.Name("distributors")).getDropCount() == 1);
+
+        Assert.assertTrue(visitor.getFields().size() == 0);
+    }
+
+   
+}