Use error names instead of codes

databricks · Nov 1, 2024 · 40a1b2a · 40a1b2a
1 parent e9314bd
commit 40a1b2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/httpclient/errors.go b/httpclient/errors.go
@@ -48,10 +48,10 @@ func DefaultErrorMapper(ctx context.Context, resp common.ResponseWrapper) error
 func DefaultErrorRetriable(ctx context.Context, err error) bool {
 	var httpError *HttpError
 	if errors.As(err, &httpError) {
-		if httpError.StatusCode == 429 {
+		if httpError.StatusCode == http.StatusTooManyRequests {
 			return true
 		}
-		if httpError.StatusCode == 504 {
+		if httpError.StatusCode == http.StatusGatewayTimeout {
 			return true
 		}
 	}