使用Scala宏生成方法

我能够通过两个注释获得概念验证.它显然不如简单地注释课程那么好,但我想不出另一种方法.

你需要：

import scala.annotation.{ StaticAnnotation,compileTimeOnly }
import scala.language.experimental.macros
import scala.reflect.macros.whitebox.Context

我们的想法是,您可以使用此批注对每个方法进行批注,以便父类上的宏注释能够找到要扩展的方法.

class alias(aliases: String *) extends StaticAnnotation

那么宏：

// Annotate the containing class to expand aliased methods within
@compileTimeOnly("You must enable the macro paradise plugin.")
class aliased extends StaticAnnotation {
    def macroTransform(annottees: Any*): Any = macro AliasMacroImpl.impl
}

object AliasMacroImpl {

  def impl(c: Context)(annottees: c.Expr[Any]*): c.Expr[Any] = {
    import c.universe._

    val result = annottees map (_.tree) match {
      // Match a class,and expand.
      case (classDef @ q"$mods class $tpname[..$tparams] $ctorMods(...$paramss) extends { ..$earlydefns } with ..$parents { $self => ..$stats }") :: _ =>

        val aliasedDefs = for {
          q"@alias(..$aliases) def $tname[..$tparams](...$paramss): $tpt = $expr" <- stats
          Literal(Constant(alias)) <- aliases
          ident = TermName(alias.toString)
        } yield {
          val args = paramss map { paramList =>
            paramList.map { case q"$_ val $param: $_ = $_" => q"$param" }
          }

          q"def $ident[..$tparams](...$paramss): $tpt = $tname(...$args)"
        }

        if(aliasedDefs.nonEmpty) {
          q"""
            $mods class $tpname[..$tparams] $ctorMods(...$paramss) extends { ..$earlydefns } with ..$parents { $self =>
              ..$stats
              ..$aliasedDefs
            }
          """
        } else classDef
        // Not a class.
        case _ => c.abort(c.enclosingPosition,"Invalid annotation target: not a class")
    }

    c.Expr[Any](result)
  }

}

请记住,这种实现将是脆弱的.它只检查注释器,检查第一个是ClassDef.然后,它查找类的成员,这些成员是使用@alias注释的方法,并创建多个别名树以拼接回到类中.如果没有带注释的方法,它只返回原始的类树.因此,这不会检测重复的方法名称,并删除修饰符(编译器不会让我同时匹配注释和修饰符).

这可以很容易地扩展到处理伴随对象,但我把它们留下来以保持代码更小.有关我使用的匹配器,请参阅quasiquotes syntax summary.处理伴随对象需要修改结果匹配来处理case classDef :: objDef :: Nil和case objDef :: Nil.

正在使用：

@aliased
class Socket {
    @alias("ask","read")
    def load(n: Int): Seq[Byte] = Seq(1,2,3).map(_.toByte)
}

scala> val socket = new Socket
socket: Socket = Socket@7407d2b8

scala> socket.load(5)
res0: Seq[Byte] = List(1,3)

scala> socket.ask(5)
res1: Seq[Byte] = List(1,3)

scala> socket.read(5)
res2: Seq[Byte] = List(1,3)

它还可以处理多个参数列表：

@aliased
class Foo {
    @alias("bar","baz")
    def test(a: Int,b: Int)(c: String) = a + b + c
}

scala> val foo = new Foo
foo: Foo = Foo@3857a375

scala> foo.baz(1,2)("4")
res0: String = 34