Merge pull request #77 from quartiq/rs/issue-76/write-fail

Handling WouldBlock errors on network send
quartiq · Dec 13, 2021 · 47fbf9d · 47fbf9d
2 parents 44f0fec + b3b1a1e
commit 47fbf9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/network_manager.rs b/src/network_manager.rs
@@ -98,9 +98,9 @@ where
         let socket = self.socket.as_mut().ok_or(Error::NotReady)?;
         self.network_stack
             .send(socket, &data)
-            .map_err(|err| match err {
-                nb::Error::WouldBlock => Error::WriteFail,
-                nb::Error::Other(err) => Error::Network(err),
+            .or_else(|err| match err {
+                nb::Error::WouldBlock => Ok(0),
+                nb::Error::Other(err) => Err(Error::Network(err)),
             })
             .and_then(|written| {
                 if written != data.len() {