luciano/amazon-kinesis-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix sts assume role credential provider logic

Browse source
This commit is contained in:
Ethan Katnic 2024-11-04 16:47:42 -08:00
parent 210d61a843
commit 88ff519d51
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
amazon-kinesis-client-multilang/src/main/java/software/amazon/kinesis/multilang/auth/KclStsAssumeRoleCredentialsProvider.java
View file

@ -17,6 +17,7 @@ import software.amazon.kinesis.multilang.NestedPropertyProcessor;
public class KclStsAssumeRoleCredentialsProvider implements AwsCredentialsProvider, NestedPropertyProcessor { public class KclStsAssumeRoleCredentialsProvider implements AwsCredentialsProvider, NestedPropertyProcessor {
private final Builder assumeRoleRequestBuilder; private final Builder assumeRoleRequestBuilder;
private final StsClientBuilder stsClientBuilder; private final StsClientBuilder stsClientBuilder;
private final StsAssumeRoleCredentialsProvider stsAssumeRoleCredentialsProvider;
public KclStsAssumeRoleCredentialsProvider(String[] params) { public KclStsAssumeRoleCredentialsProvider(String[] params) {
this(params[0], params[1], Arrays.copyOfRange(params, 2, params.length)); this(params[0], params[1], Arrays.copyOfRange(params, 2, params.length));
@ -27,15 +28,16 @@ public class KclStsAssumeRoleCredentialsProvider implements AwsCredentialsProvid
AssumeRoleRequest.builder().roleArn(roleArn).roleSessionName(roleSessionName); AssumeRoleRequest.builder().roleArn(roleArn).roleSessionName(roleSessionName);
this.stsClientBuilder = StsClient.builder(); this.stsClientBuilder = StsClient.builder();
NestedPropertyKey.parse(this, params); NestedPropertyKey.parse(this, params);
this.stsAssumeRoleCredentialsProvider = StsAssumeRoleCredentialsProvider.builder()
.refreshRequest(assumeRoleRequestBuilder.build())
.asyncCredentialUpdateEnabled(true)
.stsClient(stsClientBuilder.build())
.build();
} }
@Override @Override
public AwsCredentials resolveCredentials() { public AwsCredentials resolveCredentials() {
StsAssumeRoleCredentialsProvider provider = StsAssumeRoleCredentialsProvider.builder() return stsAssumeRoleCredentialsProvider.resolveCredentials();
.refreshRequest(assumeRoleRequestBuilder.build())
.stsClient(stsClientBuilder.build())
.build();
return provider.resolveCredentials();
} }
@Override @Override