我正在尝试生成Tree
,以用于接受案例类值并在给定位置返回案例类参数的值。这对于提取私有参数的值很有用。
import reflect.runtime.currentMirror
import scala.reflect.runtime.universe._
import scala.tools.reflect.ToolBox
val tb = currentMirror.mkToolBox()
case class A(private val first: Int)
case class B(first: Int, private val second: Int)
def get(tpe: Type, position: Option[Int]): Tree = {
val pos = s"${position.map(p => s"._${p + 1}").getOrElse("")}"
tb.parse(s"(a: $tpe) => $tpe.unapply(a).get$pos")
}
println(tb.eval(get(typeOf[A], None)).asInstanceOf[(A) => Int](A(1)))
println(tb.eval(get(typeOf[B], Some(1))).asInstanceOf[(B) => Int](B(1, 2)))
我还添加了以下依赖项:
scalaVersion := "2.11.8"
libraryDependencies ++= Seq(
"org.scala-lang" % "scala-reflect" % scalaVersion.value,
"org.scala-lang" % "scala-compiler" % scalaVersion.value
)
position
是 None
当案例类只有一个参数时。
我的解决方案正在起作用,但是如何摆脱tb.parse(s"...")
和> quasiquote q"..."
?
我尝试过,但失败了:
Don't know how to unquote here
[error] q"(a: $tpe) => $tpe.unapply(a).get$pos"
[error] ^
据我了解,我无法将某些在运行时构建的Quasiquote插入,并且与tb.parse
不同,在编译时间中解析了q"..."
。我对吗?
还可以像s"(a: $tpe) => $tpe.unapply(a).get$pos"
一样安全地插入吗?当使用q"..."
语法时,准QuaSiquote知道$tpe
是Type
,但是字符串插值从中产生了一个字符串。我不确定在更复杂和具体的情况下,这是否会起作用。
您不是不是。Quasiquotes不能仅仅插入正常字符串,它们必须插入其他AST部分。如果您自己建立了一些AST的片段,则可以将它们与q
一起使用。因此,您可以写下:
def getElem(tpe: Type, pos: Option[Int]): Tree = {
// Make something like TermName("_N")
val tupleAccess = pos.map("_" + _).map(TermName.apply)
val tupleExpr = {
val plain = q"${tpe.typeSymbol.companion}.unapply(a).get"
// Build into $companion.unapply(a).get(._N)
tupleAccess.foldLeft(plain)(Select.apply)
}
q"(a: $tpe) => $tupleExpr"
}
和voilà!
object A { object B { object C { case class D(private val x: Int, val y: String) } } }
val all = tb.eval(getElem(typeOf[A.B.C.D], None)).asInstanceOf[A.B.C.D => (Int, String)]
val int = tb.eval(getElem(typeOf[A.B.C.D], Some(1))).asInstanceOf[A.B.C.D => Int]
val str = tb.eval(getElem(typeOf[A.B.C.D], Some(2))).asInstanceOf[A.B.C.D => String]
val ds = {
val D = A.B.C.D
List(D(1, "one"), D(2, "two"), D(3, "three"))
}
ds.map(all) == List((1, "one"), (2, "two"), (3, "three"))
ds.map(int) == List(1, 2, 3)
ds.map(str) == List("one", "two", "three")