Flush underlying io stream

ntex-codec/CHANGES.md

@@ -1,5 +1,9 @@
 # Changes
 
+## [0.2.2] - 2021-01-21
+
+* Flush underlying io stream
+
 ## [0.2.1] - 2020-08-10
 
 * Require `Debug` impl for `Error`

ntex-codec/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex-codec"
-version = "0.2.1"
+version = "0.2.2"
 authors = ["ntex contributors <team@ntex.rs>"]
 description = "Utilities for encoding and decoding frames"
 keywords = ["network", "framework", "async", "futures"]
@@ -19,11 +19,11 @@ path = "src/lib.rs"
 bitflags = "1.2.1"
 bytes = "0.5.6"
 either = "1.5.3"
-futures-core = "0.3.5"
-futures-sink = "0.3.5"
+futures-core = "0.3.12"
+futures-sink = "0.3.12"
 log = "0.4"
 tokio = { version = "0.2.6", default-features=false }
 
 [dev-dependencies]
-ntex = "0.1.21"
-futures = "0.3.5"
+ntex = "0.2.0-b.2"
+futures = "0.3.12"

ntex-codec/src/framed.rs

@@ -233,10 +233,7 @@ where
         log::trace!("flushing framed transport");
 
         let len = self.write_buf.len();
-        if len == 0 {
-            return Poll::Ready(Ok(()));
-        }
-
+        if len != 0 {
             let mut written = 0;
             while written < len {
                 match Pin::new(&mut self.io).poll_write(cx, &self.write_buf[written..]) {
@@ -268,6 +265,11 @@ where
             } else {
                 self.write_buf.advance(written);
             }
+        }
+
+        // flush
+        ready!(Pin::new(&mut self.io).poll_flush(cx))?;
+
         if self.write_buf.is_empty() {
             Poll::Ready(Ok(()))
         } else {