-
-
Notifications
You must be signed in to change notification settings - Fork 45
/
managed_vec.rs
39 lines (34 loc) · 1.11 KB
/
managed_vec.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
use managed::ManagedSlice;
/// Error value indicating insufficient capacity.
#[derive(Debug, Clone, Copy, Eq, Ord, PartialEq, PartialOrd)]
pub struct CapacityError<Element>(pub Element);
/// Wraps a ManagedSlice in a vec-like interface.
pub struct ManagedVec<'a, 'b, T> {
buf: &'b mut ManagedSlice<'a, T>,
len: usize,
}
impl<'a, 'b, T> ManagedVec<'a, 'b, T> {
pub fn new_with_idx(buf: &'b mut ManagedSlice<'a, T>, len: usize) -> Self {
ManagedVec { buf, len }
}
pub fn push(&mut self, value: T) -> Result<(), CapacityError<T>> {
if self.len < self.buf.len() {
self.buf[self.len] = value;
self.len += 1;
Ok(())
} else {
match &mut self.buf {
ManagedSlice::Borrowed(_) => Err(CapacityError(value)),
#[cfg(feature = "alloc")]
ManagedSlice::Owned(buf) => {
buf.push(value);
Ok(())
}
}
}
}
#[cfg(feature = "trace-pkt")]
pub fn as_slice<'c: 'b>(&'c self) -> &'b [T] {
&self.buf[..self.len]
}
}