Update time summary calculation for tasks #446

sagiegurari · Aug 19, 2020 · e4635d1 · e4635d1
1 parent 3d82de2
commit e4635d1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/lib/runner.rs b/src/lib/runner.rs
@@ -318,7 +318,11 @@ pub(crate) fn run_task(flow_info: &FlowInfo, flow_state: &mut FlowState, step: &
                 });
 
                 match step.config.run_task {
-                    Some(ref sub_task) => run_sub_task(&flow_info, flow_state, sub_task),
+                    Some(ref sub_task) => {
+                        time_summary::add(&mut flow_state.time_summary, &step.name, start_time);
+
+                        run_sub_task(&flow_info, flow_state, sub_task);
+                    }
                     None => {
                         do_in_task_working_directory(&step, || {
                             // run script
@@ -330,11 +334,11 @@ pub(crate) fn run_task(flow_info: &FlowInfo, flow_state: &mut FlowState, step: &
                                 command::run(&updated_step);
                             };
                         });
+
+                        time_summary::add(&mut flow_state.time_summary, &step.name, start_time);
                     }
                 };
             }
-
-            time_summary::add(&mut flow_state.time_summary, &step.name, start_time);
         } else {
             let fail_message = match step.config.condition {
                 Some(ref condition) => match condition.fail_message {

diff --git a/src/lib/time_summary.rs b/src/lib/time_summary.rs
@@ -26,7 +26,7 @@ pub(crate) fn print(time_summary: &Vec<(String, u128)>) {
             }
         }
 
-        info!("=====Time Summary=====");
+        info!("================Time Summary================");
         for entry in time_summary {
             let percentage = (entry.1 as f64 / total_time as f64) * 100.0;
             let seconds = entry.1 as f64 / 1000.0;
@@ -38,6 +38,7 @@ pub(crate) fn print(time_summary: &Vec<(String, u128)>) {
                 entry.0, gap, percentage, seconds
             );
         }
+        info!("============================================");
     }
 }