需要一个解析器来仅更改“来自”部分的表引用。某些数据库不接受表达式中的完全限定名称。
此代码使用JSqlParser 0.9.5-SNAPSHOT并打印出修改后的 SQL:
public class SimpleTableReplacer {
public static void main(String args[]) throws JSQLParserException {
String sql = "select t1.name, t2.address from Table1 t1 inner join Table2 t2 on t1.id = t2.id";
Select select = (Select)CCJSqlParserUtil.parse(sql);
//Replace Table1 with hive.DB1.Table1 and Table2 with mongo.DB2.Table2
StringBuilder buffer = new StringBuilder();
ExpressionDeParser expressionDeParser = new ExpressionDeParser();
SelectDeParser deparser = new SelectDeParser(expressionDeParser,buffer ) {
@Override
public void visit(Table tableName) {
switch (tableName.getName()) {
case "Table1": getBuffer().append("hive.DB1.Table1").append(' ').append(tableName.getAlias().getName());break;
case "Table2": getBuffer().append("mongo.DB2.Table2").append(' ').append(tableName.getAlias().getName());break;
default:
super.visit(tableName);
}
}
};
expressionDeParser.setSelectVisitor(deparser);
expressionDeParser.setBuffer(buffer);
select.getSelectBody().accept(deparser);
//End of value modification
System.out.println(buffer.toString());
}
}
结果:SELECT t1.name, t2.address FROM hive.DB1.Table1 t1 INNER JOIN mongo.DB2.Table2 t2 ON t1.id = t2.id
.
当然,您可以使用此代码来修改类层次结构本身,这意味着更改表 - 对象名称。
此外,您可以使用 JSqlParser 的新功能为 SQL 的某些部分提供 AST 节点。您可以提取 SQL 中表名的确切位置并在那里进行文本替换。这可以这样编码:
SimpleNode node = (SimpleNode) CCJSqlParserUtil.parseAST(sql);
node.childrenAccept(new CCJSqlParserDefaultVisitor() {
@Override
public Object visit(SimpleNode node, Object data) {
if (node.getId() == CCJSqlParserTreeConstants.JJTTABLE) {
System.out.println("table name '" + node.jjtGetValue() + "' found at " + node.jjtGetFirstToken().beginColumn + "-" + node.jjtGetLastToken().endColumn);
}
return super.visit(node, data);
}
}, null);