ps-buffer 0.1.0-21

aligned heap buffer
Documentation 
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
49
50
51
52

use crate::{Buffer, BufferError};

pub trait ByteIteratorIntoBuffer {
    /// Collects this [`Iterator`] into a [`Buffer`].
    /// # Errors
    /// [`BufferError`] is returned if an allocation or reallocation fails.
    fn into_buffer(self) -> Result<Buffer, BufferError>;
}

impl<T: Iterator<Item = u8>> ByteIteratorIntoBuffer for T {
    fn into_buffer(self) -> Result<Buffer, BufferError> {
        let mut buffer = Buffer::with_capacity(self.size_hint().0)?;

        for byte in self {
            buffer.push(byte)?;
        }

        Ok(buffer)
    }
}

pub trait ByteRefIteratorIntoBuffer {
    /// Collects this [`Iterator`] into a [`Buffer`].
    /// # Errors
    /// [`BufferError`] is returned if an allocation or reallocation fails.
    fn into_buffer(self) -> Result<Buffer, BufferError>;
}

impl<'lt, T: Iterator<Item = &'lt u8>> ByteRefIteratorIntoBuffer for T {
    fn into_buffer(self) -> Result<Buffer, BufferError> {
        self.copied().into_buffer()
    }
}

pub trait ByteSliceIteratorIntoBuffer {
    /// Collects this [`Iterator`] into a [`Buffer`].
    /// # Errors
    /// [`BufferError`] is returned if an allocation or reallocation fails.
    fn into_buffer(self) -> Result<Buffer, BufferError>;
}

impl<Ts: AsRef<[u8]>, Ti: Iterator<Item = Ts>> ByteSliceIteratorIntoBuffer for Ti {
    fn into_buffer(self) -> Result<Buffer, BufferError> {
        let mut buffer = Buffer::new();

        for bytes in self {
            buffer.extend_from_slice(bytes.as_ref())?;
        }

        Ok(buffer)
    }
}