src/main/scala/xyz/driver/pdsuicommon/computation/FutureToComputationOps.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

package xyz.driver.pdsuicommon.computation

import xyz.driver.pdsuicommon.error.DomainError

import scala.concurrent.{ExecutionContext, Future}

final class FutureToComputationOps[T](val self: Future[T]) extends AnyVal {

  def handleDomainError[U, ER](f: PartialFunction[T, U])(implicit unsuitableToErrorsResponse: DomainError => ER,
                                                         ec: ExecutionContext): Future[Either[ER, U]] = {
    self.map {
      case x if f.isDefinedAt(x) => Right(f(x))
      case x: DomainError        => Left(unsuitableToErrorsResponse(x))
      case x                     => throw new RuntimeException(s"Can not process $x")
    }
  }

  def toComputation[U, ER](f: PartialFunction[T, U])(implicit unsuitableToErrorsResponse: DomainError => ER,
                                                     ec: ExecutionContext): Computation[ER, U] = {
    Computation(handleDomainError(f))
  }
}