tokio-rs/tokio
1
0
Fork 0
mirror of https://github.com/tokio-rs/tokio.git synced 2026-09-08 00:00:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Return NotReady from TcpStream::write_buf

Browse Source 
Closes #194
This commit is contained in:
Alex Crichton
2017-03-27 07:33:18 -07:00
parent 37e2870f04
commit 45e69bfb61
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/net/tcp.rs
+5 -6
View File
@@ -541,7 +541,7 @@ impl<'a> AsyncWrite for &'a TcpStream {
fn write_buf<B: Buf>(&mut self, buf: &mut B) -> Poll<usize, io::Error> { fn write_buf<B: Buf>(&mut self, buf: &mut B) -> Poll<usize, io::Error> {
if let Async::NotReady = <TcpStream>::poll_write(self) { if let Async::NotReady = <TcpStream>::poll_write(self) {
return Err(::would_block()) return Ok(Async::NotReady)
} }
let r = { let r = {
let mut bufs: [_; 16] = Default::default(); let mut bufs: [_; 16] = Default::default();
@@ -553,12 +553,11 @@ impl<'a> AsyncWrite for &'a TcpStream {
buf.advance(n); buf.advance(n);
Ok(Async::Ready(n)) Ok(Async::Ready(n))
} }
Err(e) => { Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
if e.kind() == io::ErrorKind::WouldBlock { self.io.need_write();
self.io.need_write(); Ok(Async::NotReady)
}
Err(e)
} }
Err(e) => Err(e),
} }
} }
} }