karyon_net/codec/
bytes_codec.rs

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
use crate::{
    codec::{Codec, Decoder, Encoder},
    Error, Result,
};

#[derive(Clone)]
pub struct BytesCodec {}
impl Codec for BytesCodec {
    type Message = Vec<u8>;
    type Error = Error;
}

impl Encoder for BytesCodec {
    type EnMessage = Vec<u8>;
    type EnError = Error;
    fn encode(&self, src: &Self::EnMessage, dst: &mut [u8]) -> Result<usize> {
        dst[..src.len()].copy_from_slice(src);
        Ok(src.len())
    }
}

impl Decoder for BytesCodec {
    type DeMessage = Vec<u8>;
    type DeError = Error;
    fn decode(&self, src: &mut [u8]) -> Result<Option<(usize, Self::DeMessage)>> {
        if src.is_empty() {
            Ok(None)
        } else {
            Ok(Some((src.len(), src.to_vec())))
        }
    }
}