Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

pg sql parser support insert

Browse Source
This commit is contained in:
温少
2012-02-23 09:35:22 +08:00
parent 27ba9c0d55
commit 97ecbaefe2
27 changed files with 677 additions and 146 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest2.java
+44
View File
@@ -0,0 +1,44 @@
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 PGInsertTest2 extends PGTest {
public void test_0() throws Exception {
String sql = "INSERT INTO films (code, title, did, date_prod, kind) VALUES"
+ " ('B6717', 'Tampopo', 110, '1985-02-10', 'Comedy'), "
+ "('HG120', 'The Dinner Game', 140, DEFAULT, 'Comedy');";
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.assertEquals(5, visitor.getFields().size());
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "kind")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "code")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "date_prod")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "title")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("films", "did")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest3.java
+40
View File
@@ -0,0 +1,40 @@
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 PGInsertTest3 extends PGTest {
public void test_0() throws Exception {
String sql = "INSERT INTO films SELECT * FROM tmp_films WHERE date_prod < '2004-05-07';";
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("tmp_films")));
Assert.assertEquals(2, visitor.getFields().size());
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("tmp_films", "date_prod")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("tmp_films", "*")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest4.java
+39
View File
@@ -0,0 +1,39 @@
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 PGInsertTest4 extends PGTest {
public void test_0() throws Exception {
String sql = "INSERT INTO distributors (did, dname) VALUES (DEFAULT, 'XYZ Widgets') RETURNING did;";
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("distributors")));
Assert.assertEquals(2, visitor.getFields().size());
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("distributors", "did")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("distributors", "dname")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/postgresql/PGInsertTest5.java
+44
View File
@@ -0,0 +1,44 @@
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 PGInsertTest5 extends PGTest {
public void test_0() throws Exception {
String sql = "WITH upd AS (" + //
" UPDATE employees SET sales_count = sales_count + 1 WHERE id =" + //
" (SELECT sales_person FROM accounts WHERE name = 'Acme Corporation')" + //
" RETURNING *" + //
")" + //
"INSERT INTO employees_log SELECT *, current_timestamp FROM upd;";
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("distributors")));
Assert.assertEquals(2, visitor.getFields().size());
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("distributors", "did")));
Assert.assertTrue(visitor.getFields().contains(new TableStat.Column("distributors", "dname")));
}
}