• Home
• io.kaizen-solutions
• virgil-cats-effect_3
• 1.2.1
• source code
• CQLExecutor.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

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

package io.kaizensolutions.virgil

import cats.effect._
import com.datastax.oss.driver.api.core.CqlSession
import com.datastax.oss.driver.api.core.CqlSessionBuilder
import com.datastax.oss.driver.api.core.metrics.Metrics
import fs2.Stream
import io.kaizensolutions.virgil.configuration.PageState
import io.kaizensolutions.virgil.internal.CQLExecutorImpl
import io.kaizensolutions.virgil.internal.Proofs.=:!=

trait CQLExecutor[F[_]] {
  def execute[A](in: CQL[A]): Stream[F, A]

  def executeMutation(in: CQL[MutationResult]): F[MutationResult]

  def executePage[A](in: CQL[A], pageState: Option[PageState])(implicit
    ev: A =:!= MutationResult
  ): F[Paged[A]]

  def metrics: F[Option[Metrics]]
}
object CQLExecutor {

  /**
   * Create a CQL Executor from an existing Datastax Java Driver's CqlSession
   * Note that the user is responsible for the lifecycle of the underlying
   * CqlSession
   * @param session
   *   is the underlying Datastax Java Driver's CqlSession
   * @return
   *   the CQLExecutor
   */
  def fromCqlSession[F[_]: Async](session: CqlSession): CQLExecutor[F] =
    new CQLExecutorImpl[F](session)

  def apply[F[_]](builder: => CqlSessionBuilder)(implicit F: Async[F]): Resource[F, CQLExecutor[F]] = {
    val acquire: F[CqlSession]         = F.delay(builder.build())
    val release: CqlSession => F[Unit] = (session: CqlSession) => F.delay(session.close())
    Resource
      .make[F, CqlSession](acquire)(release)
      .map(new CQLExecutorImpl[F](_))
  }
}