encdec_base/encode/
ext.rs

1use super::Encode;
2
3/// Extensions to [`Encode`] trait for encodable types
4pub trait EncodeExt<'a>: Encode + Sized + 'a {
5    /// Helper to encode iterables
6    fn encode_iter(
7        items: impl Iterator<Item = &'a Self>,
8        buff: &mut [u8],
9    ) -> Result<usize, Self::Error> {
10        let mut index = 0;
11        for i in items {
12            index += i.encode(&mut buff[index..])?;
13        }
14        Ok(index)
15    }
16
17    /// Helper to encode to a fixed size buffer
18    fn encode_buff<const N: usize>(&self) -> Result<([u8; N], usize), Self::Error> {
19        let mut b = [0u8; N];
20        let n = self.encode(&mut b)?;
21        Ok((b, n))
22    }
23
24    /// Helper to encode to a vector, uses `encode_len` to set length
25    #[cfg(feature = "alloc")]
26    fn encode_vec(&self) -> Result<(alloc::vec::Vec<u8>, usize), Self::Error> {
27        let n = self.encode_len()?;
28        let mut b = alloc::vec![0u8; n];
29        let n = self.encode(&mut b)?;
30        Ok((b, n))
31    }
32}
33
34/// Blanket implementation for all [`Encode`] types
35impl<'a, T: Encode + 'a> EncodeExt<'a> for T {}
encdec_base/encode/ext.rs

encdec_base/encode/
ext.rs
