Skip to content

Commit 174bd33

Browse files
Permit errors from buffered_message_sender to bubble back to the calling thread
1 parent ea90d5d commit 174bd33

File tree

2 files changed

+95
-84
lines changed

2 files changed

+95
-84
lines changed

src/replit_river/common_session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ async def buffered_message_sender(
5959
get_ws: Callable[[], WebSocketCommonProtocol | ClientConnection | None],
6060
websocket_closed_callback: Callable[[], Coroutine[Any, Any, None]],
6161
get_next_pending: Callable[[], TransportMessage | None],
62-
commit: Callable[[TransportMessage], None],
62+
commit: Callable[[TransportMessage], Awaitable[None]],
6363
get_state: Callable[[], SessionState],
6464
) -> None:
6565
our_task = asyncio.current_task()
@@ -89,7 +89,7 @@ async def buffered_message_sender(
8989
)
9090
try:
9191
await send_transport_message(msg, ws, websocket_closed_callback)
92-
commit(msg)
92+
await commit(msg)
9393
except WebsocketClosedException as e:
9494
logger.debug(
9595
"_buffered_message_sender: Connection closed while sending "

0 commit comments

Comments
 (0)