diff --git a/rw-stream-sink/src/lib.rs b/rw-stream-sink/src/lib.rs index fe6e9eeb..c59e2551 100644 --- a/rw-stream-sink/src/lib.rs +++ b/rw-stream-sink/src/lib.rs @@ -145,8 +145,10 @@ where #[inline] fn flush(&mut self) -> Result<(), IoError> { - self.inner.poll_complete()?; - Ok(()) + match self.inner.poll_complete()? { + Async::Ready(()) => Ok(()), + Async::NotReady => Err(IoError::new(IoErrorKind::WouldBlock, "not ready")) + } } }