add retry to avoid fetching token failed with some transient issues

awslabs · Jul 16, 2024 · 782f15b · 782f15b
1 parent 958d6fc
commit 782f15b
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/nodeadm/internal/aws/ecr/ecr.go b/nodeadm/internal/aws/ecr/ecr.go
@@ -19,12 +19,15 @@ func GetAuthorizationToken(awsRegion string) (string, error) {
 		return "", err
 	}
 	ecrClient := ecr.NewFromConfig(awsConfig)
-	token, err := ecrClient.GetAuthorizationToken(context.Background(), &ecr.GetAuthorizationTokenInput{})
-	if err != nil {
-		return "", err
+
+	for i := 0; i < 3; i++ {
+		token, err := ecrClient.GetAuthorizationToken(context.Background(), &ecr.GetAuthorizationTokenInput{})
+		if err == nil {
+			authData := token.AuthorizationData[0].AuthorizationToken
+			return *authData, nil
+		}
 	}
-	authData := token.AuthorizationData[0].AuthorizationToken
-	return *authData, nil
+	return "", err
 }
 
 func (r *ECRRegistry) GetSandboxImage() string {