#
actix_codec/
bcodec.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 std::io;

    use bytes::{Buf, Bytes, BytesMut};

    use super::{Decoder, Encoder};

    /// Bytes codec. Reads/writes chunks of bytes from a stream.
    #[derive(Debug, Copy, Clone)]
    pub struct BytesCodec;

    impl Encoder<Bytes> for BytesCodec {
        type Error = io::Error;

        #[inline]
        fn encode(&mut self, item: Bytes, dst: &mut BytesMut) -> Result<(), Self::Error> {
            dst.extend_from_slice(item.chunk());
            Ok(())
        }
    }

    impl Decoder for BytesCodec {
        type Item = BytesMut;
        type Error = io::Error;

        fn decode(&mut self, src: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
            if src.is_empty() {
                Ok(None)
            } else {
                Ok(Some(src.split()))
            }
        }
    }