add c.Keys

knbr13 · May 22, 2024 · 64cb5d2 · 64cb5d2
1 parent 7815019
commit 64cb5d2
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/lru_cache.go b/lru_cache.go
@@ -141,6 +141,21 @@ func (src *LRUCache[K, V]) CopyTo(dst *LRUCache[K, V]) {
 	}
 }
 
+func (c *LRUCache[K, V]) Keys() []K {
+	c.mu.RLock()
+	defer c.mu.RUnlock()
+
+	keys := make([]K, 0, c.Count())
+
+	for k, v := range c.m {
+		if v.Value.(*lruItem[K, V]).expireAt == nil || !v.Value.(*lruItem[K, V]).expireAt.Before(time.Now()) {
+			keys = append(keys, k)
+		}
+	}
+
+	return keys
+}
+
 func (c *LRUCache[K, V]) AllKeys() []K {
 	c.mu.RLock()
 	defer c.mu.RUnlock()