Question

我需要查明方法实现中是否使用了某种方法。例如，我有这个光荣的应用程序：

object Test extends App {

    def getGreetings = Array.fill(2)("hello")

    println(getGreetings.mkString("\n"))

}

我想测试方法getGreetings是否使用了Array的伴随对象的函数fill。通过上述实现，测试将成功，例如：

def getGreetings = Array("hello", "hello") // nah, fill isn't used

在this video的帮助下，我了解到我可以使用这样的宏检查实现：

def printTree(title: String)(expr: Any): Unit = macro printTreeMacro

def printTreeMacro(c: Context)(title: c.Tree)(expr: c.Tree) = {
    import c.universe._

    val code : String = showCode(expr)
    val raw  : String = showRaw(expr)

    q"""
    println(
      $title.toUpperCase + "\n\n" +
      $code              + "\n\n" +
      $raw               + "\n\n"
    )
    """
}

printTree("Method") {
    val a = Array.fill(2)("hello")
}

现在printTree显示使用了fill方法：

Block(List(ValDef(Modifiers(), TermName("a"), TypeTree(),
Apply(Apply(Apply(TypeApply(Select(Select(Ident(scala), scala.Array),
TermName("fill")), List(TypeTree())), List(Literal(Constant(2)))),
List(Literal(Constant("hello")))),
List(Typed(Apply(TypeApply(Select(Ident(scala.reflect.ClassTag),
TermName("apply")), List(TypeTree())),
List(Literal(Constant(String)))), TypeTree()))))), Literal(Constant(())))

这里缺少的部分是如何对任何方法的代码执行相同的操作，以便我可以获取方法getGreetings内部的树。

提前致谢：）

如何使用Scala宏检查实现

0 个答案: