Add retry in 502,503 and 520

doitintl · May 21, 2024 · ba62018 · ba62018
1 parent 69dc4a8
commit ba62018
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/internal/provider/client.go b/internal/provider/client.go
@@ -112,9 +112,14 @@ func (c *ClientTest) doRequest(req *http.Request) ([]byte, error) {
 			log.Println(errRetry)
 			return fmt.Errorf("no response")
 		}
-		if res.StatusCode == http.StatusTooManyRequests {
-			return fmt.Errorf("rate limit exceeded")
+		if res.StatusCode == http.StatusTooManyRequests ||
+			res.StatusCode == http.StatusGatewayTimeout{
+			return fmt.Errorf(http.StatusText(res.StatusCode))
 		}
+		if res.StatusCode == 520 {
+			return fmt.Errorf("Cloudflare 520 Error")
+		}
+
 		err = errRetry
 		return nil
 	}