wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Queuing the messages received before the data channel is ready to send them.

Browse Source
This commit is contained in:
filipi87
2026-03-09 09:47:45 -03:00
parent 30d0cccef0
commit 25165d6e2b
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/pipecat/transports/smallwebrtc/connection.py
View File

@@ -283,6 +283,7 @@ class SmallWebRTCConnection(BaseObject):
self._data_channel = None self._data_channel = None
self._renegotiation_in_progress = False self._renegotiation_in_progress = False
self._last_received_time = None self._last_received_time = None
self._message_queue = []
self._pending_app_messages = [] self._pending_app_messages = []
self._connecting_timeout_task = None self._connecting_timeout_task = None
@@ -297,6 +298,7 @@ class SmallWebRTCConnection(BaseObject):
@channel.on("open") @channel.on("open")
async def on_open(): async def on_open():
logger.debug("Data channel is open!") logger.debug("Data channel is open!")
self._flush_message_queue()
@channel.on("message") @channel.on("message")
async def on_message(message): async def on_message(message):
@@ -499,6 +501,7 @@ class SmallWebRTCConnection(BaseObject):
self._track_map.clear() self._track_map.clear()
if self._pc: if self._pc:
await self._pc.close() await self._pc.close()
self._message_queue.clear()
self._pending_app_messages.clear() self._pending_app_messages.clear()
self._track_map = {} self._track_map = {}
self._cancel_monitoring_connecting_state() self._cancel_monitoring_connecting_state()
@@ -664,8 +667,14 @@ class SmallWebRTCConnection(BaseObject):
if self._data_channel and self._data_channel.readyState == "open": if self._data_channel and self._data_channel.readyState == "open":
self._data_channel.send(json_message) self._data_channel.send(json_message)
else: else:
# The client might choose never to create a data channel. logger.debug("Data channel not ready, queuing message")
logger.trace("Data channel not ready, discarding message!") self._message_queue.append(json_message)
def _flush_message_queue(self):
logger.debug("Data channel is open, flushing queued messages")
while self._message_queue:
message = self._message_queue.pop(0)
self._data_channel.send(message)
def ask_to_renegotiate(self): def ask_to_renegotiate(self):
"""Request renegotiation of the WebRTC connection.""" """Request renegotiation of the WebRTC connection."""