Add the possibility for monads to recover from errors

1 files changed, 5 insertions, 0 deletions

diff --git a/okhttp-backend/monix/src/main/scala/com/softwaremill/sttp/okhttp/monix/OkHttpMonixBackend.scala b/okhttp-backend/monix/src/main/scala/com/softwaremill/sttp/okhttp/monix/OkHttpMonixBackend.scala
index e85d271..f579ff5 100644
--- a/okhttp-backend/monix/src/main/scala/com/softwaremill/sttp/okhttp/monix/OkHttpMonixBackend.scala
+++ b/okhttp-backend/monix/src/main/scala/com/softwaremill/sttp/okhttp/monix/OkHttpMonixBackend.scala
@@ -117,4 +117,9 @@ private[monix] object TaskMonad extends MonadAsyncError[Task] {
     }
 
   override def error[T](t: Throwable): Task[T] = Task.raiseError(t)
+
+  override protected def handleWrappedError[T](rt: Task[T])(
+      h: PartialFunction[Throwable, Task[T]]): Task[T] = rt.onErrorRecoverWith {
+    case t => h(t)
+  }
 }