Question

注意：下面有编辑！ 注意：下面有另一个编辑！

我编写了一个Scala注释宏，它正在传递一个类并创建（或者更确切地说是填充）一个case对象。 case对象的名称与传递的类的名称相同。更重要的是，对于传递类的每个字段，case对象中都会有一个同名字段。但是，case对象的字段都是String类型，它们的值是传递的类中相应字段的类型名称。例如：

// Using the annotation macro to populate a case object called `String`
@RegisterClass(classOf[String]) case object String

// The class `String` defines a field called `value` of type `char[]`.
// The case object also has a field `value`, containing `"char[]"`.
println(String.value) // Prints `"char[]"` to the console

但是，这似乎只适用于预定义的类，例如String。如果我定义case class A(...)并尝试执行@RegisterClass(classOf[A]) case object A，则会收到以下错误：

[info]   scala.tools.reflect.ToolBoxError: reflective compilation has failed:
[info]   
[info]   not found: type A

我做错了什么？我的宏代码可以在下面找到。此外，如果有人注意到非惯用的Scala或一般的不良做法，我不会介意提示。非常感谢你提前！

class RegisterClass[T](clazz: Class[T]) extends StaticAnnotation {
  def macroTransform(annottees: Any*) =
    macro RegisterClass.expandImpl[T]
}

object RegisterClass {
  def expandImpl[T](c: blackbox.Context)(annottees: c.Expr[Any]*) = {
    import c.universe._
    val clazz: Class[T] = c.prefix.tree match {
      case q"new RegisterClass($clazz)" => c.eval[Class[T]](c.Expr(clazz))
      case _ =>  c.abort(c.enclosingPosition, "RegisterClass: Annotation expects a Class[T] instance as argument.")
    }
    annottees.map(_.tree) match {
      case List(q"case object $caseObjectName") =>
        if (caseObjectName.toString != clazz.getSimpleName)
          c.abort(c.enclosingPosition, "RegisterClass: Annotated case object and class T of passed Class[T] instance" +
            "must have the same name.")
        val clazzFields = clazz.getDeclaredFields.map(field => field.getName -> field.getType.getSimpleName).toList
        val caseObjectFields = clazzFields.map(field => {
          val fieldName: TermName = field._1
          val fieldType: String = field._2
          q"val $fieldName = $fieldType"
        })
        c.Expr[Any](q"case object $caseObjectName { ..$caseObjectFields }")
      case _ => c.abort(c.enclosingPosition, "RegisterClass: Annotation must be applied to a case object definition.")
    }
  }
}

编辑：正如Eugene Burmako所指出的那样，错误发生是因为class A尚未编译，因此它不存在java.lang.Class。我现在已经开始为每个人提供100个StackOverflow积分，作为一个想法如何让它工作！

编辑2：用例的一些背景：作为我的学士论文的一部分，我正在使用Scala DSL来表达对事件处理系统的查询。这些查询传统上表示为字符串，这会导致很多问题。典型的查询看起来如下：＆＃34;从模式中选择A.id，B.timestamp [A - > B]＆＃34 ;.含义：如果发生A类型的事件，之后发生类型为B的事件，请给我id事件A和{{1} } timestamp事件。类型B和A通常是简单的Java类，我无法控制它们。 B和id是这些类的字段。我希望我的DSL查询看起来像这样：timestamp。这意味着对于表示事件类型的每个类，例如select (A.id, B.timestamp) { /* ... * / }，我需要一个伴随对象 - 理想情况下具有相同的名称。此伴随对象应具有与相应类相同的字段，以便我可以将其字段传递给A函数，如下所示：select。这样，如果我试图将select (A.id, B.timestamp) { /* ... * / }传递给A.idd函数，如果原始类中没有这样的字段，它将在编译时失败 - 因为那样就没有了伴侣对象。

Answer 1

这不是您的宏观问题的答案，但它可以解决您的一般问题。
如果您可以允许对DSL的语法进行微小更改，则可以在不使用宏的情况下进行（取决于此问题中未提及的其他要求）。

scala> class Select[A,B]{
     |   def apply[R,S](fa: A => R, fb: B => S)(body: => Unit) = ???
     | }
defined class Select

scala> def select[A,B] = new Select[A,B]
select: [A, B]=> Select[A,B]

scala> class MyA { def id = 42L }
defined class MyA

scala> class MyB { def timestamp = "foo" }
defined class MyB

scala> select[A,B](_.id, _.timestamp){ /* ... */ }
scala.NotImplementedError: an implementation is missing

我在这里使用类Select作为一种方法，能够指定事件类的类型，同时让编译器推断函数fa和fb的结果类型。如果您不需要这些结果类型，则可以将其写为def select[A,B](fa: A => Any, fb: B => Any)(body: => Unit) = ???。

如有必要，您仍然可以将select或apply方法实现为宏。但是使用这种语法，您将不再需要使用宏注释生成对象。

Scala注释宏仅适用于预定义的类

1 个答案: