Add Entry::insert method

cspr-rad · Apr 11, 2024 · 3f65cb3 · 3f65cb3
1 parent 161159e
commit 3f65cb3
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/transaction.rs b/src/transaction.rs
@@ -508,6 +508,19 @@ impl<'a, V> Entry<'a, V> {
         }
     }
 
+    /// Prefer `Transaction::insert` over `Entry::insert` if you are not using any other `Entry` methods.
+    #[inline]
+    pub fn insert(self, value: V) -> &'a mut V {
+        match self {
+            Entry::Occupied(mut o) => {
+                o.insert(value);
+                o.into_mut()
+            }
+            Entry::VacantEmptyTrie(entry) => entry.insert(value),
+            Entry::Vacant(entry) => entry.insert(value),
+        }
+    }
+
     #[inline]
     pub fn or_insert(self, value: V) -> &'a mut V {
         self.or_insert_with(|| value)