feat(on_failure): add behaviour for on_failure. When set to exit, i…

…t will do a graceful exit.

src/commands/run.rs

@@ -1,4 +1,4 @@
-use crate::config::{Config, WhenNotify};
+use crate::config::{Config, OnFailure, WhenNotify};
 use crate::fst::*;
 use crate::utils::traits::CommandConfig;
 use libc::{fork, signal};
@@ -75,10 +75,12 @@ impl Run {
 
     let mut exit_success = 0;
     let mut exit_failure = 0;
+    let mut ask_for_exit = false;
     let graph_iter = &mut graph.iter();
     loop {
       if graph_iter.has_next()
         && (graph_iter.n_in_progress() < config.concurrency() || config.concurrency() < 0)
+        && !ask_for_exit
       {
         let task = graph_iter.next().unwrap();
         let label = task.label().to_string();
@@ -100,35 +102,48 @@ impl Run {
         for id in 0..processes.len() {
           if let Some(child) = processes[id].as_mut() {
             if let Ok(Some(exit)) = child.try_wait() {
-              if exit.success() {
+              let label = graph.get_state_from_id(id).label().to_string();
+              let is_failure = if exit.success() {
                 exit_success = exit_success + 1;
+                false
               } else {
                 exit_failure = exit_failure + 1;
-              }
+                true
+              };
 
               done = done + 1;
               graph_iter.mark_done(id);
               processes[id] = None;
 
-              let msg = format!(
-                "Task {} ended with status code {}",
-                graph.get_state_from_id(id).label(),
-                exit
-              );
+              let msg = format!("Task {} ended with status code {}", label, exit);
               self.notify(&config, msg, WhenNotify::TaskEnd);
+
+              if let Some(OnFailure::Exit) = config.tasks().get(&label).unwrap().on_failure() {
+                if is_failure {
+                  ask_for_exit = true;
+                }
+              }
             }
           }
         }
 
-        if done == 0 {
+        if graph_iter.n_in_progress() == 0 && ask_for_exit {
+          break;
+        } else if done == 0 {
           std::thread::sleep(std::time::Duration::from_millis(100));
         }
       }
     }
 
     let msg = format!(
-      "All tasks ended. Got {} success and {} failure.",
-      exit_success, exit_failure
+      "All tasks ended. Got {} success and {} failure.{}",
+      exit_success,
+      exit_failure,
+      if ask_for_exit {
+        " Contains one critical failure."
+      } else {
+        ""
+      }
     );
     self.notify(&config, msg, WhenNotify::End);
 

src/config/task.rs

@@ -10,7 +10,12 @@ pub struct Task {
 }
 
 impl Task {
-  pub fn new(id: String, commands: Vec<String>, depends_on: Vec<String>, on_failure: Option<OnFailure>) -> Task {
+  pub fn new(
+    id: String,
+    commands: Vec<String>,
+    depends_on: Vec<String>,
+    on_failure: Option<OnFailure>,
+  ) -> Task {
     Task {
       id: id,
       commands: commands,

src/config/yaml_trait.rs

@@ -165,7 +165,10 @@ impl YamlTasksScheduler for LinkedHashMap<Yaml, Yaml> {
         "continue" => Ok(Some(OnFailure::Continue)),
         "exit" => Ok(Some(OnFailure::Exit)),
         "" => Ok(None),
-        _ => Err(format!("{} is an incorrect value for on_failure", on_failure)),
+        _ => Err(format!(
+          "{} is an incorrect value for on_failure",
+          on_failure
+        )),
       }
     } else {
       Ok(None)