return HTTP429 status code from HTTP API

hashicorp · Jun 28, 2022 · d94a613 · d94a613
1 parent 9930183
commit d94a613
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/command/agent/http.go b/command/agent/http.go
@@ -528,6 +528,9 @@ func errCodeFromHandler(err error) (int, string) {
 		} else if strings.HasSuffix(errMsg, structs.ErrJobRegistrationDisabled.Error()) {
 			errMsg = structs.ErrJobRegistrationDisabled.Error()
 			code = 403
+		} else if strings.HasSuffix(errMsg, structs.ErrTooManyRequests.Error()) {
+			errMsg = structs.ErrTooManyRequests.Error()
+			code = 429
 		}
 	}
 
@@ -570,6 +573,9 @@ func (s *HTTPServer) wrap(handler func(resp http.ResponseWriter, req *http.Reque
 				} else if strings.HasSuffix(errMsg, structs.ErrIncompatibleFiltering.Error()) {
 					errMsg = structs.ErrIncompatibleFiltering.Error()
 					code = 400
+				} else if strings.HasSuffix(errMsg, structs.ErrTooManyRequests.Error()) {
+					errMsg = structs.ErrTooManyRequests.Error()
+					code = 429
 				}
 			}