connection_utils/codecs/generic_lines_codec/
decoder.rs

1use std::io::{Error, ErrorKind};
2
3use bytes::BytesMut;
4use serde::{Serialize, de::DeserializeOwned};
5use tokio_util::codec::Decoder;
6
7use super::GenericLinesCodec;
8
9impl<T: Serialize + DeserializeOwned> Decoder for GenericLinesCodec<T> {
10    type Item = T;
11    type Error = Error;
12
13    fn decode(
14        &mut self,
15        buf: &mut BytesMut,
16    ) -> Result<Option<Self::Item>, Self::Error> {
17        let str_result = Decoder::decode(
18            &mut self.length_delimited_codec, 
19            buf,
20        );
21
22        let maybe_str = match str_result {
23            Ok(s) => s,
24            Err(_err) => return Err(
25                Error::new(
26                    ErrorKind::InvalidData,
27                    format!("Cannot decode input: {:0>2X?}.", buf),
28                ),
29            ),
30        };
31
32        let string = match maybe_str {
33            None => return Ok(None),
34            Some(b) => b,
35        };
36
37        let message: T = serde_json::from_str(&string)
38            .expect(
39                &format!(
40                    "Cannot deserialize string to message: {:?}.",
41                    &string,
42                ),
43            );
44
45        return Ok(Some(message));
46    } 
47}