From 4d4cb10adb59d9aa45a8db887e848d51fe06f789 Mon Sep 17 00:00:00 2001 From: Matthew Keeler Date: Mon, 23 Dec 2024 10:37:56 -0500 Subject: [PATCH] fix: Ensure blocking read stream can be shutdown --- ld_eventsource/http.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/ld_eventsource/http.py b/ld_eventsource/http.py index 940446a..05c4e3a 100644 --- 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: