Skip to content

Commit b134299

Browse files
committed
update async stuff
1 parent 9e7974a commit b134299

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

livekit-rtc/livekit/rtc/room.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -747,9 +747,11 @@ def _on_room_event(self, event: proto_room.RoomEvent):
747747
event.stream_header_received.participant_identity,
748748
)
749749
elif which == "stream_chunk_received":
750-
self._handle_stream_chunk(event.stream_chunk_received.chunk)
750+
asyncio.run(self._handle_stream_chunk(event.stream_chunk_received.chunk))
751751
elif which == "stream_trailer_received":
752-
self._handle_stream_trailer(event.stream_trailer_received.trailer)
752+
asyncio.run(
753+
self._handle_stream_trailer(event.stream_trailer_received.trailer)
754+
)
753755

754756
async def _drain_rpc_invocation_tasks(self) -> None:
755757
if self._rpc_invocation_tasks:
@@ -813,24 +815,24 @@ def _handle_stream_header(
813815
logging.warning("received unknown header type, %s", stream_type)
814816
pass
815817

816-
def _handle_stream_chunk(self, chunk: proto_room.DataStream.Chunk):
818+
async def _handle_stream_chunk(self, chunk: proto_room.DataStream.Chunk):
817819
text_reader = self._text_stream_readers.get(chunk.stream_id)
818820
file_reader = self._byte_stream_readers.get(chunk.stream_id)
819821

820822
if text_reader:
821-
text_reader._on_chunk_update(chunk)
823+
await text_reader._on_chunk_update(chunk)
822824
elif file_reader:
823-
file_reader._on_chunk_update(chunk)
825+
await file_reader._on_chunk_update(chunk)
824826

825-
def _handle_stream_trailer(self, trailer: proto_room.DataStream.Trailer):
827+
async def _handle_stream_trailer(self, trailer: proto_room.DataStream.Trailer):
826828
text_reader = self._text_stream_readers.get(trailer.stream_id)
827829
file_reader = self._byte_stream_readers.get(trailer.stream_id)
828830

829831
if text_reader:
830-
text_reader._on_stream_close(trailer)
832+
await text_reader._on_stream_close(trailer)
831833
self._text_stream_readers.pop(trailer.stream_id)
832834
elif file_reader:
833-
file_reader._on_stream_close(trailer)
835+
await file_reader._on_stream_close(trailer)
834836
self._byte_stream_readers.pop(trailer.stream_id)
835837

836838
def __repr__(self) -> str:

0 commit comments

Comments
 (0)