使用宏实例化类符号

时间:2015-03-30 17:35:13

标签: scala scala-macros

我使用以下Scala宏(受到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))
  }
}   

哪种方法正常。

我想更改宏,以便不是在包中获取所有对象,而是获取所有具体的,并提供包含实例的列表他们每个人。

创建Object实例时的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")

从我看到的情况来看,宏看起来实际上是在返回构造函数本身,而不是返回应该由构造函数构建的实例,但是我无法弄清楚我的内容是什么。我失踪了。

1 个答案:

答案 0 :(得分: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解决方案是最具前瞻性的解决方案。