doobie.syntax.stream.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 doobie.util.compat.=:=
import doobie.util.transactor.Transactor
import doobie.free.connection.ConnectionIO
import cats.data.Kleisli
import cats.effect.Concurrent
import cats.effect.kernel.{Async, MonadCancelThrow}
import fs2.{Pipe, Stream}

class StreamOps[F[_], A](fa: Stream[F, A]) {
  def transactNoPrefetch[M[_]: MonadCancelThrow](xa: Transactor[M])(implicit
      ev: Stream[F, A] =:= Stream[ConnectionIO, A]
  ): Stream[M, A] = xa.transP.apply(fa)

  def transact[M[_]: Concurrent](xa: Transactor[M])(implicit
      ev: Stream[F, A] =:= Stream[ConnectionIO, A]
  ): Stream[M, A] = transactNoPrefetch(xa).prefetchN(1)

}
class KleisliStreamOps[A, B](fa: Stream[Kleisli[ConnectionIO, A, *], B]) {
  def transactNoPrefetch[M[_]: MonadCancelThrow](xa: Transactor[M]): Stream[Kleisli[M, A, *], B] =
    xa.transPK[A].apply(fa)
  def transact[M[_]: Concurrent](xa: Transactor[M]): Stream[Kleisli[M, A, *], B] = transactNoPrefetch(xa).prefetchN(1)
}
class PipeOps[F[_], A, B](inner: Pipe[F, A, B]) {
  def transact[M[_]: Async](xa: Transactor[M])(implicit ev: Pipe[F, A, B] =:= Pipe[ConnectionIO, A, B]): Pipe[M, A, B] =
    xa.liftP(inner)
}

trait ToStreamOps {
  implicit def toDoobieStreamOps[F[_], A](fa: Stream[F, A]): StreamOps[F, A] =
    new StreamOps(fa)
  implicit def toDoobieKleisliStreamOps[A, B](fa: Stream[Kleisli[ConnectionIO, A, *], B]): KleisliStreamOps[A, B] =
    new KleisliStreamOps(fa)
  implicit def toDoobiePipeOps[F[_], A, B](inner: Pipe[F, A, B]): PipeOps[F, A, B] =
    new PipeOps(inner)
}

object stream extends ToStreamOps