doobie.h2.circe.Instances.scala Maven / Gradle / Ivy

Go to download
// Copyright (c) 2013-2020 Rob Norris and Contributors
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package doobie.h2.circe

import cats.Show
import cats.syntax.all.*
import doobie.enumerated.JdbcType
import io.circe.*
import io.circe.jawn.*
import io.circe.syntax.*
import doobie.util.*

import java.nio.charset.StandardCharsets.UTF_8

object Instances {

  private implicit val byteArrayShow: Show[Array[Byte]] = Show.show(new String(_, UTF_8))

  trait JsonInstances {
    implicit val jsonPut: Put[Json] =
      Put.Basic.one[Array[Byte]](
        jdbcTarget = JdbcType.VarChar,
        put = (ps, n, a) => ps.setBytes(n, a),
        update = (rs, n, a) => rs.updateBytes(n, a),
        checkedVendorType = Some("VARCHAR")
      )
        .tcontramap { a =>
          a.noSpaces.getBytes(UTF_8)
        }

    implicit val jsonGet: Get[Json] =
      Get.Basic.one[Array[Byte]](
        jdbcSources = JdbcType.Other,
        jdbcSourceSecondary = List.empty,
        get = (rs, n) => rs.getBytes(n),
        checkedVendorType = Some("JSON")
      )
        .temap(a =>
          parse(a.show).leftMap(_.show))

    def h2EncoderPutT[A: Encoder]: Put[A] =
      Put[Json].tcontramap(_.asJson)

    def h2EncoderPut[A: Encoder]: Put[A] =
      Put[Json].contramap(_.asJson)

    def h2DecoderGetT[A: Decoder]: Get[A] =
      Get[Json].temap(json => json.as[A].leftMap(_.show))

    def h2DecoderGet[A: Decoder]: Get[A] =
      Get[Json].map(json => json.as[A].fold(throw _, identity))
  }

}