Allow to wake up write io task

ntex/CHANGES.md

@@ -1,5 +1,7 @@
 # Changes
 
+* Allow to wake up write io task
+
 ## [0.2.0-b.7] - 2021-01-25
 
 * Fix error handling for framed disaptcher

ntex/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex"
-version = "0.2.0-b.8"
+version = "0.2.0-b.9"
 authors = ["ntex contributors <team@ntex.rs>"]
 description = "Framework for composable network services"
 readme = "README.md"

ntex/src/framed/state.rs

@@ -159,6 +159,12 @@ impl State {
         self.0.flags.get().contains(Flags::IO_STOP)
     }
 
+    #[inline]
+    /// Check if write buff is full
+    pub fn is_write_buf_full(&self) -> bool {
+        self.0.write_buf.borrow().len() >= HW
+    }
+
     #[inline]
     /// Check if read buffer has new data
     pub fn is_read_ready(&self) -> bool {
@@ -320,6 +326,12 @@ impl State {
         }
     }
 
+    #[inline]
+    /// Wake write io task
+    pub fn dsp_restart_write_task(&self) {
+        self.0.write_task.wake();
+    }
+
     #[inline]
     /// Wake read io task if it is not ready
     pub fn dsp_read_more_data(&self, waker: &Waker) {
@@ -330,12 +342,6 @@ impl State {
         self.0.dispatch_task.register(waker);
     }
 
-    #[inline]
-    /// Check if write buff is full
-    pub fn is_write_buf_full(&self) -> bool {
-        self.0.write_buf.borrow().len() >= HW
-    }
-
     #[inline]
     /// Wait until write task flushes data to socket
     ///