doobie.syntax.connectionio.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of doobie-core_2.13 Show documentation

Pure functional JDBC layer for Scala.

There is a newer version: 1.0.0-RC6

// 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.syntax

import cats.data.{EitherT, Kleisli, OptionT}
import cats.effect.kernel.MonadCancelThrow
import cats.syntax.functor.*
import doobie.ConnectionIO
import doobie.implicits.*
import doobie.util.transactor.Transactor
import doobie.hi.{connection as IHC}

class ConnectionIOOps[A](ma: ConnectionIO[A]) {
  def transact[M[_]: MonadCancelThrow](xa: Transactor[M]): M[A] = xa.trans.apply(ma)
}

class OptionTConnectionIOOps[A](ma: OptionT[ConnectionIO, A]) {
  def transact[M[_]: MonadCancelThrow](xa: Transactor[M]): OptionT[M, A] =
    OptionT(
      xa.trans.apply(ma.orElseF(IHC.rollback.as(None)).value)
    )
}

class EitherTConnectionIOOps[E, A](ma: EitherT[ConnectionIO, E, A]) {
  def transact[M[_]: MonadCancelThrow](xa: Transactor[M]): EitherT[M, E, A] =
    EitherT(
      xa.trans.apply(ma.leftSemiflatMap(IHC.rollback.as(_)).value)
    )
}

class KleisliConnectionIOOps[A, B](ma: Kleisli[ConnectionIO, A, B]) {
  def transact[M[_]: MonadCancelThrow](xa: Transactor[M]): Kleisli[M, A, B] =
    ma.mapK(xa.trans)
}

trait ToConnectionIOOps {
  implicit def toConnectionIOOps[A](ma: ConnectionIO[A]): ConnectionIOOps[A] =
    new ConnectionIOOps(ma)

  implicit def toOptionTConnectionIOOps[A](ma: OptionT[ConnectionIO, A]): OptionTConnectionIOOps[A] =
    new OptionTConnectionIOOps(ma)

  implicit def toEitherTConnectionIOOps[E, A](ma: EitherT[ConnectionIO, E, A]): EitherTConnectionIOOps[E, A] =
    new EitherTConnectionIOOps(ma)

  implicit def toKleisliConnectionIOOps[A, B](ma: Kleisli[ConnectionIO, A, B]): KleisliConnectionIOOps[A, B] =
    new KleisliConnectionIOOps[A, B](ma)
}

object connectionio extends ToConnectionIOOps