server/src/binary/handlers/utils.rs

use crate::binary::{command::ServerCommand, sender::SenderKind};
use bytes::{BufMut, BytesMut};
use iggy::error::IggyError;

pub async fn receive_and_validate(
    sender: &mut SenderKind,
    code: u32,
    length: u32,
) -> Result<ServerCommand, IggyError> {
    let mut buffer = BytesMut::with_capacity(length as usize);
    buffer.put_bytes(0, length as usize);
    if length > 0 {
        sender.read(&mut buffer).await?;
    }

    let command = ServerCommand::from_code_and_payload(code, buffer.freeze())?;
    command.validate()?;
    Ok(command)
}