Skip to content

Commit 2c02aab

Browse files
committed
Add cell#with_mut_ref for handling mutable references to the content.
1 parent 717ed51 commit 2c02aab

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

src/libcore/cell.rs

+26
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,14 @@ pub impl<T> Cell<T> {
7373
self.put_back(v);
7474
r
7575
}
76+
77+
// Calls a closure with a mutable reference to the value.
78+
fn with_mut_ref<R>(&self, op: &fn(v: &mut T) -> R) -> R {
79+
let mut v = self.take();
80+
let r = op(&mut v);
81+
self.put_back(v);
82+
r
83+
}
7684
}
7785

7886
#[test]
@@ -101,3 +109,21 @@ fn test_put_back_non_empty() {
101109
let value_cell = Cell(~10);
102110
value_cell.put_back(~20);
103111
}
112+
113+
#[test]
114+
fn test_with_ref() {
115+
let good = 6;
116+
let c = Cell(~[1, 2, 3, 4, 5, 6]);
117+
let l = do c.with_ref() |v| { v.len() };
118+
assert!(l == good);
119+
}
120+
121+
#[test]
122+
fn test_with_mut_ref() {
123+
let good = ~[1, 2, 3];
124+
let mut v = ~[1, 2];
125+
let c = Cell(v);
126+
do c.with_mut_ref() |v| { v.push(3); }
127+
let v = c.take();
128+
assert!(v == good);
129+
}

0 commit comments

Comments
 (0)