Merge pull request #4 from balena-io-experimental/job-from-array

Allow using array return on compound jobs
balena-io-experimental · Oct 29, 2024 · d1a403d · d1a403d
2 parents a8df7d3 + 939f067
commit d1a403d
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/task/handler.rs b/src/task/handler.rs
@@ -47,6 +47,23 @@ impl<S: 'static, E: std::error::Error + 'static> IntoEffect<Vec<Task<S>>, Error>
     }
 }
 
+impl<S, const N: usize> IntoEffect<Vec<Task<S>>, Error> for [Task<S>; N] {
+    fn into_effect(self, _: &System) -> Effect<Vec<Task<S>>, Error> {
+        Effect::of(self.into())
+    }
+}
+
+impl<S: 'static, E: std::error::Error + 'static, const N: usize> IntoEffect<Vec<Task<S>>, Error>
+    for Result<[Task<S>; N], E>
+{
+    fn into_effect(self, _: &System) -> Effect<Vec<Task<S>>, Error> {
+        Effect::from(
+            self.map_err(|e| Error::Other(Box::new(e)))
+                .map(|a| a.into()),
+        )
+    }
+}
+
 macro_rules! impl_action_handler {
     (
         $first:ident, $($ty:ident),*