Изменить: я обнаружил свою ошибку - в квазицитатах для моего рекурсивного случая была ошибка, из-за которой он возвращал искаженную последовательность
Я пытаюсь создать макрос, который превратит класс дела T
в updateMap: Map[String, (play.api.libs.json.JsValue) => Try[(T) => T]]
(Как использовать макросы scala для создания функционального объекта (для создания Map [String, (T) =› T])), где Ключи карты - это имена полей класса case - идея состоит в том, что с помощью JsObject("key" -> JsValue)
мы можем получить соответствующий метод обновления из updateMap
с помощью key
, а затем применить обновление с помощью JsValue
. У меня это работает в нерекурсивном случае, то есть с учетом класса case, у которого нет других классов case в качестве полей. Однако я хотел бы расширить этот макрос, чтобы он мог генерировать updateMap
для классов case, которые содержат другие классы case, например
case class Inner(innerStr: String)
case class Outer(outerStr: String, inner: Inner)
updateMap[Outer] = {
// updateMap[Inner]
val innerMap = Map("innerStr" -> (str: String) =>
Try { (i: Inner) => i.copy(innerStr = str) } )
// updateMap[Outer]
Map("outerStr" -> (str: String) =>
Try { (o: Outer) => o.copy(outerStr = str) },
"inner.innerStr" -> (str: String) =>
Try { (o: Outer) => innerMap.get("innerStr").get(str).flatMap(lens => o.copy(inner = lens(o.inner))) })}
Другими словами, учитывая updateMap[Outer]
, я мог бы напрямую обновить поле outerStr
объекта, иначе я мог бы обновить поле inner.innerStr
объекта, в любом случае получив обратно Try[Outer]
.
Код работает правильно для нерекурсивного случая (copyMapRec[Inner]()
), но рекурсивный случай (copyMapRec[Outer]()
) выдает мне ошибку «ошибка: отсутствует тип параметра» - я предполагаю, что мне нужно либо указать где-то неявный параметр, либо еще что у меня принципиальное непонимание насчет сращивания.
В приведенном ниже коде используется (String) => Try[(T) => T]
вместо (JsValue) => Try[(T) => T]
, поэтому мне не нужно импортировать платформу воспроизведения в мой REPL. Я использую неявные преобразования для преобразования JsValue
(или String
) в соответствующий тип (это происходит в строке val x: $fieldType = str
в квази-кавычках базового случая; если соответствующее неявное преобразование недоступно, я получаю ошибку компилятора).
import scala.language.experimental.macros
def copyMapImplRec[T: c.WeakTypeTag](c: scala.reflect.macros.Context)(blacklist: c.Expr[String]*): c.Expr[Map[String, (String) => scala.util.Try[(T) => T]]] = {
import c.universe._
// Fields that should be omitted from the map
val blacklistList: Seq[String] = blacklist.map(e => c.eval(c.Expr[String](c.resetAllAttrs(e.tree))))
def rec(tpe: Type): c.Expr[Map[String, (String) => scala.util.Try[(T) => T]]] = {
val typeName = tpe.typeSymbol.name.decoded
// All fields in the case class's primary constructor, minus the blacklisted fields
val fields = tpe.declarations.collectFirst {
case m: MethodSymbol if m.isPrimaryConstructor => m
}.get.paramss.head.filterNot(field => blacklistList.contains(typeName + "." + field.name.decoded))
// Split the fields into case classes and non case classes
val recursive = fields.filter(f => f.typeSignature.typeSymbol.isClass && f.typeSignature.typeSymbol.asClass.isCaseClass)
val nonRecursive = fields.filterNot(f => f.typeSignature.typeSymbol.isClass && f.typeSignature.typeSymbol.asClass.isCaseClass)
val recursiveMethods = recursive.map {
field => {
val fieldName = field.name
val fieldNameDecoded = fieldName.decoded
// Get the c.Expr[Map] for this case class
val map = rec(field.typeSignature)
// Construct an "inner.innerStr -> " seq of tuples from the "innerStr -> " seq of tuples
q"""{
val innerMap = $map
innerMap.toSeq.map(tuple => ($fieldNameDecoded + "." + tuple._1) -> {
(str: String) => {
val innerUpdate = tuple._2(str)
innerUpdate.map(innerUpdate => (outer: $tpe) => outer.copy($fieldName = innerUpdate(outer.$fieldName)))
}
})}"""
}
}
val nonRecursiveMethods = nonRecursive.map {
field => {
val fieldName = field.name
val fieldNameDecoded = fieldName.decoded
val fieldType = field.typeSignature
val fieldTypeName = fieldType.toString
q"""{
$fieldNameDecoded -> {
(str: String) => scala.util.Try {
val x: $fieldType = str
(t: $tpe) => t.copy($fieldName = x)
}.recoverWith {
case e: Exception => scala.util.Failure(new IllegalArgumentException("Failed to parse " + str + " as " + $typeName + "." + $fieldNameDecoded + ": " + $fieldTypeName))
}
}}"""
}
}
// Splice in all of the sequences of tuples, flatten the sequence, and construct a map
c.Expr[Map[String, (String) => scala.util.Try[(T) => T]]] {
q"""{ Map((List(..$recursiveMethods).flatten ++ List(..$nonRecursiveMethods)):_*) }"""
}
}
rec(weakTypeOf[T])
}
def copyMapRec[T](blacklist: String*) = macro copyMapImplRec[T]