From fec8b64975f952e2fa5fa8478b09dced87f0fa7b Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Fri, 6 Mar 2026 04:43:03 +0800 Subject: internal/lru: Split --- internal/lru/remove.go | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 internal/lru/remove.go (limited to 'internal/lru/remove.go') diff --git a/internal/lru/remove.go b/internal/lru/remove.go new file mode 100644 index 00000000..79735edf --- /dev/null +++ b/internal/lru/remove.go @@ -0,0 +1,31 @@ +package lru + +import "container/list" + +// Remove deletes key from the cache. +func (cache *Cache[K, V]) Remove(key K) (V, bool) { + elem, ok := cache.items[key] + if !ok { + var zero V + + return zero, false + } + + ent := cache.removeElem(elem) + + return ent.value, true +} + +func (cache *Cache[K, V]) removeElem(elem *list.Element) *entry[K, V] { + //nolint:forcetypeassert + ent := elem.Value.(*entry[K, V]) + cache.lru.Remove(elem) + delete(cache.items, ent.key) + + cache.weight -= ent.weight + if cache.onEvict != nil { + cache.onEvict(ent.key, ent.value) + } + + return ent +} -- cgit v1.3.1-10-gc9f91