Add cell#with_mut_ref for handling mutable references to the content.
This commit is contained in:
parent
717ed51f12
commit
2c02aab758
@ -73,6 +73,14 @@ fn with_ref<R>(&self, op: &fn(v: &T) -> R) -> R {
|
||||
self.put_back(v);
|
||||
r
|
||||
}
|
||||
|
||||
// Calls a closure with a mutable reference to the value.
|
||||
fn with_mut_ref<R>(&self, op: &fn(v: &mut T) -> R) -> R {
|
||||
let mut v = self.take();
|
||||
let r = op(&mut v);
|
||||
self.put_back(v);
|
||||
r
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
@ -101,3 +109,21 @@ fn test_put_back_non_empty() {
|
||||
let value_cell = Cell(~10);
|
||||
value_cell.put_back(~20);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_with_ref() {
|
||||
let good = 6;
|
||||
let c = Cell(~[1, 2, 3, 4, 5, 6]);
|
||||
let l = do c.with_ref() |v| { v.len() };
|
||||
assert!(l == good);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_with_mut_ref() {
|
||||
let good = ~[1, 2, 3];
|
||||
let mut v = ~[1, 2];
|
||||
let c = Cell(v);
|
||||
do c.with_mut_ref() |v| { v.push(3); }
|
||||
let v = c.take();
|
||||
assert!(v == good);
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user