Allow DynamoDB consumers to bring their own config

src/services/dynamodb/mod.rs

@@ -1,3 +1,4 @@
+use aws_config::ConfigLoader;
 use aws_sdk_dynamodb::{config::Builder as ConfigBuilder, Client as DynamoDBClient, Endpoint};
 use aws_types::region::Region;
 use cached::proc_macro::cached;
@@ -11,6 +12,19 @@ pub use client_extensions::{
     DynamoDBClientExt, DynamoDBModel, Version, VersionField, Versioned, VersionedPutError,
 };
 
+fn determine_endpoint(config: SdkConfig) -> DynamoDBClient {
+    match std::env::var("TARGET").map(|target| target.eq_ignore_ascii_case("local")) {
+        Ok(true) => {
+            let dynamo_config = ConfigBuilder::from(&config)
+                .endpoint_resolver(Endpoint::immutable("http://localhost:8000").unwrap())
+                .build();
+
+            DynamoDBClient::from_conf(dynamo_config)
+        }
+        _ => DynamoDBClient::new(&config),
+    }
+}
+
 async fn dynamodb_client(region: Option<&'static str>) -> DynamoDBClient {
     let config_builder = aws_config::from_env();
 
@@ -21,19 +35,27 @@ async fn dynamodb_client(region: Option<&'static str>) -> DynamoDBClient {
     .load()
     .await;
 
-    match std::env::var("TARGET").map(|target| target.eq_ignore_ascii_case("local")) {
-        Ok(true) => {
-            let dynamo_config = ConfigBuilder::from(&config)
-                .endpoint_resolver(Endpoint::immutable("http://localhost:8000").unwrap())
-                .build();
+    determine_endpoint(config)
+}
 
-            DynamoDBClient::from_conf(dynamo_config)
-        }
-        _ => DynamoDBClient::new(&config),
-    }
+async fn dynamodb_client(config_loader: Option<ConfigLoader>) -> DynamoDBClient {
+    let config = {
+        let Some(config) = config_loader else {
+            aws_config::from_env()
+        };
+
+        config.load().await
+    };
+
+    determine_endpoint(config)
+}
+
+#[cached]
+pub async fn dynamodb(config_loader: Option<ConfigLoader>) -> DynamoDBClient {
+    dynamodb_client(config_loader).await
 }
 
 #[cached]
 pub async fn dynamodb(region: Option<&'static str>) -> DynamoDBClient {
-    dynamodb_client(region).await
+    dynamodb_client(config_loader).await
 }