在Scala宏中处理按名称参数

时间:2015-04-20 20:18:22

标签: scala scala-macros callbyname

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

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,以防它{&#39}。一个=> T,因为=> T会在生成的代码中导致进一步的问题。

1 个答案:

答案 0 :(得分:4)

通常,使用提取器。有时候我必须看看运行时类是什么,要记住要使用的提取器。我不是每天都使用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>