Skip to content

Commit 5a71e0a

Browse files
Maybe this is where the slowdown is?
The while := should be a tight loop, but maybe too tight
1 parent abd1584 commit 5a71e0a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/replit_river/common_session.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,7 @@ async def buffered_message_sender(
130130
logger.debug("Entering buffered_message_sender")
131131
while True:
132132
logger.debug("buffered_message_sender: LOOP")
133+
sent = False
133134
while msg := get_next_pending():
134135
ws = get_ws()
135136
logger.debug("buffered_message_sender: Dequeued %r to send over %r", msg, ws)
@@ -140,6 +141,7 @@ async def buffered_message_sender(
140141
await send_transport_message(msg, ws, websocket_closed_callback)
141142
logger.debug("buffered_message_sender: committing!")
142143
commit(msg)
144+
sent = True
143145
except WebsocketClosedException as e:
144146
logger.debug(
145147
"Connection closed while sending message %r, waiting for "
@@ -157,7 +159,8 @@ async def buffered_message_sender(
157159
except Exception:
158160
logger.exception("Error attempting to send buffered messages")
159161
break
160-
await asyncio.sleep(0.25)
162+
if not sent:
163+
await asyncio.sleep(0.25)
161164

162165

163166
async def add_msg_to_stream(

0 commit comments

Comments
 (0)