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

use std::io::{Error, ErrorKind};

use bytes::BytesMut;
use serde::{Serialize, de::DeserializeOwned};
use tokio_util::codec::Encoder;

use super::GenericLinesCodec;

impl<T: Serialize + DeserializeOwned> Encoder<T> for GenericLinesCodec<T> {
    type Error = Error;

    fn encode(
        &mut self,
        message: T,
        dist_buff: &mut BytesMut,
    ) -> Result<(), Self::Error> {
        let data = serde_json::to_string(&message)
            .expect("Cannot serialize message to bytes.");
        
        let result = Encoder::encode(
            &mut self.length_delimited_codec, 
            &data,
            dist_buff,
        );

        match result {
            Ok(_) => return Ok(()),
            Err(_err) => return Err(
                Error::new(
                    ErrorKind::InvalidData,
                    format!("Cannot encode message: {:?}.", &data),
                ),
            ),
        };
    }
}