From bdd30f7ab8c2280c368b86cb12baec0578669d44 Mon Sep 17 00:00:00 2001
From: Héctor Ramón Jiménez <hector@hecrj.dev>
Date: Thu, 20 Jun 2024 01:11:21 +0200
Subject: Introduce `and_then` methods for fallible `Task`s

---
 runtime/src/task.rs | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

(limited to 'runtime')

diff --git a/runtime/src/task.rs b/runtime/src/task.rs
index 740360ac..b8a83d6d 100644
--- a/runtime/src/task.rs
+++ b/runtime/src/task.rs
@@ -242,6 +242,39 @@ impl<T> Task<T> {
     }
 }
 
+impl<T> Task<Option<T>> {
+    /// Executes a new [`Task`] after this one, only when it produces `Some` value.
+    ///
+    /// The value is provided to the closure to create the subsequent [`Task`].
+    pub fn and_then<A>(
+        self,
+        f: impl Fn(T) -> Task<A> + MaybeSend + 'static,
+    ) -> Task<A>
+    where
+        T: MaybeSend + 'static,
+        A: MaybeSend + 'static,
+    {
+        self.then(move |option| option.map_or_else(Task::none, &f))
+    }
+}
+
+impl<T, E> Task<Result<T, E>> {
+    /// Executes a new [`Task`] after this one, only when it succeeds with an `Ok` value.
+    ///
+    /// The success value is provided to the closure to create the subsequent [`Task`].
+    pub fn and_then<A>(
+        self,
+        f: impl Fn(T) -> Task<A> + MaybeSend + 'static,
+    ) -> Task<A>
+    where
+        T: MaybeSend + 'static,
+        E: MaybeSend + 'static,
+        A: MaybeSend + 'static,
+    {
+        self.then(move |option| option.map_or_else(|_| Task::none(), &f))
+    }
+}
+
 impl<T> From<()> for Task<T>
 where
     T: MaybeSend + 'static,
-- 
cgit