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

Example 26 with PGSQLStatementParser

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

the class PGDropTableIfExistsTest method test_0.

public void test_0() throws Exception {
    String sql = "DROP TABLE IF EXISTS t_report_1_19";
    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("t_report_1_19")));
    Assert.assertTrue(visitor.getTables().get(new TableStat.Name("t_report_1_19")).getDropCount() == 1);
    Assert.assertTrue(visitor.getColumns().size() == 0);
}

Example 27 with PGSQLStatementParser

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

the class PGRovokeTest0 method test_0.

public void test_0() throws Exception {
    String sql = "REVOKE ALL ON accounts FROM PUBLIC;";
    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("accounts")));
    Assert.assertTrue(visitor.getTables().get(new TableStat.Name("accounts")).getDropCount() == 0);
    Assert.assertTrue(visitor.getTables().get(new TableStat.Name("accounts")).getAlterCount() == 0);
    Assert.assertTrue(visitor.getColumns().size() == 0);
}

Example 28 with PGSQLStatementParser

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

the class PGSelectTest18 method test_0.

public void test_0() throws Exception {
    String sql = "WITH RECURSIVE t(n) AS (" + "    SELECT 1" + "  UNION ALL" + "    SELECT n+1 FROM t" + ")" + "SELECT n FROM t LIMIT 100;";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement stmt = statementList.get(0);
    {
        String result = SQLUtils.toPGString(stmt);
        Assert.assertEquals("WITH RECURSIVE t (n) AS (\n" + "\t\tSELECT 1\n" + "\t\tUNION ALL\n" + "\t\tSELECT n + 1\n" + "\t\tFROM t\n" + "\t)\n" + "SELECT n\n" + "FROM t\n" + "LIMIT 100;", result);
    }
    {
        String result = SQLUtils.toPGString(stmt, SQLUtils.DEFAULT_LCASE_FORMAT_OPTION);
        Assert.assertEquals("with recursive t (n) as (\n" + "\t\tselect 1\n" + "\t\tunion all\n" + "\t\tselect n + 1\n" + "\t\tfrom t\n" + "\t)\n" + "select n\n" + "from t\n" + "limit 100;", result);
    }
    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(0, visitor.getTables().size());
}

Example 29 with PGSQLStatementParser

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

the class PGSelectTest2 method test_0.

public void test_0() throws Exception {
    String sql = "SELECT * FROM t1 NATURAL INNER JOIN t2;";
    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());
    // System.out.println("coditions : " + visitor.getConditions());
    Assert.assertEquals(2, visitor.getColumns().size());
    Assert.assertEquals(2, visitor.getTables().size());
}

Example 30 with PGSQLStatementParser

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

the class PGSelectTest21 method test_0.

public void test_0() throws Exception {
    String sql = "SELECT DISTINCT(type) FROM dbmis2_databases";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    List<SQLStatement> statementList = parser.parseStatementList();
    SQLStatement statemen = statementList.get(0);
    // print(statementList);
    Assert.assertEquals("SELECT DISTINCT type\nFROM dbmis2_databases", 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.getColumns());
    // System.out.println("coditions : " + visitor.getConditions());
    Assert.assertEquals(1, visitor.getColumns().size());
    Assert.assertEquals(1, visitor.getTables().size());
}