Question

以下宏生成案例类Person并返回此类的实例。它使用whitebox宏，因此编译器可以推断类型Person。这允许宏客户端调用p.name，即使该字段是在宏内生成的。

import scala.language.experimental.macros

class MacroDefs(val c: scala.reflect.macros.whitebox.Context) {
  import c.universe._

  def createCaseImpl(): c.Expr[Product] = {
    val code: Tree = q""" case class Person(name:String); new Person("Joe") """
    c.Expr(code)
  }
}

object Macros {
  def createCase(): Product = macro MacrosDef.createCaseImpl
}

object Test {
  def main(args: Array[String]) {
    val p = Macros.createCase()
    println("Name: " + p.name)
  }
}

代码有效，但编译器使用结构类型来访问p.name，正如您从下面的警告消息中看到的那样（我通过反编译生成的字节码来确认它）：

Warning:(5, 54) inferred existential type Person forSome { type Person <: Product with Serializable{val name: String; def copy(name: String): Person; def copy$default$1: String @scala.annotation.unchecked.uncheckedVariance} }, which cannot be expressed by wildcards,  should be enabled
by making the implicit value scala.language.existentials visible.
    val c = Macros.createCase()
                                                 ^

由于结构类型依赖于Java Reflection，我关注性能。

我的问题是，是否有更好的方法可以使编译器使用标准方法调用而不是结构类型和反射。我使用的是Scala 2.11.6。

第二个小问题：Intellij似乎不能很好地使用whitebox宏，并将对p.name的访问标记为无效，表示该字段未知，即使它将使用scalac进行编译和运行。有没有办法让Intellij知道whitebox宏？

Answer 1

根据@MartinRing和@TravisBrown的建议，我使用Macro Annotations来解决问题。这篇文章也很有用：http://www.47deg.com/blog/scala-macros-annotate-your-case-classes。我在这里发布代码以供将来参考。

在客户端代码中，我注释了一个包装器对象，然后由宏扩展它以包含一个新的case类Person和这个类的一个实例。与使用结构类型和反射的匿名类型提供程序的解决方案相反，生成的代码使用标准方法调用。

@personGenerator object Wrapper

object Main {
  def main(args: Array[String]) {
    println("Name: " + Wrapper.thePerson.name)
    val other = new Wrapper.Person("Joan")
    println("Other: " + other.name)
  }
}

以下是注释和宏的实现：

class personGenerator extends StaticAnnotation {
  def macroTransform(annottees: Any*): Any = macro MacroDefs.personGenerator_impl
}

class MacroDefs(val c: scala.reflect.macros.whitebox.Context) {
  import c.universe._

  def personGenerator_impl(annottees: c.Expr[Any]*): c.Expr[Any] = {
    annottees.map(_.tree) match {
      case List(q"object $name { ..$body }") =>
        val code = q"""
            object $name {
              ..$body
              case class Person(name:String)
              def thePerson = new Person("Joe")
            }
          """
        c.Expr[Any](code)
    }
  }
}

编译器使用在whitebox宏内生成的case类的结构类型

1 个答案: