Fix Null STS Client when role_arn is used.

Signed-off-by: Anas <[email protected]>
prometheus · Nov 6, 2020 · ecd822f · ecd822f
1 parent c7fd02f
commit ecd822f
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/main/java/io/prometheus/cloudwatch/CloudWatchCollector.java b/src/main/java/io/prometheus/cloudwatch/CloudWatchCollector.java
@@ -42,6 +42,7 @@
 import software.amazon.awssdk.services.resourcegroupstaggingapi.model.ResourceTagMapping;
 import software.amazon.awssdk.services.resourcegroupstaggingapi.model.Tag;
 import software.amazon.awssdk.services.resourcegroupstaggingapi.model.TagFilter;
+import software.amazon.awssdk.services.sts.StsClient;
 import software.amazon.awssdk.services.sts.auth.StsAssumeRoleCredentialsProvider;
 import software.amazon.awssdk.services.sts.model.AssumeRoleRequest;
 
@@ -280,12 +281,13 @@ private void loadConfig(ArrayList<MetricRule> rules, CloudWatchClient cloudWatch
     }
 
     private AwsCredentialsProvider getRoleCredentialProvider(Map<String, Object> config) {
-        StsAssumeRoleCredentialsProvider credentialsProvider = StsAssumeRoleCredentialsProvider.builder()
-                .refreshRequest(AssumeRoleRequest.builder()
-                        .roleArn((String) config.get("role_arn"))
-                        .roleSessionName("cloudwatch_exporter").build()
-                ).build();
-      return credentialsProvider;
+      StsClient stsClient = StsClient.builder().region(Region.of((String) config.get("region"))).build();
+      AssumeRoleRequest assumeRoleRequest = AssumeRoleRequest.builder()
+              .roleArn((String) config.get("role_arn"))
+              .roleSessionName("cloudwatch_exporter").build();
+      return StsAssumeRoleCredentialsProvider.builder()
+              .stsClient(stsClient)
+              .refreshRequest(assumeRoleRequest).build();
     }
 
     private List<ResourceTagMapping> getResourceTagMappings(MetricRule rule, ResourceGroupsTaggingApiClient taggingClient) {