connection_utils/codecs/generic_lines_codec/
encoder.rs

1use std::io::{Error, ErrorKind};
2
3use bytes::BytesMut;
4use serde::{Serialize, de::DeserializeOwned};
5use tokio_util::codec::Encoder;
6
7use super::GenericLinesCodec;
8
9impl<T: Serialize + DeserializeOwned> Encoder<T> for GenericLinesCodec<T> {
10    type Error = Error;
11
12    fn encode(
13        &mut self,
14        message: T,
15        dist_buff: &mut BytesMut,
16    ) -> Result<(), Self::Error> {
17        let data = serde_json::to_string(&message)
18            .expect("Cannot serialize message to bytes.");
19        
20        let result = Encoder::encode(
21            &mut self.length_delimited_codec, 
22            &data,
23            dist_buff,
24        );
25
26        match result {
27            Ok(_) => return Ok(()),
28            Err(_err) => return Err(
29                Error::new(
30                    ErrorKind::InvalidData,
31                    format!("Cannot encode message: {:?}.", &data),
32                ),
33            ),
34        };
35    }
36}