Clear potential connections if a connection succeeded (#468)

Novage · Feb 8, 2025 · c7b5738 · c7b5738
1 parent a5f68ba
commit c7b5738
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/p2p-media-loader-core/src/p2p/tracker-client.ts b/packages/p2p-media-loader-core/src/p2p/tracker-client.ts
@@ -97,23 +97,29 @@ export class P2PTrackerClient {
   ) => {
     const itemId = Peer.getPeerIdFromConnection(peerConnection);
     let peerItem = this._peers.get(itemId);
+
     if (peerItem?.peer) {
       peerConnection.destroy();
       return;
-    } else if (!peerItem) {
+    }
+
+    if (!peerItem) {
       peerItem = { potentialConnections: new Set() };
       peerConnection.idUtf8 = itemId;
-      peerItem.potentialConnections.add(peerConnection);
       this._peers.set(itemId, peerItem);
     }
 
+    peerItem.potentialConnections.add(peerConnection);
+
     peerConnection.on("connect", () => {
       if (peerItem.peer) return;
 
       for (const connection of peerItem.potentialConnections) {
         if (connection !== peerConnection) connection.destroy();
       }
+
       peerItem.potentialConnections.clear();
+
       peerItem.peer = new Peer(
         peerConnection,
         {