1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-09-30 15:22:02 +00:00

ensure poll_flush on h1 connection disconnect (#1974)

This commit is contained in:
fakeshadow 2021-02-10 02:11:53 -08:00 committed by GitHub
parent 949d14ae2b
commit dcad9724bc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

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