Fix: reuse peerId by info hash

Novage · Feb 5, 2025 · a3cb325 · a3cb325
1 parent 22d48fb
commit a3cb325
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/p2p-media-loader-core/src/p2p/tracker-client.ts b/packages/p2p-media-loader-core/src/p2p/tracker-client.ts
@@ -37,6 +37,8 @@ function isSafariOrWkWebview() {
 }
 
 export class P2PTrackerClient {
+  private static readonly PEER_ID_BY_INFO_HASH = new Map<string, string>();
+
   private readonly streamShortId: string;
   private readonly client: TrackerClient;
   private readonly _peers = new Map<string, PeerItem>();
@@ -52,7 +54,11 @@ export class P2PTrackerClient {
     const streamHash = PeerUtil.getStreamHash(streamSwarmId);
     this.streamShortId = LoggerUtils.getStreamString(stream);
 
-    const peerId = PeerUtil.generatePeerId(config.trackerClientVersionPrefix);
+    let peerId = P2PTrackerClient.PEER_ID_BY_INFO_HASH.get(streamHash);
+    if (!peerId) {
+      peerId = PeerUtil.generatePeerId(config.trackerClientVersionPrefix);
+      P2PTrackerClient.PEER_ID_BY_INFO_HASH.set(streamHash, peerId);
+    }
 
     this.client = new TrackerClient({
       infoHash: utf8ToUintArray(streamHash),