Search in sources :

Example 1 with PGExportParameterVisitor

use of com.alibaba.druid.sql.dialect.postgresql.visitor.PGExportParameterVisitor in project druid by alibaba.

the class PGExportParameterVisitorTest method test_pg.

public void test_pg() throws Exception {
    String sql = "select fname, count(*) from t where fid = 1 group by fname order by 1";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    SQLStatement stmt = parser.parseStatementList().get(0);
    List<Object> parameters = new ArrayList<Object>();
    PGExportParameterVisitor visitor = new PGExportParameterVisitor(parameters);
    stmt.accept(visitor);
    Assert.assertEquals(1, parameters.size());
    Assert.assertEquals(1, parameters.get(0));
}
Also used : PGExportParameterVisitor(com.alibaba.druid.sql.dialect.postgresql.visitor.PGExportParameterVisitor) ArrayList(java.util.ArrayList) SQLStatement(com.alibaba.druid.sql.ast.SQLStatement) PGSQLStatementParser(com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser)

Example 2 with PGExportParameterVisitor

use of com.alibaba.druid.sql.dialect.postgresql.visitor.PGExportParameterVisitor in project druid by alibaba.

the class PGExportParameterVisitorTest1 method test_pg.

public void test_pg() throws Exception {
    String sql = "select fname, count(*) from t where fid = 1 group by fname order by 1";
    PGSQLStatementParser parser = new PGSQLStatementParser(sql);
    SQLStatement stmt = parser.parseStatementList().get(0);
    PGExportParameterVisitor visitor = new PGExportParameterVisitor();
    stmt.accept(visitor);
    List<Object> parameters = visitor.getParameters();
    Assert.assertEquals(1, parameters.size());
    Assert.assertEquals(1, parameters.get(0));
}
Also used : PGExportParameterVisitor(com.alibaba.druid.sql.dialect.postgresql.visitor.PGExportParameterVisitor) SQLStatement(com.alibaba.druid.sql.ast.SQLStatement) PGSQLStatementParser(com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser)

Aggregations

SQLStatement (com.alibaba.druid.sql.ast.SQLStatement)2 PGSQLStatementParser (com.alibaba.druid.sql.dialect.postgresql.parser.PGSQLStatementParser)2 PGExportParameterVisitor (com.alibaba.druid.sql.dialect.postgresql.visitor.PGExportParameterVisitor)2 ArrayList (java.util.ArrayList)1