highperfocused/khatru
1
0
Fork 0
mirror of https://github.com/fiatjaf/khatru.git synced 2025-03-28 18:51:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

policies: shoehorn an iterator in a xsync map Range().

Browse Source
This commit is contained in:
fiatjaf 2024-09-16 15:59:08 -03:00
parent eb92c2aac5
commit 635e91f360
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
policies/helpers.go
View File

@ -1,6 +1,7 @@
package policies package policies
import ( import (
"iter"
"sync/atomic" "sync/atomic"
"time" "time"
@ -18,13 +19,12 @@ func startRateLimitSystem[K comparable](
go func() { go func() {
for { for {
time.Sleep(interval) time.Sleep(interval)
negativeBuckets.Range(func(key K, bucket *atomic.Int32) bool { for key, bucket := range iter.Seq2[K, *atomic.Int32](negativeBuckets.Range) {
newv := bucket.Add(int32(-tokensPerInterval)) newv := bucket.Add(int32(-tokensPerInterval))
if newv <= 0 { if newv <= 0 {
negativeBuckets.Delete(key) negativeBuckets.Delete(key)
} }
return true }
})
} }
}() }()