Adding error message on connection limit met

master
Frederick F. Kautz IV 10 years ago
parent 8d1d309630
commit f596c744ed
  1. 2
      pkg/api/quota/conn_limit.go
  2. 6
      pkg/api/quota/errors.go

@ -61,6 +61,8 @@ func (c *connLimit) ServeHTTP(w http.ResponseWriter, req *http.Request) {
if c.TestAndAdd(longIP) {
defer c.Remove(longIP)
c.handler.ServeHTTP(w, req)
} else {
writeErrorResponse(w, req, ConnectionLimitExceeded, req.RequestURI)
}
}

@ -47,6 +47,7 @@ const (
BandWidthQuotaExceeded
BandWidthInsufficientToProceed
SlowDown
ConnectionLimitExceeded
)
// Golang http doesn't implement these
@ -89,6 +90,11 @@ var errorCodeResponse = map[int]Error{
Description: "Reduce your request rate.",
HTTPStatusCode: StatusTooManyRequests,
},
ConnectionLimitExceeded: {
Code: "ConnectionLimit",
Description: "Connection Limit Met",
HTTPStatusCode: StatusTooManyRequests,
},
}
// Write error response headers

Loading…
Cancel
Save