revert Poll match in poll_flush

This commit is contained in:
fakeshadow 2021-01-11 07:24:41 +08:00
parent 7659318afe
commit 98491dd5a6
1 changed files with 6 additions and 9 deletions

View File

@ -298,16 +298,13 @@ where
let mut written = 0;
while written < len {
match io.as_mut().poll_write(cx, &write_buf[written..]) {
Poll::Ready(Ok(n)) => {
if n == 0 {
return Err(DispatchError::Io(io::Error::new(
io::ErrorKind::WriteZero,
"",
)));
} else {
written += n;
}
Poll::Ready(Ok(0)) => {
return Err(DispatchError::Io(io::Error::new(
io::ErrorKind::WriteZero,
"",
)))
}
Poll::Ready(Ok(n)) => written += n,
Poll::Pending => {
write_buf.advance(written);
return Ok(true);