1 files changed, 24 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/pdsuicommon/computation/FutureToComputationOps.scala b/src/main/scala/xyz/driver/pdsuicommon/computation/FutureToComputationOps.scala
new file mode 100644
index 0000000..c5800dc
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/computation/FutureToComputationOps.scala
@@ -0,0 +1,24 @@
+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))
+  }
+}