From 3571f1dfd6f88cf9fa9acfcd748c6f9bb09716b6 Mon Sep 17 00:00:00 2001 From: Min Kim Date: Fri, 10 May 2024 14:04:15 -0700 Subject: [PATCH] Use `io::Error` instead of `SendError` --- src/frame.rs | 8 ++++---- src/message.rs | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/frame.rs b/src/frame.rs index 7335e02..f377761 100644 --- a/src/frame.rs +++ b/src/frame.rs @@ -165,10 +165,10 @@ where /// /// # Errors /// -/// * `SendError::MessageTooLarge`: if the message is too large -/// * `SendError::WriteError`: if the message could not be written -pub async fn send_raw(send: &mut SendStream, buf: &[u8]) -> Result<(), SendError> { - let len = u32::try_from(buf.len()).map_err(|_| SendError::MessageTooLarge)?; +/// Returns an error if the message is too large or could not be written. +pub async fn send_raw(send: &mut SendStream, buf: &[u8]) -> io::Result<()> { + let len = + u32::try_from(buf.len()).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?; send.write_all(&len.to_be_bytes()).await?; send.write_all(buf).await?; Ok(()) diff --git a/src/message.rs b/src/message.rs index 41f5200..ad684ce 100644 --- a/src/message.rs +++ b/src/message.rs @@ -46,7 +46,7 @@ pub async fn send_request( buf: &mut Vec, code: C, body: B, -) -> Result<(), SendError> +) -> io::Result<()> where C: Into, B: Serialize, @@ -57,7 +57,7 @@ where buf.extend_from_slice(&code.into().to_le_bytes()); bincode::DefaultOptions::new() .serialize_into(buf as &mut dyn Write, &body) - .map_err(|_| SendError::MessageTooLarge)?; + .map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?; frame::send_raw(send, buf).await?; buf.clear(); Ok(())