set timeout for ecs ram role

aliyun · Aug 13, 2024 · a97e863 · a97e863
1 parent c2314f2
commit a97e863
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/sdk/auth/credentials/credentials.go b/sdk/auth/credentials/credentials.go
@@ -513,7 +513,9 @@ func (provider *ECSRAMRoleCredentialsProvider) getRoleName() (roleName string, e
 		err = fmt.Errorf("get role name failed: %s", err.Error())
 		return
 	}
-	httpClient := &http.Client{}
+	httpClient := &http.Client{
+		Timeout: 5 * time.Second,
+	}
 	httpResponse, err := hookDo(httpClient.Do)(httpRequest)
 	if err != nil {
 		err = fmt.Errorf("get role name failed: %s", err.Error())
@@ -551,7 +553,9 @@ func (provider *ECSRAMRoleCredentialsProvider) getCredentials() (sessionCredenti
 		err = fmt.Errorf("refresh Ecs sts token err: %s", err.Error())
 		return
 	}
-	httpClient := &http.Client{}
+	httpClient := &http.Client{
+		Timeout: 5 * time.Second,
+	}
 	httpResponse, err := hookDo(httpClient.Do)(httpRequest)
 	if err != nil {
 		err = fmt.Errorf("refresh Ecs sts token err: %s", err.Error())