fix: preemptively fix short lived workloads having no output

This can be useful if a user hasn't had time to view the output yet.
The output will always be buffered for a few minutes before it is completely removed with the workload.

Signed-off-by: Nicholas Farshidmehr <nicholas@profian.com>

Author: definitelynobody

src/main.rs

@@ -460,6 +460,14 @@ async fn root_post(
     tokio::spawn(async move {
         // Convert from minutes to seconds.
         sleep(Duration::from_secs(ctx.timeout * 60)).await;
+
+        // Stop running the workload
+        if let Some(state) = OUT.write().await.get(&uuid) {
+            let _ = state.lock().await.exec.kill().await;
+        }
+
+        // Remove the workload state completely if it still exists
+        sleep(Duration::from_secs(5 * 60)).await;
         OUT.write().await.remove(&uuid);
     });