zio.nebula.net.NebulaClientLive.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zio-nebula_2.13 Show documentation
zio-nebula
The newest version!
package zio.nebula.net

import scala.jdk.CollectionConverters._

import zio._
import zio.nebula._
import zio.nebula.NebulaPoolConfig

import com.vesoft.nebula.client.graph.{ NebulaPoolConfig => _ }
import com.vesoft.nebula.client.graph.data.HostAddress
import com.vesoft.nebula.client.graph.net.{ NebulaPool => NebulaPl }

/**
 * @author
 *   梦境迷离
 * @version 1.0,2023/8/29
 */
private[nebula] final class NebulaClientLive(underlying: NebulaPl) extends NebulaClient {

  def init(): ZIO[NebulaSessionPoolConfig & NebulaPoolConfig, Throwable, Boolean] =
    for {
      config <- ZIO.service[NebulaPoolConfig]
      status <-
        ZIO.serviceWithZIO[NebulaSessionPoolConfig](sessionConfig =>
          ZIO.attempt(
            underlying.init(sessionConfig.address.map(d => new HostAddress(d.host, d.port)).asJava, config.toJava)
          )
        )
    } yield status

  def close(): Task[Unit] = ZIO.attempt(underlying.close())

  def openSession(sessionPoolConfig: NebulaSessionPoolConfig): ZIO[Any, Throwable, NebulaSession] =
    for {
      session <- ZIO.attempt(
                   new NebulaSession(
                     underlying.getSession(
                       sessionPoolConfig.auth.username,
                       sessionPoolConfig.auth.password,
                       sessionPoolConfig.reconnect
                     )
                   )
                 )
      _       <- session.execute(Stmt.str(s"USE `${sessionPoolConfig.spaceName}`"))
    } yield session

  def openSession(): ZIO[NebulaSessionPoolConfig, Throwable, NebulaSession] =
    for {
      sessionConfig <- ZIO.service[NebulaSessionPoolConfig]
      session       <- ZIO.attempt(
                         new NebulaSession(
                           underlying.getSession(
                             sessionConfig.auth.username,
                             sessionConfig.auth.password,
                             sessionConfig.reconnect
                           )
                         )
                       )
      _             <- session.execute(Stmt.str(s"USE `${sessionConfig.spaceName}`"))

    } yield session

  def activeConnNum: Task[Int] = ZIO.attempt(underlying.getActiveConnNum)

  def idleConnNum: Task[Int] = ZIO.attempt(underlying.getIdleConnNum)

  def waitersNum: Task[Int] = ZIO.attempt(underlying.getWaitersNum)

}