2015-03-30 83 views
1

我用下面的斯卡拉宏(由this SO question代碼嚴重的啓發)獲得包含在一個給定的包,繼承一個特定性狀的所有對象的列表:實例化類符號宏

object Macros { 
    def allObjects[T <: AnyRef](packageName: String): List[Any] = macro allObjectsImpl[T] 

    def allObjectsImpl[T <: AnyRef: c.WeakTypeTag](c: Context)(packageName: c.Expr[String]) = { 
    import c.universe._ 

    val baseTraitSymbol = c.weakTypeOf[T].typeSymbol 
    val pkg = packageName.tree match { 
     case Literal(Constant(name: String)) => c.mirror.staticPackage(name) 
    } 

    val types = pkg.typeSignature.members.collect { 
     case moduleSymbol: ModuleSymbol if moduleSymbol.moduleClass.asClass.baseClasses contains baseTraitSymbol => Ident(moduleSymbol) 
    }.toList 

    val listApply = Select(reify(List).tree, TermName("apply")) 
    c.Expr[List[T]](Apply(listApply, types)) 
    } 
} 

哪些工作正常。

我想改變宏,以便而不是讓所有的對象在包中,它得到所有的具體,並且提供一個包含每個人的實例的列表。

創建對象的實例時,AST是這樣的:

scala> import scala.reflect.runtime.{universe => u} 
import scala.reflect.runtime.{universe=>u} 

scala> u showRaw (u reify {new Object}) 
res42: String = Expr(Apply(Select(New(Ident(java.lang.Object)), termNames.CONSTRUCTOR), List())) 

所以我想改變我的代碼,這會工作:

object Macros { 
    def allInstances[T <: AnyRef](packageName: String): List[Any] = macro allInstancesImpl[T] 

    def allInstancesImpl[T <: AnyRef: c.WeakTypeTag](c: Context)(packageName: c.Expr[String]) = { 
    import c.universe._ 

    val baseTraitSymbol = c.weakTypeOf[T].typeSymbol 
    val pkg = packageName.tree match { 
     case Literal(Constant(name: String)) => c.mirror.staticPackage(name) 
    } 

    def isConcreteChildClass(child: ClassSymbol, base: Symbol) = { 
     !child.isAbstract && (child.baseClasses contains base) 
    } 

    val types = pkg.typeSignature.members.collect { 
     case classSymbol: ClassSymbol if isConcreteChildClass(classSymbol, baseTraitSymbol) => { 
     Apply(Select(New(Ident(classSymbol.primaryConstructor)), termNames.CONSTRUCTOR), List()) 
     } 
    }.toList 

    val listApply = Select(reify(List).tree, TermName("apply")) 
    c.Expr[List[T]](Apply(listApply, types)) 
    } 
} 

然而,當我嘗試使用更新測試包上的宏代碼,我得到以下錯誤:

scala> Macros.allInstances[AnyRef]("test") 
<console>:9: error: class type required but()test.TestClass found 
       Macros.allInstances[AnyRef]("test") 

從我是什麼eing,它看起來像宏實際上是返回構造函數本身,而不是返回應該由構造函數構建的實例,但我無法弄清楚我錯過了什麼。

回答

2

的問題是在這條線(重新格式化爲清晰起見):

Apply(
    Select(New(Ident(classSymbol.primaryConstructor)), termNames.CONSTRUCTOR), 
    List() 
) 

你基本上選擇兩次構造。你可以只下降primaryConstructor

Apply(
    Select(New(Ident(classSymbol)), termNames.CONSTRUCTOR), 
    List() 
) 

使用ApplyConstructor也將工作:

ApplyConstructor(Ident(classSymbol), Nil) 

或者你可以只用quasiquotes去:

q"new ${Ident(classSymbol)}()" 

的quasiquote解決方案是最面向未來。