From 036952656fad23e653f09dd1659006d4e2104e94 Mon Sep 17 00:00:00 2001
From: Vincent Viloria <vvilo@amazon.com>
Date: Fri, 28 Feb 2025 09:37:35 -0800
Subject: [PATCH] update KclStsAssumeRoleCredentialsProvider to have
 resolveCredentials()

---
 .../KclStsAssumeRoleCredentialsProvider.java  | 26 +++++++++++++++----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/amazon-kinesis-client-multilang/src/main/java/software/amazon/kinesis/multilang/auth/KclStsAssumeRoleCredentialsProvider.java b/amazon-kinesis-client-multilang/src/main/java/software/amazon/kinesis/multilang/auth/KclStsAssumeRoleCredentialsProvider.java
index 65c096ae..96ecfdd6 100644
--- a/amazon-kinesis-client-multilang/src/main/java/software/amazon/kinesis/multilang/auth/KclStsAssumeRoleCredentialsProvider.java
+++ b/amazon-kinesis-client-multilang/src/main/java/software/amazon/kinesis/multilang/auth/KclStsAssumeRoleCredentialsProvider.java
@@ -1,3 +1,17 @@
+/*
+ * Copyright 2024 Amazon.com, Inc. or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
 package software.amazon.kinesis.multilang.auth;
 
 import java.net.URI;
@@ -17,6 +31,7 @@ import software.amazon.kinesis.multilang.NestedPropertyProcessor;
 public class KclStsAssumeRoleCredentialsProvider implements AwsCredentialsProvider, NestedPropertyProcessor {
     private final Builder assumeRoleRequestBuilder;
     private final StsClientBuilder stsClientBuilder;
+    private final StsAssumeRoleCredentialsProvider stsAssumeRoleCredentialsProvider;
 
     public KclStsAssumeRoleCredentialsProvider(String[] params) {
         this(params[0], params[1], Arrays.copyOfRange(params, 2, params.length));
@@ -27,15 +42,16 @@ public class KclStsAssumeRoleCredentialsProvider implements AwsCredentialsProvid
                 AssumeRoleRequest.builder().roleArn(roleArn).roleSessionName(roleSessionName);
         this.stsClientBuilder = StsClient.builder();
         NestedPropertyKey.parse(this, params);
+        this.stsAssumeRoleCredentialsProvider = StsAssumeRoleCredentialsProvider.builder()
+                .refreshRequest(assumeRoleRequestBuilder.build())
+                .asyncCredentialUpdateEnabled(true)
+                .stsClient(stsClientBuilder.build())
+                .build();
     }
 
     @Override
     public AwsCredentials resolveCredentials() {
-        StsAssumeRoleCredentialsProvider provider = StsAssumeRoleCredentialsProvider.builder()
-                .refreshRequest(assumeRoleRequestBuilder.build())
-                .stsClient(stsClientBuilder.build())
-                .build();
-        return provider.resolveCredentials();
+        return stsAssumeRoleCredentialsProvider.resolveCredentials();
     }
 
     @Override