sttp.client4.impl.zio.RIOMonadAsyncError.scala Maven / Gradle / Ivy

Go to download
package sttp.client4.impl.zio

import sttp.monad.{Canceler, MonadAsyncError}
import zio.{RIO, ZIO}

class RIOMonadAsyncError[R] extends MonadAsyncError[RIO[R, *]] {
  override def unit[T](t: T): RIO[R, T] = ZIO.succeed(t)

  override def map[T, T2](fa: RIO[R, T])(f: T => T2): RIO[R, T2] = fa.map(f)

  override def flatMap[T, T2](fa: RIO[R, T])(f: T => RIO[R, T2]): RIO[R, T2] =
    fa.flatMap(f)

  override def async[T](register: (Either[Throwable, T] => Unit) => Canceler): RIO[R, T] =
    ZIO.asyncInterrupt { cb =>
      val canceler = register {
        case Left(t)  => cb(ZIO.fail(t))
        case Right(t) => cb(ZIO.succeed(t))
      }

      Left(ZIO.succeed(canceler.cancel()))
    }

  override def error[T](t: Throwable): RIO[R, T] = ZIO.fail(t)

  override protected def handleWrappedError[T](rt: RIO[R, T])(h: PartialFunction[Throwable, RIO[R, T]]): RIO[R, T] =
    rt.catchSome(h)

  override def eval[T](t: => T): RIO[R, T] = ZIO.attempt(t)

  override def suspend[T](t: => RIO[R, T]): RIO[R, T] = ZIO.suspend(t)

  override def flatten[T](ffa: RIO[R, RIO[R, T]]): RIO[R, T] = ffa.flatten

  override def ensure[T](f: RIO[R, T], e: => RIO[R, Unit]): RIO[R, T] = f.ensuring(e.ignore)
}