0

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 来获取昨天的日期。

4

0 回答 0