chore: handle edge case

waku-org · Aug 28, 2024 · f387f59 · f387f59
1 parent 2b61774
commit f387f59
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/packages/sdk/src/protocols/filter/reliability_monitor.ts b/packages/sdk/src/protocols/filter/reliability_monitor.ts
@@ -6,9 +6,7 @@ import { Logger } from "@waku/utils";
 
 type ReceivedMessageHashes = {
   all: Set<string>;
-  nodes: {
-    [peerId: PeerIdStr]: Set<string>;
-  };
+  nodes: Record<PeerIdStr, Set<string>>;
 };
 
 const DEFAULT_MAX_MISSED_MESSAGES_THRESHOLD = 3;
@@ -60,6 +58,13 @@ export class ReliabilityMonitor {
     this.receivedMessagesHashes.all.add(hashedMessageStr);
 
     if (peerIdStr) {
+      const x = this.receivedMessagesHashes.nodes[peerIdStr];
+      if (!x) {
+        log.warn(
+          `Peer ${peerIdStr} not initialized in receivedMessagesHashes.nodes, adding it.`
+        );
+        this.receivedMessagesHashes.nodes[peerIdStr] = new Set();
+      }
       this.receivedMessagesHashes.nodes[peerIdStr].add(hashedMessageStr);
     }