ldbc.connector.codec.TextCodecs.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ldbc-connector_sjs1_3 Show documentation

MySQL connector written in pure Scala3

The newest version!

/**
 * Copyright (c) 2023-2024 by Takahiko Tominaga
 * This software is licensed under the MIT License (MIT).
 * For more information see LICENSE or https://opensource.org/licenses/MIT
 */

package ldbc.connector.codec

import cats.syntax.all.*

import scodec.bits.ByteVector

import ldbc.connector.data.Type

trait TextCodecs:

  def char(length:    Int): Codec[String] = Codec.simple(s => s, _.asRight, Type.char(length))
  def varchar(length: Int): Codec[String] = Codec.simple(s => s, _.asRight, Type.varchar(length))

  def binary(length: Int): Codec[Array[Byte]] = Codec.simple(
    bytes => new String(bytes),
    _.getBytes("UTF-8").asRight[String],
    Type.binary(length)
  )
  def varbinary(length: Int): Codec[String] = Codec.simple(
    s => s,
    _.asRight,
    Type.varbinary(length)
  )

  private def blob(`type`: Type): Codec[String] = Codec.simple(
    str => ByteVector.view(str.getBytes("UTF-8")).toHex,
    hex =>
      ByteVector.fromHexDescriptive(hex) match {
        case Left(_)      => hex.asRight
        case Right(bytes) => bytes.decodeUtf8Lenient.asRight
      },
    `type`
  )

  val tinyblob:   Codec[String] = blob(Type.tinyblob)
  val blob:       Codec[String] = blob(Type.blob)
  val mediumblob: Codec[String] = blob(Type.mediumblob)
  val longblob:   Codec[String] = blob(Type.longblob)

  val tinytext:   Codec[String] = Codec.simple(s => s, _.asRight, Type.tinytext)
  val text:       Codec[String] = Codec.simple(s => s, _.asRight, Type.text)
  val mediumtext: Codec[String] = Codec.simple(s => s, _.asRight, Type.mediumtext)
  val longtext:   Codec[String] = Codec.simple(s => s, _.asRight, Type.longtext)

  def `enum`(values: String*): Codec[String] = Codec.simple(s => s, _.asRight, Type.`enum`(values.toList))

  def set(values: String*): Codec[List[String]] = Codec.simple[List[String]](
    s => s.mkString(","),
    str => str.split(",").toList.asRight,
    Type.set(values.toList)
  )

  val json: Codec[String] = Codec.simple(s => s, _.asRight, Type.json)

object text extends TextCodecs