假设我想要一个宏,它接受一个表达式并在它是一个元组文字时返回它。像这样的东西适用于元组,但返回Some(1)
而不是None
其他一切:
import scala.reflect.macros.blackbox.Context
class ArityMacros(val c: Context) {
import c.universe._
def arity[A: c.WeakTypeTag](a: c.Expr[A]): c.Tree = a.tree match {
case q"(..$xs)" => q"_root_.scala.Some(${ xs.size })"
case _ => q"_root_.scala.None"
}
}
import scala.language.experimental.macros
def arity[A](a: A): Option[Int] = macro ArityMacros.arity[A]
我知道我可以做这样的事情:
class ArityMacros(val c: Context) {
import c.universe._
def arity[A: c.WeakTypeTag](a: c.Expr[A]): c.Tree = a.tree match {
case q"scala.Tuple1.apply[$_]($_)" => q"_root_.scala.Some(1)"
case q"(..$xs)" if xs.size > 1 => q"_root_.scala.Some(${ xs.size })"
case other => q"_root_.scala.None"
}
}
但是感觉应该有更好的方法来区分Tuple1
和非元组情况(也许我过去曾使用过它?)。