From 444a4ddf9aa984865db51c702dc78e505d2954f7 Mon Sep 17 00:00:00 2001
From: Ethan Katnic <ekatnic@amazon.com>
Date: Tue, 27 Aug 2024 10:19:27 -0700
Subject: [PATCH] Add https check

---
 .../multilang/auth/KclStsAssumeRoleCredentialsProvider.java    | 3 +++
 1 file changed, 3 insertions(+)

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 ce9de0f5..65c096ae 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
@@ -40,6 +40,9 @@ public class KclStsAssumeRoleCredentialsProvider implements AwsCredentialsProvid
 
     @Override
     public void acceptEndpoint(String serviceEndpoint, String signingRegion) {
+        if (!serviceEndpoint.startsWith("http://") && !serviceEndpoint.startsWith("https://")) {
+            serviceEndpoint = "https://" + serviceEndpoint;
+        }
         stsClientBuilder.endpointOverride(URI.create(serviceEndpoint));
         stsClientBuilder.region(Region.of(signingRegion));
     }