Simplify control flow

elabit · Nov 8, 2023 · f2b21ee · f2b21ee
1 parent 0d11c8b
commit f2b21ee
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/v2/robotmk/src/bin/scheduler/child_process_supervisor.rs b/v2/robotmk/src/bin/scheduler/child_process_supervisor.rs
@@ -79,20 +79,17 @@ impl ChildProcessSupervisor<'_> {
                 return Ok(ChildProcessOutcome::Terminated);
             }
 
-            if let Some(exit_status) = child
-                .try_wait()
-                .context(format!(
-                    "Failed to query exit status of process {}, killing",
-                    child.id()
-                ))
-                .map_err(|err| {
+            match child.try_wait() {
+                Ok(Some(exit_status)) => return Ok(ChildProcessOutcome::Exited(exit_status)),
+                Ok(None) => {}
+                e => {
                     kill_child_tree(child);
-                    err
-                })?
-            {
-                return Ok(ChildProcessOutcome::Exited(exit_status));
+                    e.context(format!(
+                        "Failed to query exit status of process {}, killing",
+                        child.id()
+                    ))?;
+                }
             }
-
             sleep(Duration::from_millis(250)).await
         }
     }