Scala 3 (Dotty) Pattern correspond à une fonction avec une citation de macro
Dec 08 2020
J'essaie d'obtenir le nom de la fonction via des macros dans Scala 3.0.0-M2 La solution que j'ai trouvée utilise 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)
}
Lorsqu'il est appelé, ce code produit le nom de la fonction:
case class B(field1: String)
println(getName[B](_.field1)) // "field1"
Je me demande si cela peut être fait de manière plus simple en utilisant des guillemets.
Réponses
2 DmytroMitin Dec 13 2020 at 14:44
Je suppose que c'est suffisant pour définir
def getNameImpl[T: Type](f: Expr[T => Any])(using Quotes): Expr[String] = {
import quotes.reflect._
Expr(TypeTree.of[T].symbol.caseFields.head.name)
}
En fait, je n'utilise pas f
.
Essai:
println(getName[B](_.field1)) // "field1"
Testé dans 3.0.0-M3-bin-20201211-dbc1186-NIGHTLY.
Comment accéder à la liste des paramètres de la classe de cas dans une macro dotty
Sinon, vous pouvez essayer
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)
}