feat: add retry for run export (#1349)

prefix-dev · Jan 17, 2025 · c9c7adc · c9c7adc
1 parent 715f6b4
commit c9c7adc
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 6 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -3,7 +3,7 @@ members = ["rust-tests"]
 
 [package]
 name = "rattler-build"
-version = "0.35.2"
+version = "0.35.3"
 authors = ["rattler-build contributors <[email protected]>"]
 repository = "https://github.com/prefix-dev/rattler-build"
 edition = "2021"

diff --git a/py-rattler-build/Cargo.lock b/py-rattler-build/Cargo.lock
diff --git a/py-rattler-build/Cargo.toml b/py-rattler-build/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "py-rattler-build"
-version = "0.35.2"
+version = "0.35.3"
 edition = "2021"
 license = "BSD-3-Clause"
 publish = false

diff --git a/src/run_exports.rs b/src/run_exports.rs
@@ -6,6 +6,7 @@ use rattler_conda_types::{
     package::{PackageFile, RunExportsJson},
     RepoDataRecord,
 };
+use rattler_networking::retry_policies::default_retry_policy;
 use reqwest_middleware::ClientWithMiddleware;
 use thiserror::Error;
 use tokio::sync::Semaphore;
@@ -96,7 +97,13 @@ impl RunExportExtractor {
             .expect("semaphore error");
 
         match package_cache
-            .get_or_fetch_from_url(cache_key, url, client, Some(Arc::new(progress_reporter)))
+            .get_or_fetch_from_url_with_retry(
+                cache_key,
+                url,
+                client,
+                default_retry_policy(),
+                Some(Arc::new(progress_reporter)),
+            )
             .await
         {
             Ok(package_dir) => Ok(RunExportsJson::from_package_directory(package_dir.path()).ok()),