nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 12:17:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move request limiter to middleware.

Browse Source
This commit is contained in:
nyne
2025-07-09 17:00:39 +08:00
parent 0021a73951
commit b568b234c4
11 changed files with 73 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server/utils/request_limit.go
View File

@@ -6,12 +6,12 @@ import (
)
type RequestLimiter struct {
limit int
limit func() int
requestsByIP map[string]int
mu sync.Mutex
}
func NewRequestLimiter(limit int, duration time.Duration) *RequestLimiter {
func NewRequestLimiter(limit func() int, duration time.Duration) *RequestLimiter {
l := &RequestLimiter{
limit: limit,
requestsByIP: make(map[string]int),
@@ -38,7 +38,7 @@ func (rl *RequestLimiter) AllowRequest(ip string) bool {
count = 0
}
if count >= rl.limit {
if count >= rl.limit() {
return false // Exceeded request limit for this IP
}