Skip to content

Commit 67acb73

Browse files
author
Siddharth Yagnik
committed
Allowing async refresh for role credential provider
1 parent df5d25d commit 67acb73

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/main/java/software/amazon/msk/auth/iam/internals/MSKCredentialProvider.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -166,8 +166,9 @@ protected AwsCredentialsProvider getDefaultProvider() {
166166

167167
@Override
168168
public AwsCredentials resolveCredentials() {
169+
log.error("RESOLVING CREDENTIALS!!");
169170
AwsCredentials credentials = loadCredentialsWithRetry();
170-
if (credentials != null && shouldDebugCreds && log.isDebugEnabled()) {
171+
if (credentials != null && shouldDebugCreds) {
171172
logCallerIdentity(credentials);
172173
}
173174
return credentials;
@@ -218,7 +219,7 @@ private void logCallerIdentity(AwsCredentials credentials) {
218219
try {
219220
StsClient stsClient = getStsClientForDebuggingCreds(credentials);
220221
GetCallerIdentityResponse response = stsClient.getCallerIdentity();
221-
log.debug("The identity of the credentials is {}", response.toString());
222+
log.error("The identity of the credentials is {}", response.toString());
222223
} catch (Exception e) {
223224
//If we run into an exception logging the caller identity, we should log the exception but
224225
//continue running.
@@ -362,6 +363,7 @@ StsAssumeRoleCredentialsProvider createSTSRoleCredentialProvider(
362363
return StsAssumeRoleCredentialsProvider.builder()
363364
.stsClient(stsClient)
364365
.refreshRequest(roleRequest)
366+
.asyncCredentialUpdateEnabled(true)
365367
.build();
366368
}
367369

@@ -379,6 +381,7 @@ StsAssumeRoleCredentialsProvider createSTSRoleCredentialProvider(
379381
return StsAssumeRoleCredentialsProvider.builder()
380382
.stsClient(stsClient)
381383
.refreshRequest(roleRequest)
384+
.asyncCredentialUpdateEnabled(true)
382385
.build();
383386
}
384387

@@ -395,6 +398,7 @@ StsAssumeRoleCredentialsProvider createSTSRoleCredentialProvider(
395398
return StsAssumeRoleCredentialsProvider.builder()
396399
.stsClient(getStsClientBuilder(Region.of(stsRegion)).build())
397400
.refreshRequest(roleRequest)
401+
.asyncCredentialUpdateEnabled(true)
398402
.build();
399403
}
400404
}

0 commit comments

Comments
 (0)