fix: downloader shutdown leads to verification failure

bytebeamio · Apr 4, 2024 · 8192348 · 8192348
1 parent 2e984f4
commit 8192348
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/uplink/src/collector/downloader.rs b/uplink/src/collector/downloader.rs
@@ -207,18 +207,19 @@ impl FileDownloader {
             }
         };
         select! {
-            // NOTE: if download has timedout don't do anything, else ensure errors are forwarded after three retries
-            o = timeout_at(deadline, self.continuous_retry(state)) => match o {
-                Ok(r) => r?,
-                Err(_) => error!("Last download has timedout"),
-            },
-
             _ = shutdown_rx.recv_async() => {
                 if let Err(e) = state.save(&self.config) {
                     error!("Error saving current_download: {e}");
                 }
-            }
 
+                return Ok(());
+            },
+
+            // NOTE: if download has timedout don't do anything, else ensure errors are forwarded after three retries
+            o = timeout_at(deadline, self.continuous_retry(state)) => match o {
+                Ok(r) => r?,
+                Err(_) => error!("Last download has timedout"),
+            }
         }
 
         state.current.meta.verify_checksum()?;