io.getquill.CassandraZioSession.scala Maven / Gradle / Ivy
package io.getquill
import com.datastax.oss.driver.api.core.CqlSession
import com.typesafe.config.Config
import io.getquill.context.{AsyncFutureCache, CassandraSession, SyncCache}
import io.getquill.util.LoadConfig
import zio.{ZIO, ZLayer}
case class CassandraZioSession(
override val session: CqlSession,
override val preparedStatementCacheSize: Long
) extends CassandraSession
with SyncCache
with AsyncFutureCache
with AutoCloseable
object CassandraZioSession {
val live: ZLayer[CassandraContextConfig, Throwable, CassandraZioSession] =
ZLayer.scoped {
(for {
config <- ZIO.service[CassandraContextConfig]
// Evaluate the configuration inside of 'effect' and then create the session from it
session <- ZIO.fromAutoCloseable(
ZIO.attempt(CassandraZioSession(config.session, config.preparedStatementCacheSize))
)
} yield session)
}
def fromContextConfig(config: CassandraContextConfig): ZLayer[Any, Throwable, CassandraZioSession] =
ZLayer.succeed(config) >>> live
def fromConfig(config: Config) = fromContextConfig(CassandraContextConfig(config))
// Call the by-name constructor for the construction to fail inside of the effect if it fails
def fromPrefix(configPrefix: String) = fromContextConfig(CassandraContextConfig(LoadConfig(configPrefix)))
}