Add DeleteIf to CopyOnWriteMap

openziti · Nov 12, 2024 · 1603635 · 1603635
1 parent 2ee5fb4
commit 1603635
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/concurrenz/copy_on_write_map.go b/concurrenz/copy_on_write_map.go
@@ -63,3 +63,21 @@ func (self *CopyOnWriteMap[K, V]) Clear() {
 	defer self.lock.Unlock()
 	self.value.Store(map[K]V{})
 }
+
+func (self *CopyOnWriteMap[K, V]) DeleteIf(f func(key K, val V) bool) bool {
+	self.lock.Lock()
+	defer self.lock.Unlock()
+
+	matched := false
+	var current = self.value.Load()
+	mapCopy := map[K]V{}
+	for k, v := range current {
+		if !f(k, v) {
+			mapCopy[k] = v
+		} else {
+			matched = true
+		}
+	}
+	self.value.Store(mapCopy)
+	return matched
+}