2

我正在尝试访问作为宏实现的方法的参数。

object Macros {
    def impl()(using Quotes): Expr[Unit] = {    
        import quotes.reflect._
        val params: List[List[ValDef]] = {
            def nearestEnclosingMethodParams(owner: Symbol): List[List[ValDef]] =
                owner match {
                    case defSym if defSym.isDefDef =>
                        defSym.tree.asInstanceOf[DefDef].paramss
                    case _ =>
                        nearestEnclosingMethod(owner.owner)
                }
            nearestEnclosingMethodParams(Symbol.spliceOwner)
        }
        println(params) // I would do something useful with params names and types here
        '{()}
    }
}

呼叫站点可能类似于：

object Test {
    def foo(a: String, b: Int) = Foo.impl
    @main def run(): Unit = {
        val x = foo("blah", 24)
        ()
    }
}

object Foo {
    inline def impl = ${ Macros.impl() }
}

现在，CyclicReference当宏展开时，我在defSym.tree. 我知道这defSym.tree是循环的，因为它包含当前扩展宏的代码，但我仍然需要访问方法定义的“树”版本来访问其名称和参数，而不需要方法的主体。我怎样才能在不骑自行车的情况下获得这些信息？

4

1 回答 1

1

此问题已在最近发布的 Scala 3.0.0-RC1 版本中得到修复。

defSym.tree在示例中使用def foo(a: String, b: Int) = Foo.impl现在可以正常工作，并且只使用EmptyTree宏体。您可以轻松地从方法参数或foo定义的类型层次结构中提取所需的任何信息，这是我的用例。

DefDef(
    foo,
    List(
        List(
            ValDef(
                a,
                TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class scala)),object Predef),type String)],
                EmptyTree
            ), 
            ValDef(
                b,
                TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class <root>)),object scala),class Int)],
                EmptyTree
            )
        )
    ),
    TypeTree[TypeRef(ThisType(TypeRef(NoPrefix,module class scala)),class Unit)],
    EmptyTree
)
于 2021-02-19T07:26:03.527 回答