clear peerPromises map when fullfilling a promise

libp2p · Oct 29, 2021 · 85a1f01 · 85a1f01
1 parent cee11bb
commit 85a1f01
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/gossip_tracer.go b/gossip_tracer.go
@@ -118,11 +118,19 @@ var _ RawTracer = (*gossipTracer)(nil)
 
 func (gt *gossipTracer) fulfillPromise(msg *Message) {
 	mid := gt.msgID(msg.Message)
+	p := msg.ReceivedFrom
 
 	gt.Lock()
 	defer gt.Unlock()
 
 	delete(gt.promises, mid)
+	peerPromises, ok := gt.peerPromises[p]
+	if ok {
+		delete(peerPromises, mid)
+		if len(peerPromises) == 0 {
+			delete(gt.peerPromises, p)
+		}
+	}
 }
 
 func (gt *gossipTracer) DeliverMessage(msg *Message) {