Merge pull request #1593 from jokly/error-handler

Add WithErrorHandler client option
xanzy · Dec 12, 2022 · 65b39a6 · 65b39a6
2 parents 837fee3 + db9dda5
commit 65b39a6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/client_options.go b/client_options.go
@@ -94,6 +94,14 @@ func WithCustomRetryWaitMinMax(waitMin, waitMax time.Duration) ClientOptionFunc
 	}
 }
 
+// WithErrorHandler can be used to configure a custom error handler.
+func WithErrorHandler(handler retryablehttp.ErrorHandler) ClientOptionFunc {
+	return func(c *Client) error {
+		c.client.ErrorHandler = handler
+		return nil
+	}
+}
+
 // WithHTTPClient can be used to configure a custom HTTP client.
 func WithHTTPClient(httpClient *http.Client) ClientOptionFunc {
 	return func(c *Client) error {