Add WithErrorHandler client option

xanzy · Dec 12, 2022 · db9dda5 · db9dda5
1 parent ac5896f
commit db9dda5
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 {