我JSqlParser
用来做一些简单的表名修改,但是当我在语句中使用一些 udf 函数时会抛出异常。如何让它忽略功能检查?
def replaceTableName(sql: String) = {
val select = CCJSqlParserUtil.parse(sql).asInstanceOf[Select]
val buffer = new java.lang.StringBuilder()
val expressionDeParser = new ExpressionDeParser() {
override def visit(tableColumn: Column) = {
if (tableColumn.getTable.getName != null && tmpTables.contains(resolveName(tableColumn.getTable.getName))) {
tableColumn.getTable().setName(resolveName(tableColumn.getTable().getName()))
}
super.visit(tableColumn)
}
}
val deparser = new SelectDeParser(expressionDeParser, buffer) {
override def visit(tableName: Table) {
if(tmpTables.contains(resolveName(tableName.getName)))
tableName.setName(resolveName(tableName.getName()))
super.visit(tableName)
}
}
expressionDeParser.setSelectVisitor(deparser)
expressionDeParser.setBuffer(buffer)
select.getSelectBody().accept(deparser)
buffer.toString
}
此代码将抛出
原因:net.sf.jsqlparser.parser.ParseException:遇到意外令牌:
<K_TIME_KEY_EXPR>
第 1 行第 xx 列的“current_date”
如果我的输入是select * from table where date = sub_date(current_date('yyyyMMdd'),'1','yyyyMMdd');
sub_date
并且current_date
是我的 udfs 来获取昨天的日期。