Add Entry get_mut and into_mut

cspr-rad · Apr 11, 2024 · 161159e · 161159e
1 parent 06ec8ec
commit 161159e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/transaction.rs b/src/transaction.rs
@@ -492,6 +492,22 @@ impl<'a, V> Entry<'a, V> {
         }
     }
 
+    #[inline]
+    pub fn get_mut(&mut self) -> Option<&mut V> {
+        match self {
+            Entry::Occupied(OccupiedEntry { leaf }) => Some(&mut leaf.value),
+            _ => None,
+        }
+    }
+
+    #[inline]
+    pub fn into_mut(self) -> Option<&'a mut V> {
+        match self {
+            Entry::Occupied(OccupiedEntry { leaf }) => Some(&mut leaf.value),
+            _ => None,
+        }
+    }
+
     #[inline]
     pub fn or_insert(self, value: V) -> &'a mut V {
         self.or_insert_with(|| value)