在 Scala 3 中查找 lambda 捕获的值（或其类）-解网

问：

我正在寻找一种方法来查找 Scala 3 中由 lambda 捕获的值（或其类）（用于序列化 - 类似于 Spark）（我不需要 Scala 2 支持）：

val a = "abc"
val f = () => a + "xyz"
serialize(f) // Should detect a / String as captured value

在运行时执行此操作有点容易（迭代），但我想在编译时执行此操作。f.getClass.getDeclaredFields

我试图在 Macro 中检查 lambda 的时间，但它被检测为没有任何有趣信息。scala.Function0

我想知道我是否可以做一些树检查，但我真的很想避免这种情况 - 我觉得我必须复制编译器内部结构才能捕获所有边缘情况。

lambda 闭包 Scala-3 dotty

import scala.quoted.*

inline def serialize(x: Any): Unit = ${serializeImpl('x)}

def serializeImpl(x: Expr[Any])(using Quotes): Expr[Unit] = {
  import quotes.reflect.*

  def owners(s: Symbol): List[Symbol] = s :: List.unfold(s)(s1 => Option.when(s1.maybeOwner != Symbol.noSymbol)((s1.maybeOwner, s1.maybeOwner)))

  val symbol = x.asTerm.underlying.symbol
  val rhs = symbol.tree match {
    case ValDef(_, _, Some(rhs)) => rhs
  }

  val traverser = new TreeTraverser {
    override def traverseTree(tree: Tree)(owner: Symbol): Unit = {
      tree match {
        case Ident(name) =>
          val symbol1 = tree.symbol
          val pos1 = symbol1.pos.get
          println(s"identifier: $name, defined inside lambda: ${owners(symbol1).contains(symbol)}, defined in current file: ${pos1.sourceFile == SourceFile.current}")

        case _ =>
      }

      super.traverseTree(tree)(owner)
    }
  }

  traverser.traverseTree(rhs)(rhs.symbol)

  '{()}
}

用法：

object App1 {
  val b = "bbb"
}

import App1.b

object App {
  val a = "abc"
  val f = () => { val x = "uvw"; a + b + x + "xyz"}
  serialize(f)
}

//scalac: identifier: a, defined inside lambda: false, defined in current file: true
//scalac: identifier: b, defined inside lambda: false, defined in current file: false
//scalac: identifier: x, defined inside lambda: true, defined in current file: true
//scalac: identifier: $anonfun, defined inside lambda: true, defined in current file: true

@Andrew 为了方便起见，我在 github.com/DmytroMitin/scala3-macro-closure-demo 发布了该项目如果您的编译错误是因为定义右侧的 AST 为空，那么很可能您丢失了。如果您仍然遇到编译错误并需要帮助，请提供复制（在此问题或新问题中）。MatchErrorvalscalacOptions += "-Yretain-trees"

上一个：为什么我返回 lambda 的函数似乎被转换为将 int 转换为 int 的函数？

下一个：为什么 lambda 捕获中的“this”与指向同一对象的指针的处理方式不同？

在 Scala 3 中查找 lambda 捕获的值（或其类）

Finding lambda captured values (or their classes) in Scala 3

评论

评论