Merge pull request #302 from Scalingo/feat/301/handle_too_many_requests

feat(status code): handle too many requests exception
Scalingo · Feb 6, 2023 · 54f56e9 · 54f56e9
2 parents a82d55a + d7bef4c
commit 54f56e9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,7 @@
 # Changelog
 
 ## To Be Released
+* feat(status code): handle too many requests exception
 
 ## 6.1.0
 

diff --git a/http/errors.go b/http/errors.go
@@ -37,6 +37,11 @@ type (
 		Errors map[string][]string `json:"errors"`
 	}
 
+	TooManyRequestsError struct {
+		ErrMessage string `json:"error"`
+		Code       string `json:"code"`
+	}
+
 	APIError struct {
 		Error string `json:"error"`
 	}
@@ -94,6 +99,10 @@ func (err UnprocessableEntity) Error() string {
 	return strings.Join(errArray, "\n")
 }
 
+func (err TooManyRequestsError) Error() string {
+	return fmt.Sprintf("429 Too Many Requests → %v", err.ErrMessage)
+}
+
 func (err ForbiddenError) Error() string {
 	return fmt.Sprintf("Request forbidden (403): %v", err.Err)
 }
@@ -141,6 +150,13 @@ func NewRequestFailedError(res *http.Response, req *APIRequest) error {
 			return err
 		}
 		return &RequestFailedError{Code: res.StatusCode, APIError: unprocessableError, Req: req}
+	case 429:
+		var tooManyRequestsError TooManyRequestsError
+		err := parseJSON(res, &tooManyRequestsError)
+		if err != nil {
+			return err
+		}
+		return &RequestFailedError{Code: res.StatusCode, APIError: tooManyRequestsError, Req: req}
 	case 500:
 		return &RequestFailedError{Code: res.StatusCode, APIError: errgo.New("server internal error - our team has been notified"), Req: req}
 	case 503: