impl Purge func

knbr13 · May 29, 2024 · 4c929f4 · 4c929f4
1 parent 60b97a9
commit 4c929f4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lfu_cache.go b/lfu_cache.go
@@ -169,6 +169,14 @@ func (l *LFUCache[K, V]) Keys() []K {
 	return keys
 }
 
+func (l *LFUCache[K, V]) Purge() {
+	l.mu.Lock()
+	defer l.mu.Unlock()
+
+	l.m = make(map[K]*list.Element)
+	l.evictionList.Init()
+}
+
 func (l *LFUCache[K, V]) Count() int {
 	l.mu.RLock()
 	defer l.mu.RUnlock()