Add WithErrorHandler client option

xanzy · Dec 9, 2022 · 858e600 · 858e600
1 parent ac5896f
commit 858e600
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/client_options.go b/client_options.go
@@ -118,6 +118,14 @@ func WithResponseLogHook(hook retryablehttp.ResponseLogHook) 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
+	}
+}
+
 // WithoutRetries disables the default retry logic.
 func WithoutRetries() ClientOptionFunc {
 	return func(c *Client) error {