mirror of
https://github.com/containers/podman.git
synced 2025-09-27 16:54:42 +08:00
9ac0ebb079
* Make sure that all vendored dependencies are in sync with the code and the vendor.conf by running `make vendor` with a follow-up status check of the git tree. * Vendor ginkgo and gomega to include the test dependencies. Signed-off-by: Chris Evic <cevich@redhat.com> Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
59 lines
975 B
Go
59 lines
975 B
Go
package ratelimiter
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
)
|
|
|
|
const GC_SIZE int = 100
|
|
|
|
type Memory struct {
|
|
store map[string]LeakyBucket
|
|
lastGCCollected time.Time
|
|
}
|
|
|
|
func NewMemory() *Memory {
|
|
m := new(Memory)
|
|
m.store = make(map[string]LeakyBucket)
|
|
m.lastGCCollected = time.Now()
|
|
return m
|
|
}
|
|
|
|
func (m *Memory) GetBucketFor(key string) (*LeakyBucket, error) {
|
|
|
|
bucket, ok := m.store[key]
|
|
if !ok {
|
|
return nil, errors.New("miss")
|
|
}
|
|
|
|
return &bucket, nil
|
|
}
|
|
|
|
func (m *Memory) SetBucketFor(key string, bucket LeakyBucket) error {
|
|
|
|
if len(m.store) > GC_SIZE {
|
|
m.GarbageCollect()
|
|
}
|
|
|
|
m.store[key] = bucket
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Memory) GarbageCollect() {
|
|
now := time.Now()
|
|
|
|
// rate limit GC to once per minute
|
|
if now.Add(60*time.Second).Unix() > m.lastGCCollected.Unix() {
|
|
|
|
for key, bucket := range m.store {
|
|
// if the bucket is drained, then GC
|
|
if bucket.DrainedAt().Unix() > now.Unix() {
|
|
delete(m.store, key)
|
|
}
|
|
}
|
|
|
|
m.lastGCCollected = now
|
|
}
|
|
}
|