add using retryablehttp with custom CheckRetry

timohirt · Feb 18, 2022 · 3c9338b · 3c9338b
1 parent 5ae261d
commit 3c9338b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/hetznerdns/api/client.go b/hetznerdns/api/client.go
@@ -2,6 +2,7 @@ package api
 
 import (
 	"bytes"
+	"context"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -47,7 +48,13 @@ func NewClient(apiToken string) (*Client, error) {
 
 func (c *Client) doHTTPRequest(apiToken string, method string, url string, body io.Reader) (*http.Response, error) {
 	retryableClient := retryablehttp.NewClient()
-	retryableClient.Backoff = retryablehttp.DefaultBackoff
+	retryableClient.CheckRetry = func(ctx context.Context, resp *http.Response, err error) (bool, error) {
+		ok, err := retryablehttp.DefaultRetryPolicy(ctx, resp, err)
+		if !ok && resp.StatusCode == http.StatusUnprocessableEntity {
+			return true, nil
+		}
+		return ok, err
+	}
 	retryableClient.RetryMax = 10
 	client := retryableClient.StandardClient()