Stop write task if io is closed (#416)

ntex-tokio/CHANGES.md

@@ -1,5 +1,9 @@
 # Changes
 
+## [0.5.1] - 2024-09-06
+
+* Stop write task if io is closed
+
 ## [0.4.0] - 2024-01-09
 
 * Log io tags

ntex-tokio/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex-tokio"
-version = "0.5.0"
+version = "0.5.1"
 authors = ["ntex contributors <team@ntex.rs>"]
 description = "tokio intergration for ntex framework"
 keywords = ["network", "framework", "async", "futures"]
@@ -17,7 +17,7 @@ path = "src/lib.rs"
 
 [dependencies]
 ntex-bytes = "0.1"
-ntex-io = "2.0"
-ntex-util = "2.0"
+ntex-io = "2.4"
+ntex-util = "2"
 log = "0.4"
 tokio = { version = "1", default-features = false, features = ["rt", "net", "sync", "signal"] }

ntex-tokio/src/io.rs

@@ -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