fix send return

franklee26 · Dec 31, 2024 · 6442d41 · 6442d41
1 parent 79a6407
commit 6442d41
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/src/net/connection.rs b/src/net/connection.rs
@@ -310,9 +310,6 @@ impl BluefinConnection {
 
     #[inline]
     pub fn send(&mut self, buf: &[u8]) -> BluefinResult<usize> {
-        // TODO! This returns the total bytes sent (including bluefin payload). This
-        // really should only return the total payload bytes
-        self.writer_handler.send_data(buf)?;
-        Ok(buf.len())
+        self.writer_handler.send_data(buf)
     }
 }
diff --git a/src/worker/writer.rs b/src/worker/writer.rs
@@ -74,7 +74,7 @@ impl WriterHandler {
     }
 
     #[inline]
-    pub(crate) fn send_data(&self, payload: &[u8]) -> BluefinResult<()> {
+    pub(crate) fn send_data(&self, payload: &[u8]) -> BluefinResult<usize> {
         match self.data_sender {
             Some(ref sender) => {
                 if let Err(e) = sender.send(payload.to_vec()) {
@@ -83,7 +83,7 @@ impl WriterHandler {
                         e
                     )));
                 }
-                Ok(())
+                Ok(payload.len())
             }
             None => Err(BluefinError::WriteError(
                 "Sender is not available. Cannot send.".to_string(),