io.kaizensolutions.virgil.Cursor.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of virgil-core_2.13 Show documentation
virgil-core
The newest version!
package io.kaizensolutions.virgil

import com.datastax.oss.driver.api.core.cql.Row
import com.datastax.oss.driver.api.core.data.GettableByName
import com.datastax.oss.driver.api.core.data.UdtValue
import io.kaizensolutions.virgil.codecs._

import scala.util.control.NonFatal

final case class RowCursor(
  private val history: Vector[String],
  private val current: Row
) {
  def downUdtValue(name: String): Either[DecoderException, UdtValueCursor] =
    Cursor.downUdtValue(name, current, history)

  def field[A](name: String)(implicit ev: CqlPrimitiveDecoder[A]): Either[DecoderException, A] =
    Cursor.field(current, name, history)(ev)

  def viewAs[A](implicit ev: codecs.CqlRowDecoder.Object[A]): Either[DecoderException, A] =
    ev.either
      .decode(current)
      .left
      .map(Cursor.enrichError(history))
}
object RowCursor {
  def apply(row: Row): RowCursor = RowCursor(Vector.empty, row)
}

final case class UdtValueCursor(private val history: Vector[String], private val current: UdtValue) {
  def downUdtValue(name: String): Either[DecoderException, UdtValueCursor] = Cursor.downUdtValue(name, current, history)

  def field[A](name: String)(implicit ev: CqlPrimitiveDecoder[A]): Either[DecoderException, A] =
    Cursor.field(current, name, history)(ev)

  def viewAs[A](implicit ev: CqlUdtValueDecoder.Object[A]): Either[DecoderException, A] =
    ev.either
      .decode(current)
      .left
      .map(Cursor.enrichError(history))
}
object UdtValueCursor {
  def apply(udtValue: UdtValue): UdtValueCursor = UdtValueCursor(Vector.empty, udtValue)
}

private object Cursor {
  def downUdtValue(
    fieldName: String,
    current: GettableByName,
    history: Vector[String]
  ): Either[DecoderException, UdtValueCursor] =
    try Right(UdtValueCursor(history :+ fieldName, current.getUdtValue(fieldName)))
    catch {
      case NonFatal(cause) =>
        Left(
          DecoderException.StructureReadFailure(
            s"Failed to get UdtValue from Row ${Cursor.renderHistory(history)}",
            Some(DecoderException.FieldType.Name(fieldName)),
            current,
            cause
          )
        )
    }

  def field[A](current: GettableByName, fieldName: String, history: Vector[String])(implicit
    ev: CqlPrimitiveDecoder[A]
  ): Either[DecoderException, A] =
    try { Right(CqlPrimitiveDecoder.decodePrimitiveByFieldName(current, fieldName)(ev)) }
    catch {
      case NonFatal(d: DecoderException) =>
        Left(d)

      case NonFatal(cause) =>
        Left(
          DecoderException.StructureReadFailure(
            s"Failed to get field $fieldName from Row ${Cursor.renderHistory(history)}",
            Some(DecoderException.FieldType.Name(fieldName)),
            current,
            cause
          )
        )
    }

  def renderHistory(history: Vector[String]): String =
    history.mkString(start = "History(", sep = " -> ", end = ")")

  def enrichError(history: Vector[String])(decoderException: DecoderException): DecoderException = {
    val historyRendered = renderHistory(history)
    decoderException match {
      case s @ DecoderException.StructureReadFailure(message, _, _, _) =>
        s.copy(message = s"$message. $historyRendered")

      case p @ DecoderException.PrimitiveReadFailure(message, _) =>
        p.copy(message = s"$message. $historyRendered")
    }
  }
}