Inline

Author: blast-hardcheese

src/replit_river/client_session.py

@@ -113,11 +113,8 @@ async def serve(self) -> None:
 
     async def _update_book_keeping(self, msg: TransportMessage) -> None:
         await self._seq_manager.check_seq_and_update(msg)
-        await self._remove_acked_messages_in_buffer()
-        self._reset_session_close_countdown()
-
-    async def _remove_acked_messages_in_buffer(self) -> None:
         await self._buffer.remove_old_messages(self._seq_manager.receiver_ack)
+        self._reset_session_close_countdown()
 
     async def _handle_messages_from_ws(self) -> None:
         logger.debug(

src/replit_river/server_session.py

@@ -111,11 +111,8 @@ async def serve(self) -> None:
 
     async def _update_book_keeping(self, msg: TransportMessage) -> None:
         await self._seq_manager.check_seq_and_update(msg)
-        await self._remove_acked_messages_in_buffer()
-        self._reset_session_close_countdown()
-
-    async def _remove_acked_messages_in_buffer(self) -> None:
         await self._buffer.remove_old_messages(self._seq_manager.receiver_ack)
+        self._reset_session_close_countdown()
 
     async def _handle_messages_from_ws(self, tg: asyncio.TaskGroup) -> None:
         logger.debug(