avoid KeyError when track is already unpublished

Author: davidzhao

livekit-rtc/livekit/rtc/room.py

@@ -821,11 +821,22 @@ def _on_room_event(self, event: proto_room.RoomEvent) -> None:
             rparticipant._track_publications[rpublication.sid] = rpublication
             self.emit("track_published", rpublication, rparticipant)
         elif which == "track_unpublished":
-            rparticipant = self._remote_participants[event.track_unpublished.participant_identity]
-            rpublication = rparticipant._track_publications.pop(
-                event.track_unpublished.publication_sid
-            )
-            self.emit("track_unpublished", rpublication, rparticipant)
+            # The participant or publication may already have been removed by a
+            # racing disconnect or a duplicate event, so both lookups are done
+            # defensively and the emit is skipped when the entry is gone,
+            # mirroring the local_track_unpublished handler, instead of raising a
+            # KeyError that _listen_task logs as an error.
+            identity = event.track_unpublished.participant_identity
+            sid = event.track_unpublished.publication_sid
+            rparticipant = self._remote_participants.get(identity)
+            if rparticipant is not None:
+                rpublication = rparticipant._track_publications.pop(sid, None)
+                if rpublication is not None:
+                    self.emit("track_unpublished", rpublication, rparticipant)
+                else:
+                    logging.debug("track_unpublished for untracked publication sid %s", sid)
+            else:
+                logging.debug("track_unpublished for untracked participant %s", identity)
         elif which == "track_subscribed":
             owned_track_info = event.track_subscribed.track
             track_info = owned_track_info.info