Класс Decode Case с вложенным копродукцией дискриминатора - PullRequest
Новая
       47

Класс Decode Case с вложенным копродукцией дискриминатора

3 голосов
/ 11 апреля 2019

У меня есть следующие настройки 

case class A(eventType : String, fieldOne : Int)
case class B(eventType : String, fieldOne : Int, fieldTwo : Int)

type Event = A :+: B :+: CNil

case class X(id :String, events : List[Event])

И я получаю следующее сообщение Json, X с одним событием (экземпляр B) 

{
"id" : "id",
"events" : [
    {
      "eventType" : "B",
      "fieldOne": 1,
      "fieldTwo" : 2
    }
]
}

Если я использую Circe, я могу декодировать это в экземпляр X, однако в списке событий, потому что A прибывает первым в побочном продукте, он декодирует его в A. 

val actual = X("id", [A("B", 1)])
val expected = X("id", [B("B", 1, 2)])

Я хочу иметь возможность использовать eventType в качестве дискриминатора конфигурации, чтобы определить, каким типом в Coproduct становится вложенное поле.

Я думаю, что ответ лежит здесь

Общий вывод для ADT в Scala с пользовательским представлением

но я не могу понять это для моего случая.

1 Ответ

1 голос
/ 12 апреля 2019

Самый простой способ сделать это - изменить производные декодеры для A и B, чтобы они не работали, когда eventType не является правильным значением.Это приведет к тому, что декодер сопутствующего продукта найдет соответствующий случай естественным образом: 

import shapeless._
import io.circe.Decoder, io.circe.syntax._
import io.circe.generic.semiauto.deriveDecoder
import io.circe.generic.auto._, io.circe.shapes._

case class A(eventType: String, fieldOne: Int)
case class B(eventType: String, fieldOne: Int, fieldTwo: Int)

type Event = A :+: B :+: CNil

case class X(id: String, events: List[Event])

implicit val decodeA: Decoder[A] = deriveDecoder[A].emap {
  case a @ A("A", _) => Right(a)
  case _ => Left("Invalid eventType")
}

implicit val decodeB: Decoder[B] = deriveDecoder[B].emap {
  case b @ B("B", _, _) => Right(b)
  case _ => Left("Invalid eventType")
}

val doc = """{
  "id" : "id",
  "events" : [
    {
      "eventType" : "B",
      "fieldOne": 1,
      "fieldTwo" : 2
    }
  ]
}"""

И затем: 

scala> io.circe.jawn.decode[X](doc)
res0: Either[io.circe.Error,X] = Right(X(id,List(Inr(Inl(B(B,1,2))))))

Обратите внимание, что вы все еще можете использовать автоматически полученные кодеры - вам просто нужны дополнительныепроверьте на стороне декодирования.(Это, конечно, при условии, что вы не будете создавать значения A или B с недопустимыми типами событий, но, поскольку вы спрашиваете об использовании этого члена в качестве дискриминатора, это нормально.)

Обновление: если вы не хотите перечислять декодеры, вы можете сделать что-то вроде этого: 

import io.circe.generic.decoding.DerivedDecoder

def checkType[A <: Product { def eventType: String }](a: A): Either[String, A] =
  if (a.productPrefix == a.eventType) Right(a) else Left("Invalid eventType")

implicit def decodeSomeX[A <: Product { def eventType: String }](implicit
  decoder: DerivedDecoder[A]
): Decoder[A] = decoder.emap(checkType)

… и оно должно работать точно так же, как в коде выше.Есть небольшие (почти наверняка незначительные) затраты времени выполнения для структурных типов, но они совершенно безопасны и кажутся мне разумным способом абстрагироваться от этих типов.

...