Add LazyMap::{remove, clear}

a1phyr · Jan 8, 2024 · ee4323f · ee4323f
1 parent 7f159e7
commit ee4323f
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/src/sync.rs b/src/sync.rs
@@ -910,6 +910,10 @@ impl<K, V, S, F> LazyMap<K, V, S, F> {
             init: f,
         }
     }
+
+    pub fn clear(&mut self) {
+        self.map.clear();
+    }
 }
 
 impl<K, V, S, F> LazyMap<K, V, S, F>
@@ -957,6 +961,19 @@ where
     }
 }
 
+impl<K, V, S, F> LazyMap<K, V, S, F>
+where
+    K: Eq + Hash,
+    S: BuildHasher,
+{
+    pub fn remove<Q>(&mut self, key: &Q) -> Option<V>
+    where
+        Q: Hash + Equivalent<K> + ?Sized,
+    {
+        self.map.remove(key)
+    }
+}
+
 impl<K, V, S, F, Q> core::ops::Index<&Q> for LazyMap<K, V, S, F>
 where
     K: Eq + Hash,

diff --git a/src/unsync.rs b/src/unsync.rs
@@ -475,6 +475,10 @@ impl<K, V, S, F> LazyMap<K, V, S, F> {
             init: f,
         }
     }
+
+    pub fn clear(&mut self) {
+        self.map.clear();
+    }
 }
 
 impl<K, V, S, F> LazyMap<K, V, S, F>
@@ -522,6 +526,19 @@ where
     }
 }
 
+impl<K, V, S, F> LazyMap<K, V, S, F>
+where
+    K: Eq + Hash,
+    S: BuildHasher,
+{
+    pub fn remove<Q>(&mut self, key: &Q) -> Option<V>
+    where
+        Q: Hash + Equivalent<K> + ?Sized,
+    {
+        self.map.remove(key)
+    }
+}
+
 /// Creates a `LazyMap` that fills all values with `V::default()`.
 impl<K, V: Default, S: Default> Default for LazyMap<K, V, S> {
     fn default() -> Self {