auto merge of #16903 : mahkoh/rust/move_items_unwrap, r=aturon

Closes #16879
This commit is contained in:
bors 2014-09-08 13:46:15 +00:00
commit 6f34760e41
1 changed files with 25 additions and 0 deletions

View File

@ -1618,6 +1618,19 @@ pub struct MoveItems<T> {
iter: Items<'static, T>
}
impl<T> MoveItems<T> {
#[inline]
/// Drops all items that have not yet been moved and returns the empty vector.
pub fn unwrap(mut self) -> Vec<T> {
unsafe {
for _x in self { }
let MoveItems { allocation, cap, iter: _iter } = self;
mem::forget(self);
Vec { ptr: allocation, cap: cap, len: 0 }
}
}
}
impl<T> Iterator<T> for MoveItems<T> {
#[inline]
fn next<'a>(&'a mut self) -> Option<T> {
@ -2016,6 +2029,18 @@ mod tests {
assert_eq!(vec.swap_remove(0), None);
}
#[test]
fn test_move_iter_unwrap() {
let mut vec: Vec<uint> = Vec::with_capacity(7);
vec.push(1);
vec.push(2);
let ptr = vec.as_ptr();
vec = vec.move_iter().unwrap();
assert_eq!(vec.as_ptr(), ptr);
assert_eq!(vec.capacity(), 7);
assert_eq!(vec.len(), 0);
}
#[bench]
fn bench_new(b: &mut Bencher) {
b.iter(|| {