Шаблон Scala 3 (Dotty) соответствует функции с цитатой макроса
Dec 08 2020
Я пытаюсь получить имя функции с помощью макросов в Scala 3.0.0-M2. Решение, которое я придумал, использует 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)
}
При вызове этого кода выдает имя функции:
case class B(field1: String)
println(getName[B](_.field1)) // "field1"
Интересно, можно ли это сделать проще, используя кавычки.
Ответы
2 DmytroMitin Dec 13 2020 at 14:44
Думаю, достаточно определить
def getNameImpl[T: Type](f: Expr[T => Any])(using Quotes): Expr[String] = {
import quotes.reflect._
Expr(TypeTree.of[T].symbol.caseFields.head.name)
}
Собственно, не использую f
.
Тестирование:
println(getName[B](_.field1)) // "field1"
Проверено в 3.0.0-M3-bin-20201211-dbc1186-NIGHTLY.
Как получить доступ к списку параметров класса case в макросе dotty
В качестве альтернативы вы можете попробовать
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)
}