Merge pull request #2300 from albertoadami/add-support-for-array-json

Add instancies for Array[Json] on Postgres

Author: jatcwang

modules/postgres-circe/src/main/scala/doobie/postgres/circe/Instances.scala

@@ -34,6 +34,17 @@ object Instances {
       ).temap(a =>
         parse(a.getValue).leftMap(_.show))
 
+    private implicit val showString: Show[Array[String]] = Show.show(_.mkString("Array(", ", ", ")"))
+
+    implicit val arrayJsonGet: Get[Array[Json]] =
+      Get.Advanced.array[String](NonEmptyList.of("jsonb[]", "_jsonb"))
+        .temap(_.toList.traverse(jsonStr =>
+          parse(jsonStr).leftMap(_.getMessage)).map(_.toArray))
+
+    implicit val arrayJsonPut: Put[Array[Json]] =
+      Put.Advanced.array[String](NonEmptyList.of("_jsonb"), "jsonb")
+        .tcontramap(_.map(_.noSpaces))
+
     def pgEncoderPutT[A: Encoder]: Put[A] =
       Put[Json].tcontramap(_.asJson)
 

modules/postgres-circe/src/test/scala/doobie/postgres/circe/PGJsonSuite.scala

@@ -48,6 +48,11 @@ class PGJsonSuite extends munit.CatsEffectSuite {
     testInOut("jsonb", Json.obj("something" -> Json.fromString("Yellow")), xa)
   }
 
+  {
+    import doobie.postgres.circe.jsonb.implicits.*
+    testInOut("jsonb[]", List(Json.obj("a" -> Json.fromInt(1)), Json.obj("b" -> Json.fromInt(2))), xa)
+  }
+
   // Explicit Type Checks
 
   test("json should check ok for read") {
@@ -75,6 +80,19 @@ class PGJsonSuite extends munit.CatsEffectSuite {
     a.map(_.parameterTypeErrors).assertEquals(Nil)
   }
 
+  test("array[jsonb] should check ok for read") {
+    import doobie.postgres.circe.jsonb.implicits.*
+    val a = sql"""select ARRAY['{"a":1}', '{"b":2}']::jsonb[]""".query[Array[Json]].analysis.transact(xa)
+    a.map(_.columnTypeErrors).assertEquals(Nil)
+  }
+
+  test("array[jsonb] should check ok for write") {
+    import doobie.postgres.circe.jsonb.implicits.*
+    val arr = Array(Json.obj("a" -> Json.fromInt(1)), Json.obj("b" -> Json.fromInt(2)))
+    val a = sql"select ${arr} :: jsonb[]".query[Array[Json]].analysis.transact(xa)
+    a.map(_.parameterTypeErrors).assertEquals(Nil)
+  }
+
   // Encoder / Decoders
   private case class Foo(x: Json)
   private object Foo {