Fix KeepAlive timeout handling in default dispatcher (#351)

ntex-io/CHANGES.md

@@ -1,5 +1,11 @@
 # Changes
 
+## [1.1.0] - 2024-05-01
+
+* Add IoRef::notify_timeout() helper method
+
+* Fix KeepAlive timeout handling in default dispatcher
+
 ## [1.0.2] - 2024-03-31
 
 * Add IoRef::is_wr_backpressure() method

ntex-io/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex-io"
-version = "1.0.2"
+version = "1.1.0"
 authors = ["ntex contributors <team@ntex.rs>"]
 description = "Utilities for encoding and decoding frames"
 keywords = ["network", "framework", "async", "futures"]

ntex-io/src/dispatcher.rs

@@ -575,15 +575,19 @@ where
                         self.shared.io.tag()
                     );
                 }
-                return Err(DispatchItem::ReadTimeout);
+                Err(DispatchItem::ReadTimeout)
+            } else {
+                Ok(())
             }
+        } else if self.flags.contains(Flags::KA_TIMEOUT) {
+            log::trace!(
+                "{}: Keep-alive error, stopping dispatcher",
+                self.shared.io.tag()
+            );
+            Err(DispatchItem::KeepAliveTimeout)
+        } else {
+            Ok(())
         }
-
-        log::trace!(
-            "{}: Keep-alive error, stopping dispatcher",
-            self.shared.io.tag()
-        );
-        Err(DispatchItem::KeepAliveTimeout)
     }
 }
 

ntex-io/src/io.rs

@@ -93,13 +93,12 @@ impl IoState {
     }
 
     pub(super) fn notify_timeout(&self) {
-        log::trace!("{}: Timeout, notify dispatcher", self.tag.get());
-
         let mut flags = self.flags.get();
         if !flags.contains(Flags::DSP_TIMEOUT) {
             flags.insert(Flags::DSP_TIMEOUT);
             self.flags.set(flags);
             self.dispatch_task.wake();
+            log::trace!("{}: Timer, notify dispatcher", self.tag.get());
         }
     }
 

ntex-io/src/ioref.rs

@@ -228,6 +228,12 @@ impl IoRef {
         self.0.timeout.get()
     }
 
+    #[inline]
+    /// wakeup dispatcher and send keep-alive error
+    pub fn notify_timeout(&self) {
+        self.0.notify_timeout()
+    }
+
     #[inline]
     /// Start timer
     pub fn start_timer(&self, timeout: Seconds) -> timer::TimerHandle {