liewhite.rpc.Transport.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rpc_3 Show documentation

rpc

The newest version!

package liewhite.rpc

import zio.*
import com.rabbitmq.client.*
import zio.concurrent.ConcurrentSet
import scala.util.Try

class Transport(val connection: Connection) {
  def newChannel(qos: Int = 1) = {
    val channel = connection.createChannel()
    channel.confirmSelect()
    channel.basicQos(qos)
    channel
  }

  def scopedChannel(qos: Int = 1) = {
    val channel = connection.createChannel()
    channel.confirmSelect()
    channel.basicQos(qos)
    ZIO.acquireRelease(ZIO.succeed(channel))(c =>
      ZIO.logInfo(s"close channel: ${c.getChannelNumber()}") *> ZIO.succeed(Try(c.close())).debug("close channel result: ")
    )
  }

  def close() = ZIO.succeed(
    connection.close()
  )

}

object Transport {
  def layer(uri: String): ZLayer[Scope, Nothing, Transport] = {
    val tp = ZIO.acquireRelease({
      ZIO.succeed(newTransport(uri))
    })(tp =>
      ZIO.debug("release transport") *> ZIO
        .succeed(tp.close())
    )
    ZLayer(tp)
    
  }

  def newTransport(uri: String) = {
    val factory = new ConnectionFactory()
    factory.setUri(uri)
    val connection = factory.newConnection()
    Transport(connection)
  }
}