me.archdev.foundationdb.serializers.package.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of foundation-db-scala_2.12 Show documentation
foundation-db-scala
The newest version!
package me.archdev.foundationdb

import com.apple.foundationdb.tuple.Tuple
import me.archdev.foundationdb.algebra.DeserializationError
import shapeless._

package object serializers {

  trait Tupler[A] {
    def write(tuple: Tuple, value: A): Tuple
    def read(tuple: Tuple, idx: Int): A
  }

  implicit val BooleanToTuple: Tupler[Boolean] =
    new Tupler[Boolean] {
      override def write(tuple: Tuple, value: Boolean): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Boolean      = tuple.getBoolean(idx)
    }

  implicit val CharToTuple: Tupler[Char] =
    new Tupler[Char] {
      override def write(tuple: Tuple, value: Char): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Char      = tuple.getLong(idx).toChar
    }

  implicit val StringToTuple: Tupler[String] =
    new Tupler[String] {
      override def write(tuple: Tuple, value: String): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): String      = tuple.getString(idx)
    }

  implicit val ByteToTuple: Tupler[Byte] =
    new Tupler[Byte] {
      override def write(tuple: Tuple, value: Byte): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Byte      = tuple.getLong(idx).toByte
    }

  implicit val ShortToTuple: Tupler[Short] =
    new Tupler[Short] {
      override def write(tuple: Tuple, value: Short): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Short      = tuple.getLong(idx).toShort
    }

  implicit val IntToTuple: Tupler[Int] =
    new Tupler[Int] {
      override def write(tuple: Tuple, value: Int): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Int      = tuple.getLong(idx).toInt
    }

  implicit val LongToTuple: Tupler[Long] =
    new Tupler[Long] {
      override def write(tuple: Tuple, value: Long): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Long      = tuple.getLong(idx)
    }

  implicit val FloatToTuple: Tupler[Float] =
    new Tupler[Float] {
      override def write(tuple: Tuple, value: Float): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Float      = tuple.getFloat(idx)
    }

  implicit val DoubleToTuple: Tupler[Double] =
    new Tupler[Double] {
      override def write(tuple: Tuple, value: Double): Tuple = tuple.add(value)
      override def read(tuple: Tuple, idx: Int): Double      = tuple.getDouble(idx)
    }

  implicit val hnilToTuple: Tupler[HNil] =
    new Tupler[HNil] {
      override def write(tuple: Tuple, value: HNil): Tuple = tuple
      override def read(tuple: Tuple, idx: Int): HNil      = HNil
    }

  implicit def hconsToTuple[Head, Tail <: HList](implicit headSerializer: Tupler[Head],
                                                 tailSerializer: Tupler[Tail]): Tupler[Head :: Tail] =
    new Tupler[Head :: Tail] {
      override def write(tuple: Tuple, hList: Head :: Tail): Tuple =
        tailSerializer.write(headSerializer.write(tuple, hList.head), hList.tail)
      override def read(tuple: Tuple, idx: Int): Head :: Tail =
        headSerializer.read(tuple, idx) :: tailSerializer.read(tuple, idx + 1)
    }

  implicit def lgenToTuple[T, Repr](implicit lgen: Generic.Aux[T, Repr], reprSerializer: Tupler[Repr]): Tupler[T] =
    new Tupler[T] {
      override def write(tuple: Tuple, value: T): Tuple = reprSerializer.write(tuple, lgen.to(value))
      override def read(tuple: Tuple, idx: Int): T      = lgen.from(reprSerializer.read(tuple, idx))
    }

  implicit class ToTupleOps[T](t: T)(implicit serializer: Tupler[T]) {
    def toTuple: Tuple = serializer.write(new Tuple, t)
  }

  implicit class FromTupleOps(tuple: Tuple) {
    def fromTuple[T](implicit serializer: Tupler[T]): T =
      try {
        serializer.read(tuple, 0)
      } catch {
        case ex: ClassCastException =>
          throw DeserializationError(ex.getMessage)
      }
  }

}