fix signing middleware not found bug

aws · Dec 28, 2023 · bb050be · bb050be
1 parent 71e6af0
commit bb050be
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/aws/retry/middleware.go b/aws/retry/middleware.go
@@ -328,9 +328,18 @@ func AddRetryMiddlewares(stack *smithymiddle.Stack, options AddRetryMiddlewaresO
 		middleware.LogAttempts = options.LogRetryAttempts
 	})
 
-	if err := stack.Finalize.Insert(attempt, "Signing", smithymiddle.Before); err != nil {
-		return err
+	// index retry to before signing, if signing exists
+	_, ok := stack.Finalize.Get("Signing")
+	if !ok {
+		if err := stack.Finalize.Add(attempt, smithymiddle.After); err != nil {
+			return err
+		}
+	} else {
+		if err := stack.Finalize.Insert(attempt, "Signing", smithymiddle.Before); err != nil {
+			return err
+		}
 	}
+
 	if err := stack.Finalize.Insert(&MetricsHeader{}, attempt.ID(), smithymiddle.After); err != nil {
 		return err
 	}