public class AWSAuthCredentialsProvider implements CredentialsResolver {

    private final String userName;
    private final IAMAuthTokenRequest iamAuthTokenRequest;
    private final AwsCredentialsProvider awsCredentialsProvider;

    private volatile CompletionStage future;
    private volatile Long lastTime = System.currentTimeMillis();

    public AWSAuthCredentialsProvider(String userName, IAMAuthTokenRequest iamAuthTokenRequest,
            AwsCredentialsProvider awsCredentialsProvider) {
        this.userName = userName;
        this.iamAuthTokenRequest = iamAuthTokenRequest;
        this.awsCredentialsProvider = awsCredentialsProvider;
    }

    @Override
    public CompletionStage resolve(InetSocketAddress address) {
        if (System.currentTimeMillis() - lastTime > 15 * 60 * 1000 || future == null) {
            String token = iamAuthTokenRequest.toSignedRequestUri(
                    awsCredentialsProvider.resolveCredentials());
            future = CompletableFuture.completedFuture(new Credentials(userName, token));
            lastTime = System.currentTimeMillis();
        }
        return future;
    }
}

/** The generate awsCredentialsProvider code */
public AwsCredentials getAwsCredentials(AWSCredentialConfig awsCredentialConfig, Integer durationSeconds) {
        AwsBasicCredentials awsCreds = AwsBasicCredentials.create(awsCredentialConfig.getAccessKey(), awsCredentialConfig.getSecretKey());
        StsClient stsClient = StsClient.builder()
                .region(awsCredentialConfig.getRegion())
                .credentialsProvider(StaticCredentialsProvider.create(awsCreds)).build();
        AssumeRoleRequest assumeRoleRequest = createAssumeRoleRequest(stsClient, awsCredentialConfig.getRoleName(), durationSeconds);

        Credentials sessionCredentials = stsClient.assumeRole(assumeRoleRequest).credentials();
        return AwsSessionCredentials.create(
                sessionCredentials.accessKeyId(),
                sessionCredentials.secretAccessKey(),
                sessionCredentials.sessionToken());
    }