Examples with PGSQLStatementParser com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser used on opensource projects

Example 46 with PGSQLStatementParser

use of com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser in project druid by alibaba.

the class PGSelectTest21 method test_1.

public void test_1() throws Exception {
    String sql = "with a(a1,b1) as (select * from b) select * from a";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement statemen = statementList.get(0);
    //        print(statementList);
    assertTrue(statemen instanceof PGSelectStatement);
    assertTrue(((PGSelectStatement) statemen).getWith().getWithQuery().size() == 1);
    StringBuffer sb = new StringBuffer();
    ((PGSelectStatement) statemen).getWith().getWithQuery().get(0).getName().output(sb);
    assertTrue("a".equals(sb.toString()));
    assertTrue(((PGSelectStatement) statemen).getWith().getWithQuery().get(0).getColumns().size() == 2);
}

Example 47 with PGSQLStatementParser

use of com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser in project druid by alibaba.

the class PGSelectTest23 method test_0.

public void test_0() throws Exception {
    String sql = "select id, login_name, name, password, salt, roles, register_date from user WHERE ( name like ? ) limit ? offset ?";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    //        print(statementList);
    Assert.assertEquals("SELECT id, login_name, name, password, salt" + "\n\t, roles, register_date" + "\nFROM user" + "\nWHERE name LIKE ?" + "\nLIMIT ? OFFSET ?", SQLUtils.toPGString(stmt));
    Assert.assertEquals("select id, login_name, name, password, salt" + "\n\t, roles, register_date" + "\nfrom user" + "\nwhere name like ?" + "\nlimit ? offset ?", SQLUtils.toPGString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
    Assert.assertEquals(1, statementList.size());
    PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    Assert.assertEquals(7, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getTables().size());
}

Example 48 with PGSQLStatementParser

use of com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser in project druid by alibaba.

the class PGSelectTest24 method test_0.

public void test_0() throws Exception {
    String sql = "select wm_concat(article_id) over() from t_nds_web_article";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    Assert.assertEquals("SELECT wm_concat(article_id) OVER ()" + "\nFROM t_nds_web_article", SQLUtils.toPGString(stmt));
    Assert.assertEquals("select wm_concat(article_id) over ()" + "\nfrom t_nds_web_article", SQLUtils.toPGString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
    Assert.assertEquals(1, statementList.size());
    PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    Assert.assertEquals(1, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getTables().size());
}

Example 49 with PGSQLStatementParser

use of com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser in project druid by alibaba.

the class PGSelectTest25 method test_0.

public void test_0() throws Exception {
    String sql = "select COALESCE((SELECT project_deduct_mandays from mytable_01 limit 1)) from t";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    Assert.assertEquals("SELECT COALESCE((" + "\n\t\tSELECT project_deduct_mandays" + "\n\t\tFROM mytable_01" + "\n\t\tLIMIT 1" + "\n\t\t))" + "\nFROM t", SQLUtils.toPGString(stmt));
    Assert.assertEquals("select COALESCE((" + "\n\t\tselect project_deduct_mandays" + "\n\t\tfrom mytable_01" + "\n\t\tlimit 1" + "\n\t\t))" + "\nfrom t", SQLUtils.toPGString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
    Assert.assertEquals(1, statementList.size());
    PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    Assert.assertEquals(1, visitor.getColumns().size());
    Assert.assertEquals(2, visitor.getTables().size());
}

Example 50 with PGSQLStatementParser

use of com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser in project druid by alibaba.

the class PGSelectTest26 method test_0.

public void test_0() throws Exception {
    String sql = "select '[{\"a\":\"foo\"},{\"b\":\"bar\"},{\"c\":\"baz\"}]'::json->2 from dual";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    Assert.assertEquals("SELECT '[{\"a\":\"foo\"},{\"b\":\"bar\"},{\"c\":\"baz\"}]'::json -> 2" + "\nFROM dual", SQLUtils.toPGString(stmt));
    Assert.assertEquals("select '[{\"a\":\"foo\"},{\"b\":\"bar\"},{\"c\":\"baz\"}]'::json -> 2" + "\nfrom dual", SQLUtils.toPGString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION));
    Assert.assertEquals(1, statementList.size());
    PGSchemaStatVisitor visitor = new PGSchemaStatVisitor();
    stmt.accept(visitor);
    //        System.out.println("Tables : " + visitor.getTables());
    //        System.out.println("fields : " + visitor.getColumns());
    //        System.out.println("coditions : " + visitor.getConditions());
    Assert.assertEquals(0, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getTables().size());
}