streamdata 0.2.3

Sans-IO data stream decoding utilitites.
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
53
54
55
56
57
58
59
60
61
62

//! [`tokio_util`] integration (i.e. [`tokio_util::codec::Decoder`] support).

use crate::Buffer;

/// The decoder that wraps any [`tokio_util::codec::Decoder`].
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Decoder<T> {
    /// The inner decoder type.
    pub inner: T,
}

impl<T> Decoder<T> {
    /// Create a new [`Decoder`] with a given [`tokio_util::codec::Decoder`].
    pub const fn new(inner: T) -> Self {
        Self { inner }
    }
}

impl<T> crate::Decoder<Vec<u8>> for Decoder<T>
where
    T: tokio_util::codec::Decoder,
{
    type Value = <T as tokio_util::codec::Decoder>::Item;
    type Error = <T as tokio_util::codec::Decoder>::Error;

    #[allow(clippy::arithmetic_side_effects)]
    fn decode(
        &mut self,
        input: &mut Vec<u8>,
    ) -> Result<Self::Value, crate::DecodeError<Self::Error>> {
        let buf = input.view();
        let mut buf_bytes = bytes::BytesMut::from(buf);
        match tokio_util::codec::Decoder::decode(&mut self.inner, &mut buf_bytes) {
            Ok(None) => Err(crate::DecodeError::NeedMoreData),
            Ok(Some(value)) => {
                input.advance(buf.len() - buf_bytes.len());
                Ok(value)
            }
            Err(err) => Err(crate::DecodeError::Other(err)),
        }
    }
}

impl<T> crate::Decoder<bytes::BytesMut> for Decoder<T>
where
    T: tokio_util::codec::Decoder,
{
    type Value = <T as tokio_util::codec::Decoder>::Item;
    type Error = <T as tokio_util::codec::Decoder>::Error;

    #[allow(clippy::arithmetic_side_effects)]
    fn decode(
        &mut self,
        input: &mut bytes::BytesMut,
    ) -> Result<Self::Value, crate::DecodeError<Self::Error>> {
        match tokio_util::codec::Decoder::decode(&mut self.inner, input) {
            Ok(None) => Err(crate::DecodeError::NeedMoreData),
            Ok(Some(value)) => Ok(value),
            Err(err) => Err(crate::DecodeError::Other(err)),
        }
    }
}