Merge pull request #62 from xacrimon/i61

Fixed #61.
xacrimon · Mar 9, 2020 · eb2e397 · eb2e397
2 parents 749c1da + af348f4
commit eb2e397
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/mapref/entry.rs b/src/mapref/entry.rs
@@ -68,6 +68,17 @@ impl<'a, K: Eq + Hash, V, S: BuildHasher> Entry<'a, K, V, S> {
             Entry::Vacant(entry) => entry.insert(value()),
         }
     }
+
+    #[inline]
+    pub fn or_try_insert_with<E>(
+        self,
+        value: impl FnOnce() -> Result<V, E>,
+    ) -> Result<RefMut<'a, K, V, S>, E> {
+        match self {
+            Entry::Occupied(entry) => Ok(entry.into_ref()),
+            Entry::Vacant(entry) => Ok(entry.insert(value()?)),
+        }
+    }
 }
 
 pub struct VacantEntry<'a, K, V, S> {