在Scala中的运行时获取指定变量的名称



让一个类

case class C(val i: Int) {
  var assignee: Option[String] = None
}

并且对于名为x的变量的给定赋值,

val x = C(1)
x: C = C(1)

如何获取值"x": Stringx.assignee,即

assert( x.assignee == Some("x") )

本地变量名在运行时不可用。但是您可以在编译时用宏捕获它。参见例如definingValNamehttps://github.com/sbt/sbt/blob/9c442d3aed53bdc89db1ada9d5b204bf02adb339/main/settings/src/main/scala/sbt/std/KeyMacro.scala及其用途:

def definingValName(c: Context, invalidEnclosingTree: String => String): String = {
  import c.universe.{ Apply => ApplyTree, _ }
  val methodName = c.macroApplication.symbol.name
  def processName(n: Name): String = n.decoded.trim // trim is not strictly correct, but macros don't expose the API necessary
  def enclosingVal(trees: List[c.Tree]): String = {
    trees match {
      case vd @ ValDef(_, name, _, _) :: ts => processName(name)
      case (_: ApplyTree | _: Select | _: TypeApply) :: xs => enclosingVal(xs)
      // lazy val x: X = <methodName> has this form for some reason (only when the explicit type is present, though)
      case Block(_, _) :: DefDef(mods, name, _, _, _, _) :: xs if mods.hasFlag(Flag.LAZY) => processName(name)
      case _ =>
        c.error(c.enclosingPosition, invalidEnclosingTree(methodName.decoded))
        "<error>"
    }
  }
  enclosingVal(enclosingTrees(c).toList)
}
def enclosingTrees(c: Context): Seq[c.Tree] =
  c.asInstanceOf[reflect.macros.runtime.Context].callsiteTyper.context.enclosingContextChain.map(_.tree.asInstanceOf[c.Tree])

相关内容

  • 没有找到相关文章

最新更新