斯卡拉宏：c.universe.Tree到c.universe.Name

Question

我想使用宏和靜態註釋生成case類。

有什麼辦法可以將c.universe.Tree轉換成另一種c.universe類型，特別是c.universe.Name？

Answer 1

答案取決於你有什麼樣的樹，但在任何情況下，你想從你的樹去一個字符串，然後你可以創建無論是TypeName或TermName與TypeName和TermName構造。

例如，我們這裏的樹是一個字符串：

import scala.language.experimental.macros 
import scala.reflect.macros.blackbox.Context 

def callFooImpl(c: Context)(name: c.Expr[String]): c.Expr[Unit] = { 
    import c.universe._ 

    val termName: TermName = name.tree match { 
    case Literal(Constant(s: String)) => TermName(s) 
    case _ => c.abort(c.enclosingPosition, "Not a string literal") 
    } 

    c.Expr[Unit](q"$termName.foo()") 
} 

def callFoo(name: String): Unit = macro callFooImpl 

case class Foo(i: Int) { 
    def foo(): Unit = println(s"Called foo() on $this") 
} 

然後：

scala> :paste 
// Entering paste mode (ctrl-D to finish) 

val myFoo = Foo(10) 
callFoo("myFoo") 

// Exiting paste mode, now interpreting. 

Called foo() on Foo(10) 
myFoo: Foo = Foo(10) 

這裏我們解構字符串文字樹Literal和Constant得到一個編譯期然後我們用它來創建一個TermName，我們可以在我們生成的代碼中使用它。