Add borrow_with to LazyCell

CONTRIBUTORS.md

@@ -3,5 +3,6 @@ alphabetical order.
 
 * [Carl Lerche](https://github.com/carllerche)
 * [Nikita Pekin](https://github.com/indiv0)
+* [J/A](https://github.com/archer884)
 
 [lazycell]: https://github.com/indiv0/lazycell

src/lib.rs

@@ -85,6 +85,20 @@ impl<T> LazyCell<T> {
         unsafe { &*self.inner.get() }.as_ref()
     }
 
+    /// Borrows the contents of this lazy cell for the duration of the cell
+    /// itself.
+    /// 
+    /// If the cell has not yet been filled, the cell is first filled using the
+    /// function provided.
+    pub fn borrow_with<F: FnOnce() -> T>(&self, f: F) -> &T {
+        let mut slot = unsafe { &mut *self.inner.get() };
+        if !slot.is_some() {
+            *slot = Some(f());
+        }
+
+        slot.as_ref().unwrap()
+    }
+
     /// Consumes this `LazyCell`, returning the underlying value.
     pub fn into_inner(self) -> Option<T> {
         unsafe { self.inner.into_inner() }
@@ -186,6 +200,33 @@ mod tests {
         assert_eq!(lazycell.fill(1), Err(1));
     }
 
+    #[test]
+    fn test_borrow_with() {
+        let lazycell = LazyCell::new();
+
+        let value = lazycell.borrow_with(|| 1);
+        assert_eq!(&1, value);
+    }
+
+    #[test]
+    fn test_borrow_with_already_filled() {
+        let lazycell = LazyCell::new();
+        lazycell.fill(1).unwrap();
+
+        let value = lazycell.borrow_with(|| 1);
+        assert_eq!(&1, value);
+    }
+
+    #[test]
+    fn test_borrow_with_not_called_when_filled() {
+        let lazycell = LazyCell::new();
+
+        lazycell.fill(1).unwrap();
+
+        let value = lazycell.borrow_with(|| 2);
+        assert_eq!(&1, value);
+    }
+
     #[test]
     fn test_into_inner() {
         let lazycell = LazyCell::new();