ensure poll_flush on h1 connection disconnect

2021-02-09 20:57:52 -08:00 · 2021-02-09 20:57:52 -08:00 · 79676b46f3
parent 949d14ae2b
commit 79676b46f3
1 changed files with 2 additions and 3 deletions
--- a/actix-http/src/h1/dispatcher.rs
+++ b/actix-http/src/h1/dispatcher.rs
@ -841,9 +841,8 @@ where
                    if inner.flags.contains(Flags::WRITE_DISCONNECT) {
                        Poll::Ready(Ok(()))
                    } else {
-                        // flush buffer
-                        inner.as_mut().poll_flush(cx)?;
-                        if !inner.write_buf.is_empty() {
+                        // flush buffer and wait on block.
+                        if inner.as_mut().poll_flush(cx)? {
                            Poll::Pending
                        } else {
                            Pin::new(inner.project().io.as_mut().unwrap())