mirrors/ntex
1
0
Fork 0
mirror of https://github.com/ntex-rs/ntex.git synced 2025-04-04 13:27:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

tune error handling

Browse source
This commit is contained in:
Nikolay Kim 2021-12-29 09:42:36 +06:00
parent 59eedcc51c
commit c5d43eb12d
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
ntex-io/src/io.rs
View file

@ -556,6 +556,11 @@ impl<F> Io<F> {
match self.decode(codec) { match self.decode(codec) {
Ok(Some(el)) => Poll::Ready(Ok(el)), Ok(Some(el)) => Poll::Ready(Ok(el)),
Ok(None) => { Ok(None) => {
if !self.0 .0.is_io_open() {
return Poll::Ready(Err(RecvError::PeerGone(
self.0 .0.error.take(),
)));
}
let flags = self.flags(); let flags = self.flags();
if flags.contains(Flags::DSP_STOP) { if flags.contains(Flags::DSP_STOP) {
Poll::Ready(Err(RecvError::Stop)) Poll::Ready(Err(RecvError::Stop))
@ -591,6 +596,14 @@ impl<F> Io<F> {
pub fn poll_flush(&self, cx: &mut Context<'_>, full: bool) -> Poll<io::Result<()>> { pub fn poll_flush(&self, cx: &mut Context<'_>, full: bool) -> Poll<io::Result<()>> {
// check io error // check io error
if !self.0 .0.is_io_open() { if !self.0 .0.is_io_open() {
self.0 .0.remove_flags(
Flags::DSP_KEEPALIVE
| Flags::RD_PAUSED
| Flags::RD_READY
| Flags::RD_BUF_FULL
| Flags::WR_WAIT
| Flags::WR_BACKPRESSURE,
);
return Poll::Ready(Err(self return Poll::Ready(Err(self
.0 .0
.0 .0