diff options
Diffstat (limited to 'backend')
| -rw-r--r-- | backend/admin.go | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/backend/admin.go b/backend/admin.go index e5f5696..84b5c0f 100644 --- a/backend/admin.go +++ b/backend/admin.go @@ -56,10 +56,15 @@ func handleTimeout(ctx context.Context) bool { } v.since = time.Now() GetLogger(ctx).Warn("rate limiting IP", "ip", ip, "duration", dur().String()) - go func(v *to) { + go func(v *to, ip string) { time.Sleep(3 * time.Hour) v.n = max(v.n-4, 0) - }(v) + if v.n == 0 { + timeouts.mu.Lock() + defer timeouts.mu.Unlock() + delete(timeouts.tos, ip) + } + }(v, ip) return true } @@ -70,13 +75,7 @@ func resetTimeout(ctx context.Context) { timeouts.mu.Lock() defer timeouts.mu.Unlock() - - v, ok := timeouts.tos[ip] - if !ok { - return - } - v.n = 0 - v.since = time.Unix(0, 0) + delete(timeouts.tos, ip) } func HandleAdmin(r *chi.Mux) { |