Skip to content

Commit 5924946

Browse files
authored
fix(remote): using customized retry predicate func if provided (#2135)
1 parent ccd9b15 commit 5924946

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

pkg/v1/remote/options.go

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,8 +162,14 @@ func makeOptions(opts ...Option) (*options, error) {
162162
o.transport = transport.NewLogger(o.transport)
163163
}
164164

165+
// Using customized retry predicate if provided, and fallback to default if not.
166+
predicate := o.retryPredicate
167+
if predicate == nil {
168+
predicate = defaultRetryPredicate
169+
}
170+
165171
// Wrap the transport in something that can retry network flakes.
166-
o.transport = transport.NewRetry(o.transport, transport.WithRetryPredicate(defaultRetryPredicate), transport.WithRetryStatusCodes(o.retryStatusCodes...))
172+
o.transport = transport.NewRetry(o.transport, transport.WithRetryPredicate(predicate), transport.WithRetryStatusCodes(o.retryStatusCodes...))
167173

168174
// Wrap this last to prevent transport.New from double-wrapping.
169175
if o.userAgent != "" {

0 commit comments

Comments
 (0)