6

我目前正在玩一些宏，无论如何这可能是个坏主意，但这是我的问题：

我有以下宏：

def using[A <: { def close(): Unit }, B](resource: A)(f: A => B) = macro usingImpl[A, B]

def usingImpl[A <: { def close(): Unit }, B](c: Context)(resource: c.Expr[A])(f: c.Expr[A => B]): c.Expr[B] = {

  import c.universe._


  f.tree match {
    case Function(params, body) =>
      //val ValDef(modifiers, name, tpt, _) = params.head
      c.Expr[B](
        Block(
          List(
            //ValDef(modifiers, name, tpt, resource.tree)
            ValDef(params.head.symbol, resource.tree)
          ),
          body
        )
      )

    case _: Select =>
      reify {
        val res = resource.splice
        try {
          f.splice(res)
        } finally {
          res.close()
        }
      }
  }
}

在 a 的情况下Select，我只需调用该函数并关闭资源，就可以正常工作。但是在 a 的情况下Function，我想将参数值分配给资源并调用正文。当我使用已弃用的创建者时ValDef，需要 aSymbol和 a Tree，一切正常。如果我使用的是注释掉的 4-args 创建器，我会收到一个编译器错误，指出该值x$1不在范围内。当我查看两个版本生成的代码时，它看起来完全一样：

Expr[Int]({
  <synthetic> val x$1: Test.Foo = new Test.this.Foo();
  x$1.bar.+(23)
})

有没有办法简单地使用params.head和分配一个值？谢谢你的帮助！

编辑：

我这样称呼宏：

object Test extends App {
  import Macros._

  class Foo {
    def close() {}

    def bar = 3
  }

  println(using(new Foo)(_.bar + 3))
}

正如我所说，如果我使用的是注释掉的版本，它会给我一个编译器错误，它会打印 AST，最后会显示以下消息：[error] symbol value x$1 does not exist in Test$delayedInit$body.apply

我正在使用 2.10.1。

4

1 回答 1

7

啊，现在我可以重现你的错误了。任何时候你得到“这个条目似乎已经杀死了编译器”。消息并在堆栈跟踪中看到这样的行：

at scala.reflect.internal.SymbolTable.abort(SymbolTable.scala:49)

你的下一步应该是开始在c.resetAllAttrs. 老实说，我第一次无法重现您的错误的原因是因为在复制和粘贴您的代码之后，我已经body在块中替换c.resetAllAttrs(body)为，甚至没有考虑它。在这一点上，这只是一种反射。

例如，请参阅这个小的单一抽象方法类演示，以了解需要此技巧的地方的类似实例。

在你的情况下，如果我们有这个：

import scala.language.experimental.macros
import scala.reflect.macros.Context

object Macros {
  def using[A <: { def close(): Unit }, B](resource: A)(f: A => B) =
    macro usingImpl[A, B]

  def usingImpl[A <: { def close(): Unit }, B](c: Context)
    (resource: c.Expr[A])(f: c.Expr[A => B]): c.Expr[B] = {
    import c.universe._

    val expr = f.tree match {
      case Function(ValDef(modifiers, name, tpt, _) :: Nil, body) =>
        c.Expr[B](
          Block(
            ValDef(modifiers, name, tpt, resource.tree) :: Nil,
            c.resetAllAttrs(body)
          )
        )

      case _: Select => reify {
        val res = resource.splice
        try { f.splice(res) } finally { res.close() }
      }
    }

    println(expr)
    expr
  }
}

当我们编译您的测试代码时，我们将看到以下内容：

Expr[B]({
  <synthetic> val x$1: $line3.$read.$iw.$iw.Test.Foo = new Test.this.Foo();
  x$1.bar.$plus(3)
})

完全如愿。

于 2013-05-09T00:50:04.483 回答