feat: OIDCRoleArnProvider support endpoint (#245)

TencentCloud · Dec 14, 2023 · f06b539 · f06b539
1 parent c8385f9
commit f06b539
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tencentcloud/common/oidc_role_arn_provider.go b/tencentcloud/common/oidc_role_arn_provider.go
@@ -20,6 +20,7 @@ type OIDCRoleArnProvider struct {
 	roleArn          string
 	roleSessionName  string
 	durationSeconds  int64
+	Endpoint         string
 }
 
 type oidcStsRsp struct {
@@ -91,8 +92,12 @@ func (r *OIDCRoleArnProvider) GetCredential() (CredentialIface, error) {
 	if r.durationSeconds > 43200 || r.durationSeconds <= 0 {
 		return nil, tcerr.NewTencentCloudSDKError(creErr, "AssumeRoleWithWebIdentity durationSeconds should be in the range of 0~43200s", "")
 	}
+	providerEndpoint := r.Endpoint
+	if providerEndpoint == "" {
+		providerEndpoint = endpoint
+	}
 	cpf := profile.NewClientProfile()
-	cpf.HttpProfile.Endpoint = endpoint
+	cpf.HttpProfile.Endpoint = providerEndpoint
 	cpf.HttpProfile.ReqMethod = "POST"
 
 	client := NewCommonClient(nil, r.region, cpf)