Scala Parser Combinator 替换模式

Question

我有一个下面的程序，我可以将convert(a.ACCOUNT_ID, string) 之类的模式解析为表达式，但我想用CAST(a.ACCOUNT_ID AS VARCHAR) 替换这个模式。我可以解析结果表达式并用上面的字符串替换字符串，但是有这样的表达式，因此我不想这样做。有什么方法可以进行模式替换吗？就像如果我找到convert(a.ACCOUNT_ID, string) 的模式然后将其替换为CAST(a.ACCOUNT_ID AS VARCHAR)

import scala.util.parsing.combinator._
import scala.util.parsing.combinator.lexical._
import scala.util.parsing.combinator.syntactical._
import scala.util.parsing.combinator.token._
import scala.util.parsing.input.CharSequenceReader


trait QParser  extends RegexParsers with JavaTokenParsers  {
  def knownFunction: Parser[Any] = ident ~ "(" ~ ident ~ ("." ~ ident <~ "," ~ ident ~ ")")

  def parse(inputString: String): Any = synchronized {
    phrase(knownFunction)(new CharSequenceReader(inputString)) match {
      case Success(result, _) => result
      case Failure(msg,_) => throw new DataTypeException(msg)
      case Error(msg,_) => throw new DataTypeException(msg)
    }
  }

 class DataTypeException(message: String) extends Exception(message)

}

object Parser extends QParser {
  def main(args: Array[String]) {
     println(parse("convert(a.ACCOUNT_ID, string)"));
  }
}

输出：(((convert~()~a)~(.~ACCOUNT_ID))

Answer 1

我不太确定您对 “有这样的表达式，因此我不想那样做” 的含义，但是您可以使用 @ 转换解析器函数的结果987654322@运营商。

解析器的转换函数可以是：

def knownFunction: Parser[String] = 
  ident ~ "(" ~ ident ~ "." ~ ident ~ "," ~ ident ~ ")" ^^ {
    case func ~ "(" ~ obj ~ "." ~ value ~ "," ~ castType ~ ")" =>
      val sqlFunc = Map("convert" -> "CAST")
      val sqlType = Map("string" -> "VARCHAR")
      s"${sqlFunc(func)}($obj.$value AS ${sqlType(castType)})"
  }

使用这个更新后的函数，您的应用程序的输出将是：

CAST(a.ACCOUNT_ID AS VARCHAR)

有关 Scala 组合器解析的更多信息，请参阅 Scala 编程，第 1 章的chapter。