1

我有 scala 宏，可以即时为类创建构造函数。
例如，如果我们有一个 class case class PersonConfig(name: String, age: Int, isFemale: Boolean)。我有类名的树结构和传递给类的参数，如下所示

@ val className = q"PersonConfig"
className: Ident = Ident(PersonConfig)

@ val args = List(q""""Jyn Erso"""", q"26", q"true")
args: List[Literal] = List(Literal(Constant("Jyn Erso")), Literal(Constant(26)), Literal(Constant(true)))

现在要创建将创建PersonConfig(ie PersonConfig("Jyn Erso", 26, true)) 实例的 AST 结构，我必须将 className 和 args 值结合起来。这里的挑战是args可以是任意大小，因为这个宏可以用来为许多不同的类构造构造函数。

目前明显但不那么 DRY 和冗长的解决方案是对args参数进行模式匹配并创建AST如下所示的结构。

import scala.reflect.runtime.universe
def makeExpr(className: universe.Tree, args: List[universe.Tree]): universe.Tree = {
  args.reverse match {
    case node1 :: Nil => q"$className($node1)"
    case arg1 :: arg2 :: Nil => q"$className($arg1, $arg2)"
    case arg1 :: arg2 :: arg3 :: Nil => q"$className($arg1, $arg2, $arg3)"
    case arg1 :: arg2 :: arg3 :: arg4 :: Nil => q"$className($arg1, $arg2, $arg3, $arg4)"
    case arg1 :: arg2 :: arg3 :: arg4 :: arg5 :: Nil => q"$className($arg1, $arg2, $arg3, $arg4, $arg5)"
    case Nil => throw new Exception(s"argument list for class ${className.toString} cannot be empty")
    case _ => throw new Exception(s"argument list for class ${className.toString} is too long")
  }

}

但是有没有更好的方法来有效地处理这个问题，哪个更干燥？比如使用 foldLeft 或者其他等效的方法来实现什么makeExpr功能呢？

4

1 回答 1

0

我设法使用 foldLeft 完成了这项工作，如下所示。

  def makeExpr(c: blackbox.Context)(className: c.Tree, args: List[c.Tree]): c.universe.Tree = {
    import c.universe._
    args.reverse match {
      case head :: tail => tail.foldLeft(q"$className($head)")({
       case (q"$_(..$params)", node) => q"$className(..${params :+ node})"          })
      case Nil => throw new MacroException(s"argument list for class ${className.toString} cannot be empty")
    }
  }
于 2018-07-16T18:25:59.157 回答