src/resp/buffer_decoder.rs

use super::RespDeserializer;
use crate::{resp::RespBuf, Error, Result};
use bytes::BytesMut;
use serde::{de::IgnoredAny, Deserialize};
use tokio_util::codec::Decoder;

pub(crate) struct BufferDecoder;

impl Decoder for BufferDecoder {
    type Item = RespBuf;
    type Error = Error;

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

        let bytes = src.as_ref();
        let mut deserializer = RespDeserializer::new(bytes);
        let result = IgnoredAny::deserialize(&mut deserializer);
        match result {
            Ok(_) => Ok(Some(RespBuf::new(
                src.split_to(deserializer.get_pos()).freeze(),
            ))),
            Err(Error::EOF) => Ok(None),
            Err(e) => Err(e),
        }
    }
}