implement CopyTo func

knbr13 · May 29, 2024 · c751660 · c751660
1 parent 03a3286
commit c751660
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lfu_cache.go b/lfu_cache.go
@@ -143,6 +143,17 @@ func (src *LFUCache[K, V]) TransferTo(dst *LFUCache[K, V]) {
 	}
 }
 
+func (src *LFUCache[K, V]) CopyTo(dst *LFUCache[K, V]) {
+	src.mu.RLock()
+	defer src.mu.RUnlock()
+
+	for k, v := range src.m {
+		if v.Value.(*lfuItem[K, V]).expireAt == nil || !v.Value.(*lfuItem[K, V]).expireAt.Before(time.Now()) {
+			dst.Set(k, v.Value.(*lfuItem[K, V]).value)
+		}
+	}
+}
+
 func (l *LFUCache[K, V]) Delete(k K) {
 	l.mu.Lock()
 	defer l.mu.Unlock()