diff options
| author | Jason Cameron <git@jasoncameron.dev> | 2025-03-29 23:24:06 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-03-29 23:24:06 -0400 |
| commit | 0f41388bd78668ceae6d5c12b05868bd0ca8fd1f (patch) | |
| tree | e60f01da1b446e8ec98aa66a70da2969a7f4e9ad /decaymap/decaymap.go | |
| parent | 052316ba25c86d5dcc870cd5b4c045ea67110562 (diff) | |
| download | anubis-0f41388bd78668ceae6d5c12b05868bd0ca8fd1f.tar.xz anubis-0f41388bd78668ceae6d5c12b05868bd0ca8fd1f.zip | |
Add periodic cleanup job for DecayMap (#8) (#158)
* Add periodic cleanup job for DecayMap
see https://github.com/TecharoHQ/anubis/issues/8
* Refactor: Improve DecayMap cleanup tests and add Len method
- Refactored DecayMap cleanup tests to use the new Len method
for more precise assertions.
- Added a Len method to DecayMap to retrieve the number of
entries.
- Simplified conditional checks in Get method.
* chore(changelog): add entry
* fix(tests): Use Impl.expire for decaymap cleanup
Signed-off-by: Jason Cameron <git@jasoncameron.dev>
---------
Signed-off-by: Jason Cameron <git@jasoncameron.dev>
Diffstat (limited to 'decaymap/decaymap.go')
| -rw-r--r-- | decaymap/decaymap.go | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/decaymap/decaymap.go b/decaymap/decaymap.go index 7498bb5..57ee6c2 100644 --- a/decaymap/decaymap.go +++ b/decaymap/decaymap.go @@ -85,3 +85,23 @@ func (m *Impl[K, V]) Set(key K, value V, ttl time.Duration) { expiry: time.Now().Add(ttl), } } + +// Cleanup removes all expired entries from the DecayMap. +func (m *Impl[K, V]) Cleanup() { + m.lock.Lock() + defer m.lock.Unlock() + + now := time.Now() + for key, entry := range m.data { + if now.After(entry.expiry) { + delete(m.data, key) + } + } +} + +// Len returns the number of entries in the DecayMap. +func (m *Impl[K, V]) Len() int { + m.lock.RLock() + defer m.lock.RUnlock() + return len(m.data) +} |
