File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -24,6 +24,7 @@ import (
24
24
"time"
25
25
26
26
awsmiddleware "github.com/aws/aws-sdk-go-v2/aws/middleware"
27
+ "github.com/aws/aws-sdk-go-v2/aws/retry"
27
28
"github.com/aws/smithy-go/middleware"
28
29
smithyhttp "github.com/aws/smithy-go/transport/http"
29
30
"github.com/prometheus/client_golang/prometheus"
@@ -100,8 +101,11 @@ func WithMiddlewares(controller string, target runtime.Object) func(stack *middl
100
101
if err := stack .Finalize .Add (getRequestMetricContextMiddleware (), middleware .Before ); err != nil {
101
102
return fmt .Errorf ("failed to add request metric context middleware: %w" , err )
102
103
}
103
- if err := stack .Finalize .Insert (getAttemptContextMiddleware (), "Retry" , middleware .After ); err != nil {
104
- return fmt .Errorf ("failed to add attempt context middleware: %w" , err )
104
+ // Check if the Default Retry Middleware exists.
105
+ if _ , ok := stack .Finalize .Get ((* retry .Attempt )(nil ).ID ()); ok {
106
+ if err := stack .Finalize .Insert (getAttemptContextMiddleware (), (* retry .Attempt )(nil ).ID (), middleware .After ); err != nil {
107
+ return fmt .Errorf ("failed to add attempt context middleware: %w" , err )
108
+ }
105
109
}
106
110
return stack .Finalize .Add (getRecordAWSPermissionsIssueMiddleware (target ), middleware .After )
107
111
}
You can’t perform that action at this time.
0 commit comments