From f068edd95f960dea97bb874430c43e55d6b7d186 Mon Sep 17 00:00:00 2001
From: diogo464 <diogo464@d464.sh>
Date: Thu, 24 Jul 2025 12:46:32 +0100
Subject: added basic retry logic to the machine_containers_wait function

---
 src/main.rs | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index cba59b8..4fd002a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -544,7 +544,23 @@ async fn machine_containers_wait(
 ) -> Result<()> {
     tracing::info!("waiting for {} containers to exit", containers.len());
     let script = machine_containers_wait_script(containers);
-    machine_run_script(ctx, machine, &script).await?;
+    let retry_seconds = 5;
+    let mut retries = 10;
+    loop {
+        match machine_run_script(ctx, machine, &script).await {
+            Ok(_) => break,
+            Err(err) => {
+                tracing::debug!("failed to run script: {err}, {retries} left");
+                if retries == 0 {
+                    return Err(err);
+                }
+                retries -= 1;
+                tracing::debug!("waiting {retry_seconds} before retrying...");
+                tokio::time::sleep(Duration::from_secs(retry_seconds)).await;
+                tracing::debug!("retrying");
+            }
+        }
+    }
     tracing::info!("all containers exited");
     Ok(())
 }
-- 
cgit