fix(middleware): add mutex for rapid-request counter
This commit is contained in:
@@ -5,6 +5,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -220,18 +221,29 @@ func isSuspiciousUserAgent(userAgent string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
var requestCounts = make(map[string]int)
|
type rapidRequestTracker struct {
|
||||||
var lastReset = time.Now()
|
mu sync.Mutex
|
||||||
|
counts map[string]int
|
||||||
|
lastReset time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
var rapidRequests = rapidRequestTracker{
|
||||||
|
counts: make(map[string]int),
|
||||||
|
lastReset: time.Now(),
|
||||||
|
}
|
||||||
|
|
||||||
func isRapidRequest(ip string) bool {
|
func isRapidRequest(ip string) bool {
|
||||||
|
rapidRequests.mu.Lock()
|
||||||
|
defer rapidRequests.mu.Unlock()
|
||||||
|
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
|
|
||||||
if now.Sub(lastReset) > time.Minute {
|
if now.Sub(rapidRequests.lastReset) > time.Minute {
|
||||||
requestCounts = make(map[string]int)
|
rapidRequests.counts = make(map[string]int)
|
||||||
lastReset = now
|
rapidRequests.lastReset = now
|
||||||
}
|
}
|
||||||
|
|
||||||
requestCounts[ip]++
|
rapidRequests.counts[ip]++
|
||||||
|
|
||||||
return requestCounts[ip] > 100
|
return rapidRequests.counts[ip] > 100
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user