doobie.enum.autogeneratedkeys.scala Maven / Gradle / Ivy

Go to download
package doobie.enum

import doobie.util.invariant._

import java.sql.Statement._

import cats.kernel.Eq
import cats.kernel.instances.int._

object autogeneratedkeys {

  /** @group Implementation */
  sealed abstract class AutoGeneratedKeys(val toInt: Int)

  /** @group Values */ case object ReturnGeneratedKeys extends AutoGeneratedKeys(RETURN_GENERATED_KEYS)
  /** @group Values */ case object NoGeneratedKeys extends AutoGeneratedKeys(NO_GENERATED_KEYS)

  /** @group Implementation */
  object AutoGeneratedKeys {

    def fromInt(n: Int): Option[AutoGeneratedKeys] =
      Some(n) collect {
        case ReturnGeneratedKeys.toInt => ReturnGeneratedKeys
        case NoGeneratedKeys.toInt => NoGeneratedKeys
      }

    def unsafeFromInt(n: Int): AutoGeneratedKeys =
      fromInt(n).getOrElse(throw InvalidOrdinal[AutoGeneratedKeys](n))

    implicit val EqAutoGeneratedKeys: Eq[AutoGeneratedKeys] =
      Eq.by(_.toInt)

  }

}