Merge pull request #488 from getAlby/rate-limit-response

Use proper JSON response for rate limit error
getAlby · Jun 7, 2024 · 8fe603c · 8fe603c
2 parents badb700 + 0b8c8e0
commit 8fe603c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lib/responses/errors.go b/lib/responses/errors.go
@@ -99,6 +99,13 @@ var AccountDeactivatedError = ErrorResponse{
 	HttpStatusCode: 401,
 }
 
+var TooManyRequestsError = ErrorResponse{
+	Error:          true,
+	Code:           1,
+	Message:        "Too many requests",
+	HttpStatusCode: 429,
+}
+
 func HTTPErrorHandler(err error, c echo.Context) {
 	if c.Response().Committed {
 		return

diff --git a/lib/transport/echo.go b/lib/transport/echo.go
@@ -77,6 +77,10 @@ func CreateRateLimitMiddleware(requestsPerSecond int, burst int) echo.Middleware
 
 			return id, nil
 		},
+		ErrorHandler: func(context echo.Context, err error) error {
+			res := responses.TooManyRequestsError
+			return context.JSON(res.HttpStatusCode, res)
+		},
 	}
 
 	return middleware.RateLimiterWithConfig(config)