Noratrieb/vechonk
1
0
Fork 0
mirror of https://github.com/Noratrieb/vechonk.git synced 2026-03-16 06:06:05 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

use &mut T again

Browse source
This commit is contained in:
nora 2021-12-22 21:34:38 +01:00
parent e5b7c40028
commit 9a0d125d8d
3 changed files with 37 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/iter.rs
View file

@ -1,4 +1,4 @@
use crate::{MutGuard, RawVechonk, Vechonk};
use crate::{RawVechonk, Vechonk};
use alloc::boxed::Box;
use core::marker::PhantomData;
use core::mem;
@ -68,19 +68,20 @@ impl<'a, T: ?Sized> IterMut<'a, T> {
}
impl<'a, T: ?Sized> Iterator for IterMut<'a, T> {
type Item = MutGuard<T>;
type Item = &'a mut T;
fn next(&mut self) -> Option<Self::Item> {
if self.current_index == self.raw.len {
return None;
}
let old_index = self.current_index;
// SAFETY: We just did a bounds check above
let ptr = unsafe { self.raw.get_unchecked_ptr(self.current_index) };
self.current_index += 1;
// SAFETY: We did a bounds check above, and taken `&mut Vechonk`
unsafe { Some(MutGuard::new(self.raw.copy(), old_index)) }
// SAFETY: We rely on `get_unchecked_ptr` returning a valid pointer, which is does, see its SAFETY comments
unsafe { Some(&mut *ptr) }
}
fn size_hint(&self) -> (usize, Option<usize>) {