From 161159e810b6b2596c1bd9aca3fef6a92f9e5ac8 Mon Sep 17 00:00:00 2001
From: Avi Dessauer <avi.the.coder@gmail.com>
Date: Thu, 11 Apr 2024 15:12:51 +0300
Subject: [PATCH] Add Entry get_mut and into_mut

---
 src/transaction.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/transaction.rs b/src/transaction.rs
index d801cbc..d4cfc59 100644
--- 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)