Scala 3 (Dotty) -Muster stimmen eine Funktion mit einem Makro-Zitat ab
Dec 08 2020
Ich versuche, den Funktionsnamen über Makros in Scala 3.0.0-M2 abzurufen. Die Lösung, die ich mir ausgedacht habe, verwendet TreeAccumulator
import scala.quoted._
inline def getName[T](inline f: T => Any): String = ${getNameImpl('f)}
def getNameImpl[T](f: Expr[T => Any])(using Quotes): Expr[String] = {
import quotes.reflect._
val acc = new TreeAccumulator[String] {
def foldTree(names: String, tree: Tree)(owner: Symbol): String = tree match {
case Select(_, name) => name
case _ => foldOverTree(names, tree)(owner)
}
}
val fieldName = acc.foldTree(null, Term.of(f))(Symbol.spliceOwner)
Expr(fieldName)
}
Beim Aufrufen erzeugt dieser Code den Namen der Funktion:
case class B(field1: String)
println(getName[B](_.field1)) // "field1"
Ich frage mich, ob dies mit Anführungszeichen einfacher gemacht werden kann.
Antworten
2 DmytroMitin Dec 13 2020 at 14:44
Ich denke, es ist genug zu definieren
def getNameImpl[T: Type](f: Expr[T => Any])(using Quotes): Expr[String] = {
import quotes.reflect._
Expr(TypeTree.of[T].symbol.caseFields.head.name)
}
Eigentlich benutze ich nicht f
.
Testing:
println(getName[B](_.field1)) // "field1"
Tested in 3.0.0-M3-bin-20201211-dbc1186-NIGHTLY.
How to access parameter list of case class in a dotty macro
Alternatively you can try
def getNameImpl[T](f: Expr[T => Any])(using Quotes): Expr[String] = {
import quotes.reflect._
val fieldName = f.asTerm match {
case Inlined(
_,
List(),
Block(
List(DefDef(
_,
List(),
List(List(ValDef(_, _, _))),
_,
Some(Select(Ident(_), fn))
)),
Closure(_, _)
)
) => fn
}
Expr(fieldName)
}