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

Stop write task if io is closed (#416)

Browse source
This commit is contained in:
Nikolay Kim 2024-09-07 10:04:08 +05:00 • committed by GitHub
parent 3edb54ffdf
commit db6d3a6e4c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 59 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

14
ntex-tokio/src/io.rs
View file

@ -137,6 +137,10 @@ impl Future for WriteTask {
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
let this = self.as_mut().get_mut();
if this.state.poll_close(cx).is_ready() {
return Poll::Ready(());
}
match this.st {
IoWriteState::Processing(ref mut delay) => {
match ready!(this.state.poll_ready(cx)) {
@ -215,6 +219,9 @@ impl Future for WriteTask {
// close WRITE side and wait for disconnect on read side.
// use disconnect timeout, otherwise it could hang forever.
loop {
if this.state.poll_close(cx).is_ready() {
return Poll::Ready(());
}
match st {
Shutdown::None => {
// flush write buffer
@ -564,6 +571,10 @@ mod unixstream {
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
let this = self.as_mut().get_mut();
if this.state.poll_close(cx).is_ready() {
return Poll::Ready(());
}
match this.st {
IoWriteState::Processing(ref mut delay) => {
match this.state.poll_ready(cx) {
@ -630,6 +641,9 @@ mod unixstream {
// close WRITE side and wait for disconnect on read side.
// use disconnect timeout, otherwise it could hang forever.
loop {
if this.state.poll_close(cx).is_ready() {
return Poll::Ready(());
}
match st {
Shutdown::None => {
// flush write buffer