fix: Ensure blocking read stream can be shutdown (#37)

launchdarkly · Dec 23, 2024 · 58c4702 · 58c4702
1 parent 1872b4a
commit 58c4702
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ld_eventsource/http.py b/ld_eventsource/http.py
@@ -87,7 +87,15 @@ def connect(self, last_event_id: Optional[str]) -> Tuple[Iterator[bytes], Callab
             raise HTTPContentTypeError(content_type or '')
 
         stream = resp.stream(_CHUNK_SIZE)
-        return stream, resp.release_conn
+
+        def close():
+            try:
+                resp.shutdown()
+            except AttributeError:
+                pass
+            resp.release_conn()
+
+        return stream, close
 
     def close(self):
         if self.__should_close_pool: