Add optional retry_config to FileStore::new

helium · Dec 10, 2024 · 5484f91 · 5484f91
1 parent 4f5a52d
commit 5484f91
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/file_store/src/file_store.rs b/file_store/src/file_store.rs
@@ -3,7 +3,7 @@ use crate::{
     settings::{self, Settings},
     BytesMutStream, Error, FileInfo, FileInfoStream, Result,
 };
-use aws_config::{meta::region::RegionProviderChain, timeout::TimeoutConfig};
+use aws_config::{meta::region::RegionProviderChain, retry::RetryConfig, timeout::TimeoutConfig};
 use aws_sdk_s3::{types::ByteStream, Client, Endpoint, Region};
 use chrono::{DateTime, Utc};
 use futures::FutureExt;
@@ -64,6 +64,7 @@ impl FileStore {
         endpoint: Option<String>,
         region: Option<String>,
         timeout_config: Option<TimeoutConfig>,
+        retry_config: Option<RetryConfig>,
     ) -> Result<Self> {
         let endpoint: Option<Endpoint> = match &endpoint {
             Some(endpoint) => Uri::from_str(endpoint)
@@ -84,6 +85,10 @@ impl FileStore {
             config = config.timeout_config(timeout);
         }
 
+        if let Some(retry) = retry_config {
+            config = config.retry_config(retry);
+        }
+
         let config = config.load().await;
 
         let client = Client::new(&config);