Retry on HTTP/2 internal errors. (#652)

digitalocean · Nov 7, 2023 · 2036ebb · 2036ebb
1 parent 39cdba9
commit 2036ebb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/godo.go b/godo.go
@@ -348,6 +348,16 @@ func New(httpClient *http.Client, opts ...ClientOpt) (*Client, error) {
 			return resp, err
 		}
 
+		retryableClient.CheckRetry = func(ctx context.Context, resp *http.Response, err error) (bool, error) {
+			// In addition to the default retry policy, we also retry HTTP/2 INTERNAL_ERROR errors.
+			// See: https://github.com/golang/go/issues/51323
+			if err != nil && strings.Contains(err.Error(), "INTERNAL_ERROR") && strings.Contains(reflect.TypeOf(err).String(), "http2") {
+				return true, nil
+			}
+
+			return retryablehttp.DefaultRetryPolicy(ctx, resp, err)
+		}
+
 		var source *oauth2.Transport
 		if _, ok := c.HTTPClient.Transport.(*oauth2.Transport); ok {
 			source = c.HTTPClient.Transport.(*oauth2.Transport)