DB2SelectQueryBlock add toString support. for issue #2034

This commit is contained in:
wenshao
2017-10-09 20:48:21 +08:00
parent f4d3310734
commit 8f44fb09b4
2 changed files with 9 additions and 2 deletions
@@ -20,6 +20,7 @@ import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
import com.alibaba.druid.sql.dialect.db2.ast.DB2Object;
import com.alibaba.druid.sql.dialect.db2.visitor.DB2ASTVisitor;
import com.alibaba.druid.sql.dialect.db2.visitor.DB2OutputVisitor;
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
import com.alibaba.druid.util.JdbcConstants;
@@ -91,4 +92,9 @@ public class DB2SelectQueryBlock extends SQLSelectQueryBlock implements DB2Objec
throw new UnsupportedOperationException("not support offset");
}
}
public void output(StringBuffer buf) {
DB2OutputVisitor visitor = new DB2OutputVisitor(buf);
this.accept(visitor);
}
}
@@ -18,6 +18,7 @@ package com.alibaba.druid.bvt.sql.db2;
import com.alibaba.druid.sql.DB2Test;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
import com.alibaba.druid.sql.dialect.db2.parser.DB2StatementParser;
import com.alibaba.druid.sql.dialect.db2.visitor.DB2SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
@@ -33,8 +34,8 @@ public class DB2SelectTest_25_concat extends DB2Test {
DB2StatementParser parser = new DB2StatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement stmt = statementList.get(0);
System.out.println(SQLUtils.toDB2String(stmt));
SQLSelectStatement stmt = (SQLSelectStatement) statementList.get(0);
System.out.println(stmt.getSelect().getQuery());
Assert.assertEquals(1, statementList.size());