代码之家  ›  专栏  ›  技术社区  ›  Travis Brown

带有自定义表示的Scala中adt的泛型派生

  •  7
  • Travis Brown  · 技术社区  · 7 年前

    我在解释 a question from the circe Gitter channel 在这里。

    假设我有这样一个Scala密封的特征层次结构(或ADT):

    sealed trait Item
    case class Cake(flavor: String, height: Int) extends Item
    case class Hat(shape: String, material: String, color: String) extends Item
    

    我希望能够在ADT和JSON表示之间来回映射,如下所示:

    { "tag": "Cake", "contents": ["cherry", 100] }
    { "tag": "Hat", "contents": ["cowboy", "felt", "black"] }
    

    默认情况下,circe的泛型派生使用不同的表示:

    scala> val item1: Item = Cake("cherry", 100)
    item1: Item = Cake(cherry,100)
    
    scala> val item2: Item = Hat("cowboy", "felt", "brown")
    item2: Item = Hat(cowboy,felt,brown)
    
    scala> import io.circe.generic.auto._, io.circe.syntax._
    import io.circe.generic.auto._
    import io.circe.syntax._
    
    scala> item1.asJson.noSpaces
    res0: String = {"Cake":{"flavor":"cherry","height":100}}
    
    scala> item2.asJson.noSpaces
    res1: String = {"Hat":{"shape":"cowboy","material":"felt","color":"brown"}}
    

    我们可以更接近circe generic extras:

    import io.circe.generic.extras.Configuration
    import io.circe.generic.extras.auto._
    
    implicit val configuration: Configuration =
       Configuration.default.withDiscriminator("tag")
    

    然后:

    scala> item1.asJson.noSpaces
    res2: String = {"flavor":"cherry","height":100,"tag":"Cake"}
    
    scala> item2.asJson.noSpaces
    res3: String = {"shape":"cowboy","material":"felt","color":"brown","tag":"Hat"}
    

    但这仍然不是我们想要的。

    1 回复  |  直到 7 年前
        1
  •  12
  •   Travis Brown    7 年前

    将case类表示为JSON数组

    首先要注意的是circe shapes模块提供了Shapeless的实例 HList

    scala> import io.circe.shapes._
    import io.circe.shapes._
    
    scala> import shapeless._
    import shapeless._
    
    scala> ("foo" :: 1 :: List(true, false) :: HNil).asJson.noSpaces
    res4: String = ["foo",1,[true,false]]
    

    Shapeless本身提供了case类和 H列表

    import io.circe.{ Decoder, Encoder }
    import io.circe.shapes.HListInstances
    import shapeless.{ Generic, HList }
    
    trait FlatCaseClassCodecs extends HListInstances {
      implicit def encodeCaseClassFlat[A, Repr <: HList](implicit
        gen: Generic.Aux[A, Repr],
        encodeRepr: Encoder[Repr]
      ): Encoder[A] = encodeRepr.contramap(gen.to)
    
      implicit def decodeCaseClassFlat[A, Repr <: HList](implicit
        gen: Generic.Aux[A, Repr],
        decodeRepr: Decoder[Repr]
      ): Decoder[A] = decodeRepr.map(gen.from)
    }
    
    object FlatCaseClassCodecs extends FlatCaseClassCodecs
    

    然后:

    scala> import FlatCaseClassCodecs._
    import FlatCaseClassCodecs._
    
    scala> Cake("cherry", 100).asJson.noSpaces
    res5: String = ["cherry",100]
    
    scala> Hat("cowboy", "felt", "brown").asJson.noSpaces
    res6: String = ["cowboy","felt","brown"]
    

    注意,我正在使用 io.circe.shapes.HListInstances

    这是一个很好的第一步,但它并没有让我们得到我们想要的表现 Item

    import io.circe.{ JsonObject, ObjectEncoder }
    import shapeless.{ :+:, CNil, Coproduct, Inl, Inr, Witness }
    import shapeless.labelled.FieldType
    
    trait ReprEncoder[C <: Coproduct] extends ObjectEncoder[C]
    
    object ReprEncoder {
      def wrap[A <: Coproduct](encodeA: ObjectEncoder[A]): ReprEncoder[A] =
        new ReprEncoder[A] {
          def encodeObject(a: A): JsonObject = encodeA.encodeObject(a)
        }
    
      implicit val encodeCNil: ReprEncoder[CNil] = wrap(
        ObjectEncoder.instance[CNil](_ => sys.error("Cannot encode CNil"))
      )
    
      implicit def encodeCCons[K <: Symbol, L, R <: Coproduct](implicit
        witK: Witness.Aux[K],
        encodeL: Encoder[L],
        encodeR: ReprEncoder[R]
      ): ReprEncoder[FieldType[K, L] :+: R] = wrap[FieldType[K, L] :+: R](
        ObjectEncoder.instance {
          case Inl(l) => JsonObject("tag" := witK.value.name, "contents" := (l: L))
          case Inr(r) => encodeR.encodeObject(r)
        }
      )
    }
    

    这告诉我们如何对 Coproduct ,Shapeless将其用作Scala中密封特征层次结构的通用表示。一开始代码可能很吓人,但这是一种非常常见的模式,如果你花大量时间处理Shapeless,你会发现90%的代码基本上是样板文件,你可以在任何时候看到你像这样归纳地构建实例。

    解码这些副产品

    解码实现甚至有点糟糕,但遵循相同的模式:

    import io.circe.{ DecodingFailure, HCursor }
    import shapeless.labelled.field
    
    trait ReprDecoder[C <: Coproduct] extends Decoder[C]
    
    object ReprDecoder {
      def wrap[A <: Coproduct](decodeA: Decoder[A]): ReprDecoder[A] =
        new ReprDecoder[A] {
          def apply(c: HCursor): Decoder.Result[A] = decodeA(c)
        }
    
      implicit val decodeCNil: ReprDecoder[CNil] = wrap(
        Decoder.failed(DecodingFailure("CNil", Nil))
      )
    
      implicit def decodeCCons[K <: Symbol, L, R <: Coproduct](implicit
        witK: Witness.Aux[K],
        decodeL: Decoder[L],
        decodeR: ReprDecoder[R]
      ): ReprDecoder[FieldType[K, L] :+: R] = wrap(
        decodeL.prepare(_.downField("contents")).validate(
          _.downField("tag").focus
            .flatMap(_.as[String].right.toOption)
            .contains(witK.value.name),
          witK.value.name
        )
        .map(l => Inl[FieldType[K, L], R](field[K](l)))
        .or(decodeR.map[FieldType[K, L] :+: R](Inr(_)))
      )
    }
    

    总的来说,我们的计划会有更多的逻辑性 Decoder

    我们的ADT代表

    import shapeless.{ LabelledGeneric, Lazy }
    
    object Derivation extends FlatCaseClassCodecs {
      implicit def encodeAdt[A, Repr <: Coproduct](implicit
        gen: LabelledGeneric.Aux[A, Repr],
        encodeRepr: Lazy[ReprEncoder[Repr]]
      ): ObjectEncoder[A] = encodeRepr.value.contramapObject(gen.to)
    
      implicit def decodeAdt[A, Repr <: Coproduct](implicit
        gen: LabelledGeneric.Aux[A, Repr],
        decodeRepr: Lazy[ReprDecoder[Repr]]
      ): Decoder[A] = decodeRepr.value.map(gen.from)
    }
    

    这看起来与我们的 FlatCaseClassCodecs ,再次为用户最小化导入。

    在行动中

    scala> import Derivation._
    import Derivation._
    
    scala> item1.asJson.noSpaces
    res7: String = {"tag":"Cake","contents":["cherry",100]}
    
    scala> item2.asJson.noSpaces
    res8: String = {"tag":"Hat","contents":["cowboy","felt","brown"]}
    

    这正是我们想要的。最好的一点是,这将适用于Scala中任何密封的trait层次结构,不管它有多少case类或这些case类有多少成员(尽管一旦您进入其中的几十个case类,编译时间将开始受到影响),假设所有成员类型都有JSON表示。