blob: 311e62cc578455443160b7ea611fb9555660db58 (
plain)
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
40
41
42
43
44
45
46
47
48
|
// SPDX-License-Identifier: GPL-2.0
//! Extensions to [`Vec`] for fallible allocations.
use alloc::{collections::TryReserveError, vec::Vec};
use core::result::Result;
/// Extensions to [`Vec`].
pub trait VecExt<T>: Sized {
/// Creates a new [`Vec`] instance with at least the given capacity.
fn try_with_capacity(capacity: usize) -> Result<Self, TryReserveError>;
/// Appends an element to the back of the [`Vec`] instance.
fn try_push(&mut self, v: T) -> Result<(), TryReserveError>;
/// Pushes clones of the elements of slice into the [`Vec`] instance.
fn try_extend_from_slice(&mut self, other: &[T]) -> Result<(), TryReserveError>
where
T: Clone;
}
impl<T> VecExt<T> for Vec<T> {
fn try_with_capacity(capacity: usize) -> Result<Self, TryReserveError> {
let mut v = Vec::new();
v.try_reserve(capacity)?;
Ok(v)
}
fn try_push(&mut self, v: T) -> Result<(), TryReserveError> {
if let Err(retry) = self.push_within_capacity(v) {
self.try_reserve(1)?;
let _ = self.push_within_capacity(retry);
}
Ok(())
}
fn try_extend_from_slice(&mut self, other: &[T]) -> Result<(), TryReserveError>
where
T: Clone,
{
self.try_reserve(other.len())?;
for item in other {
self.try_push(item.clone())?;
}
Ok(())
}
}
|
