feat: support ListMap (#1177)

zio · Oct 20, 2024 · 7066a54 · 7066a54
1 parent e8b1b79
commit 7066a54
Show file tree

Hide file tree

Showing 6 changed files with 29 additions and 0 deletions.
diff --git a/zio-json/shared/src/main/scala/zio/json/JsonCodec.scala b/zio-json/shared/src/main/scala/zio/json/JsonCodec.scala
@@ -157,6 +157,10 @@ private[json] trait CodecLowPriority1 extends CodecLowPriority2 { this: JsonCode
 
   implicit def sortedSet[A: Ordering: JsonEncoder: JsonDecoder]: JsonCodec[immutable.SortedSet[A]] =
     JsonCodec(JsonEncoder.sortedSet[A], JsonDecoder.sortedSet[A])
+
+  implicit def listMap[K: JsonFieldEncoder: JsonFieldDecoder, V: JsonEncoder: JsonDecoder]
+    : JsonCodec[immutable.ListMap[K, V]] =
+    JsonCodec(JsonEncoder.listMap[K, V], JsonDecoder.listMap[K, V])
 }
 
 private[json] trait CodecLowPriority2 extends CodecLowPriority3 { this: JsonCodec.type =>

diff --git a/zio-json/shared/src/main/scala/zio/json/JsonDecoder.scala b/zio-json/shared/src/main/scala/zio/json/JsonDecoder.scala
@@ -597,6 +597,13 @@ private[json] trait DecoderLowPriority1 extends DecoderLowPriority2 {
       def unsafeDecode(trace: List[JsonError], in: RetractReader): collection.SortedMap[K, V] =
         keyValueBuilder(trace, in, collection.SortedMap.newBuilder[K, V])
     }
+
+  implicit def listMap[K: JsonFieldDecoder, V: JsonDecoder]: JsonDecoder[immutable.ListMap[K, V]] =
+    new JsonDecoder[immutable.ListMap[K, V]] {
+
+      def unsafeDecode(trace: List[JsonError], in: RetractReader): immutable.ListMap[K, V] =
+        keyValueBuilder(trace, in, immutable.ListMap.newBuilder[K, V])
+    }
 }
 
 // We have a hierarchy of implicits for two reasons:

diff --git a/zio-json/shared/src/main/scala/zio/json/JsonEncoder.scala b/zio-json/shared/src/main/scala/zio/json/JsonEncoder.scala
@@ -377,6 +377,9 @@ private[json] trait EncoderLowPriority1 extends EncoderLowPriority2 {
 
   implicit def sortedMap[K: JsonFieldEncoder, V: JsonEncoder]: JsonEncoder[collection.SortedMap[K, V]] =
     keyValueIterable[K, V, collection.SortedMap]
+
+  implicit def listMap[K: JsonFieldEncoder, V: JsonEncoder]: JsonEncoder[immutable.ListMap[K, V]] =
+    keyValueIterable[K, V, immutable.ListMap]
 }
 
 private[json] trait EncoderLowPriority2 extends EncoderLowPriority3 {

diff --git a/zio-json/shared/src/test/scala/zio/json/CodecSpec.scala b/zio-json/shared/src/test/scala/zio/json/CodecSpec.scala
@@ -182,6 +182,12 @@ object CodecSpec extends ZIOSpecDefault {
 
           assert(jsonStr.fromJson[Map[String, Int]])(isRight(equalTo(expected)))
         },
+        test("ListMap") {
+          val jsonStr  = """{"5XL":3,"2XL":14,"XL":159}"""
+          val expected = collection.immutable.ListMap("5XL" -> 3, "2XL" -> 14, "XL" -> 159)
+
+          assert(jsonStr.fromJson[collection.immutable.ListMap[String, Int]])(isRight(equalTo(expected)))
+        },
         test("zio.Chunk") {
           val jsonStr  = """["5XL","2XL","XL"]"""
           val expected = Chunk("5XL", "2XL", "XL")

diff --git a/zio-json/shared/src/test/scala/zio/json/DecoderSpec.scala b/zio-json/shared/src/test/scala/zio/json/DecoderSpec.scala
@@ -494,6 +494,12 @@ object DecoderSpec extends ZIOSpecDefault {
 
           assert(json.as[SortedMap[String, Int]])(isRight(equalTo(expected)))
         },
+        test("ListMap") {
+          val json     = Json.Obj("5XL" -> Json.Num(3), "2XL" -> Json.Num(14), "XL" -> Json.Num(159))
+          val expected = immutable.ListMap("5XL" -> 3, "2XL" -> 14, "XL" -> 159)
+
+          assert(json.as[immutable.ListMap[String, Int]])(isRight(equalTo(expected)))
+        },
         test("Map, custom keys") {
           val json     = Json.Obj("1" -> Json.Str("a"), "2" -> Json.Str("b"))
           val expected = Map(1 -> "a", 2 -> "b")

diff --git a/zio-json/shared/src/test/scala/zio/json/EncoderSpec.scala b/zio-json/shared/src/test/scala/zio/json/EncoderSpec.scala
@@ -256,6 +256,9 @@ object EncoderSpec extends ZIOSpecDefault {
           assert(Map("hello" -> "world").toJsonPretty)(equalTo("{\n  \"hello\" : \"world\"\n}")) &&
           assert(Map("hello" -> Some("world"), "goodbye" -> None).toJsonPretty)(
             equalTo("{\n  \"hello\" : \"world\"\n}")
+          ) &&
+          assert(immutable.ListMap("hello" -> "world", "goodbye" -> "world").toJson)(
+            equalTo("""{"hello":"world","goodbye":"world"}""")
           )
         },
         test("Map, custom keys") {