在 Scala 宏中处理按名称的参数

Question

我有一个宏可以对嵌套函数应用程序进行一些分析。它匹配应用程序并以这种方式检索参数类型：

case q"$f[..$targs](..$args)(...$otherArgs)" =>

    // retrieve the list of all parameter types
    val paramTpes = f.tpe match {
      case pmt: PolyType if pmt.paramLists.size == 0 =>
        Seq()
      case pmt: PolyType =>
        pmt.paramLists(0) map {_.typeSignature
             .substituteTypes(pmt.typeParams, targs map (_.tpe))}
      case pmt: MethodType if pmt.paramLists.size == 0 =>
        Seq()
      case pmt: MethodType =>
        pmt.paramLists(0) map (_.typeSignature)
    }

现在，如果碰巧有别名参数，我得到的是一些奇怪的类型，它打印 => T 但无法与任何东西匹配。反射 API 中似乎没有适当的工具来正确处理这些问题。我想做的是检索 T 以防它是 => T，因为 => T 会导致生成的代码出现更多问题。

Answer 1

通常，使用提取器。有时我必须查看运行时 class 是什么，以记住要使用的提取器。我不是每天都用API。

scala> import reflect.runtime._, universe._
import reflect.runtime._
import universe._

scala> class X { def x(i: => Int) = i * 2 }
defined class X

scala> typeOf[X].member(TermName("x"))
res0: reflect.runtime.universe.Symbol = method x

scala> .typeSignature
res1: reflect.runtime.universe.Type = (i: => scala.Int)scala.Int

scala> res1 match { case MethodType(ps, res) => ps }
res2: List[reflect.runtime.universe.Symbol] = List(value i)

scala> .head
res3: reflect.runtime.universe.Symbol = value i

scala> .typeSignature
res4: reflect.runtime.universe.Type = => scala.Int

scala> res4.getClass
res5: Class[_ <: reflect.runtime.universe.Type] = class scala.reflect.internal.Types$ClassArgsTypeRef

scala> res4 match { case TypeRef(pre, sym, args) => sym }
res6: reflect.runtime.universe.Symbol = class <byname>

scala> res4 match { case TypeRef(pre, sym, args) => args }
res7: List[reflect.runtime.universe.Type] = List(scala.Int)

scala> definitions
res8: reflect.runtime.universe.DefinitionsApi = scala.reflect.internal.Definitions$definitions$@4e80a001

scala> definitions.By
ByNameParamClass   ByteClass   ByteTpe

scala> definitions.ByNameParamClass
res9: reflect.runtime.universe.ClassSymbol = class <byname>

我依稀记得这个特殊的名字，但是我能得到一个稳定的模式匹配前缀吗？我猜不是。

scala> res4 match { case TypeRef(pre, definitions.ByNameParamClass, args) => args }
<console>:20: error: stable identifier required, but scala.reflect.runtime.`package`.universe.definitions.ByNameParamClass found.
 Note that method ByNameParamClass is not stable because its type, => reflect.runtime.universe.ClassSymbol, is volatile.
              res4 match { case TypeRef(pre, definitions.ByNameParamClass, args) => args }
                                                         ^

scala> val k = definitions.ByNameParamClass
k: reflect.runtime.universe.ClassSymbol = class <byname>

scala> res4 match { case TypeRef(pre, k, args) => args }
res11: List[reflect.runtime.universe.Type] = List(scala.Int)

scala>

在 Scala 宏中处理按名称的参数

Handling by-name parameters in Scala macro

scala

callbyname

scala-macros