🤔 Avoid sending more messages when we know we are closed

Author: blast-hardcheese

src/replit_river/v2/client_session.py

@@ -75,6 +75,7 @@ def __init__(
 
         async def do_close_websocket() -> None:
             if self._ws_unwrapped:
+                self._ws_connected = False
                 self._task_manager.create_task(self._ws_unwrapped.close())
                 if self._retry_connection_callback:
                     self._task_manager.create_task(self._retry_connection_callback())
@@ -99,7 +100,7 @@ def get_next_pending() -> TransportMessage | None:
 
         self._task_manager.create_task(
             buffered_message_sender(
-                get_ws=lambda: self._ws_unwrapped,
+                get_ws=lambda: self._ws_unwrapped if self.is_websocket_open() else None,
                 websocket_closed_callback=self._begin_close_session_countdown,
                 get_next_pending=get_next_pending,
                 commit=commit,

src/replit_river/v2/session.py

@@ -169,6 +169,7 @@ async def replace_with_new_websocket(
                 )
             )
         self._ws_unwrapped = new_ws
+        self._ws_connected = True
 
     async def _get_current_time(self) -> float:
         return asyncio.get_event_loop().time()