Merge pull request #174 from Charliekenney23/feat/retry-on-408

add retry condition for 408s
linode · Jan 8, 2021 · f260595 · f260595
2 parents 897602d + da8be28
commit f260595
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -173,6 +173,7 @@ func (c *Client) SetRetries() *Client {
 		addRetryConditional(linodeBusyRetryCondition).
 		addRetryConditional(tooManyRequestsRetryCondition).
 		addRetryConditional(serviceUnavailableRetryCondition).
+		addRetryConditional(requestTimeoutRetryCondition).
 		SetRetryMaxWaitTime(APIRetryMaxWaitTime)
 	configureRetries(c)
 	return c

diff --git a/retries.go b/retries.go
@@ -54,6 +54,10 @@ func serviceUnavailableRetryCondition(r *resty.Response, _ error) bool {
 	return r.StatusCode() == http.StatusServiceUnavailable
 }
 
+func requestTimeoutRetryCondition(r *resty.Response, _ error) bool {
+	return r.StatusCode() == http.StatusRequestTimeout
+}
+
 func respectRetryAfter(client *resty.Client, resp *resty.Response) (time.Duration, error) {
 	retryAfterStr := resp.Header().Get(retryAfterHeaderName)
 	if retryAfterStr == "" {