Ensure TCP connection is properly shut down when session is dropped

actix · Nov 1, 2024 · c8d9d06 · c8d9d06
1 parent 77406cb
commit c8d9d06
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/actix-ws/CHANGELOG.md b/actix-ws/CHANGELOG.md
@@ -1,6 +1,7 @@
 # Changelog
 
 ## Unreleased
+- Ensure TCP connection is properly shut down when session is dropped
 
 ## 0.3.0
 

diff --git a/actix-ws/src/stream.rs b/actix-ws/src/stream.rs
@@ -145,6 +145,10 @@ impl Stream for StreamingBody {
             return Poll::Ready(Some(Ok(mem::take(&mut this.buf).freeze())));
         }
 
+        if this.closing {
+            return Poll::Ready(None);
+        }
+
         Poll::Pending
     }
 }