update DDB client to a 3 retry max to avoid lambda timeout + update to auth token

Author: tinyg210

FIS-experiments/README.md

@@ -34,7 +34,7 @@ cd lambda-functions && mvn clean package shade:shade
 ### Starting LocalStack
 
 ```bash
-export LOCALSTACK_API_KEY = <your_localstack_api_key>
+export LOCALSTACK_AUTH_TOKEN = <your_localstack_auth_token>
 docker compose up
 ```
 

FIS-experiments/docker-compose.yml

@@ -17,7 +17,7 @@ services:
       - LOCALSTACK_HOST=localstack # where services are available from other containers
       - ENFORCE_IAM=0 # enforce IAM policies
       - LAMBDA_DOCKER_NETWORK=ls_network
-      - LOCALSTACK_API_KEY=${LOCALSTACK_API_KEY}
+      - LOCALSTACK_AUTH_TOKEN=${LOCALSTACK_AUTH_TOKEN}
       - LAMBDA_RUNTIME_ENVIRONMENT_TIMEOUT=60
       - PERSIST_ALL=false
 

FIS-experiments/lambda-functions/src/main/java/lambda/ProductApi.java

@@ -2,8 +2,11 @@
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import java.net.URI;
+
 import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
 import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
+import software.amazon.awssdk.core.client.config.ClientOverrideConfiguration;
+import software.amazon.awssdk.core.retry.RetryPolicy;
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
 import software.amazon.awssdk.services.sns.SnsClient;
@@ -15,6 +18,17 @@ public class ProductApi {
   protected static final String topicArn = "arn:aws:sns:us-east-1:000000000000:ProductEventsTopic";
   protected ObjectMapper objectMapper = new ObjectMapper();
 
+  // Define a custom retry policy
+  // Set maximum number of retries
+  RetryPolicy customRetryPolicy = RetryPolicy.builder()
+          .numRetries(3)
+          .build();
+
+  // Apply the custom retry policy to ClientOverrideConfiguration
+  ClientOverrideConfiguration clientOverrideConfig = ClientOverrideConfiguration.builder()
+          .retryPolicy(customRetryPolicy)
+          .build();
+
   protected SnsClient snsClient = SnsClient.builder()
       .endpointOverride(URI.create(String.format("http://%s:4566", LOCALSTACK_HOSTNAME)))
       .credentialsProvider(
@@ -28,5 +42,6 @@ public class ProductApi {
           StaticCredentialsProvider.create(AwsBasicCredentials.create("test", "test")))
       .region(Region.of(AWS_REGION))
       .endpointDiscoveryEnabled(true)
+      .overrideConfiguration(clientOverrideConfig)
       .build();
 }

FIS-experiments/lambda-functions/target/product-lambda.jar

@@ -15,7 +15,7 @@ services:
       - LOCALSTACK_HOST=localstack # where services are available from other containers
       - LAMBDA_DOCKER_NETWORK=ls_network
       - DISABLE_CUSTOM_CORS_APIGATEWAY=1
-      - LOCALSTACK_API_KEY=${LOCALSTACK_API_KEY}
+      - LOCALSTACK_AUTH_TOKEN=${LOCALSTACK_AUTH_TOKEN}
       - EXTENSION_AUTO_INSTALL=localstack-extension-outages
       - LAMBDA_RUNTIME_ENVIRONMENT_TIMEOUT=60
       - PERSIST_ALL=false

extension-outages/docker-compose.yml

@@ -17,7 +17,7 @@ services:
       - LOCALSTACK_HOST=localstack # where services are available from other containers
       - ENFORCE_IAM=0 # enforce IAM policies
       - LAMBDA_DOCKER_NETWORK=ls_network
-      - LOCALSTACK_API_KEY=${LOCALSTACK_API_KEY}
+      - LOCALSTACK_AUTH_TOKEN=${LOCALSTACK_AUTH_TOKEN}
       - LAMBDA_RUNTIME_ENVIRONMENT_TIMEOUT=60
       - DNS_ADDRESS=127.0.0.1
 #      - EXTENSION_AUTO_INSTALL=localstack-extension-outages

route53-failover/docker-compose.yml

@@ -4,6 +4,8 @@
 import java.net.URI;
 import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
 import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
+import software.amazon.awssdk.core.client.config.ClientOverrideConfiguration;
+import software.amazon.awssdk.core.retry.RetryPolicy;
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
 import software.amazon.awssdk.services.sns.SnsClient;
@@ -15,6 +17,17 @@ public class ProductApi {
   protected static final String topicArn = "arn:aws:sns:us-east-1:000000000000:ProductEventsTopic";
   protected ObjectMapper objectMapper = new ObjectMapper();
 
+  // Define a custom retry policy
+  // Set maximum number of retries
+  RetryPolicy customRetryPolicy = RetryPolicy.builder()
+          .numRetries(3)
+          .build();
+
+  // Apply the custom retry policy to ClientOverrideConfiguration
+  ClientOverrideConfiguration clientOverrideConfig = ClientOverrideConfiguration.builder()
+          .retryPolicy(customRetryPolicy)
+          .build();
+
   protected SnsClient snsClient = SnsClient.builder()
       .endpointOverride(URI.create(String.format("http://%s:4566", LOCALSTACK_HOSTNAME)))
       .credentialsProvider(
@@ -28,5 +41,6 @@ public class ProductApi {
           StaticCredentialsProvider.create(AwsBasicCredentials.create("test", "test")))
       .region(Region.of(AWS_REGION))
       .endpointDiscoveryEnabled(true)
+      .overrideConfiguration(clientOverrideConfig)
       .build();
 }