fix: ensure TCP connection is properly shut down when Session is drop…

…ped (#476) * Ensure TCP connection is properly shut down when session is dropped * Update CHANGELOG.md --------- Co-authored-by: Rob Ede <[email protected]>
actix · Dec 29, 2024 · 98847b9 · 98847b9
1 parent cd1b771
commit 98847b9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/actix-ws/CHANGELOG.md b/actix-ws/CHANGELOG.md
@@ -2,6 +2,8 @@
 
 ## Unreleased
 
+- Ensure TCP connection is properly shut down when session is dropped.
+
 ## 0.3.0
 
 - Add `AggregatedMessage[Stream]` types.

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
     }
 }