devbennett7/goma-gateway
1
0
Fork 0
Code Pull Requests Activity

chore: add warning message from Too many access client

Browse Source
This commit is contained in:
Jonas Kaninda
2024-10-28 02:43:13 +01:00
parent 37f3ff9868
commit 95777bfd33
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/middleware/rate_limiter.go
View File

@@ -18,6 +18,7 @@ limitations under the License.
import ( import (
"encoding/json" "encoding/json"
"github.com/gorilla/mux" "github.com/gorilla/mux"
"github.com/jkaninda/goma-gateway/internal/logger"
"net/http" "net/http"
"time" "time"
) )
@@ -65,6 +66,7 @@ func (rl *RateLimiter) RateLimitMiddleware() mux.MiddlewareFunc {
rl.mu.Unlock() rl.mu.Unlock()
if client.RequestCount > rl.Requests { if client.RequestCount > rl.Requests {
logger.Warn("Too many request from IP: %s %s %s", clientID, r.URL, r.UserAgent())
w.Header().Set("Content-Type", "application/json") w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusTooManyRequests) w.WriteHeader(http.StatusTooManyRequests)
err := json.NewEncoder(w).Encode(ProxyResponseError{ err := json.NewEncoder(w).Encode(ProxyResponseError{