cleanup peerresponsetracker

ipfs · Nov 18, 2024 · ea61bc6 · ea61bc6
1 parent ecd68dd
commit ea61bc6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/bitswap/client/internal/session/peerresponsetracker.go b/bitswap/client/internal/session/peerresponsetracker.go
@@ -61,3 +61,8 @@ func (prt *peerResponseTracker) getPeerCount(p peer.ID) int {
 	// will be chosen
 	return prt.firstResponder[p] + 1
 }
+
+// cleanup stops tracking a peer
+func (prt *peerResponseTracker) cleanup(p peer.ID) {
+	delete(prt.firstResponder, p)
+}
diff --git a/bitswap/client/internal/session/sessionwantsender.go b/bitswap/client/internal/session/sessionwantsender.go
@@ -332,6 +332,7 @@ func (sws *sessionWantSender) processAvailability(availability map[peer.ID]bool)
 			// Reset the count of consecutive DONT_HAVEs received from the
 			// peer
 			delete(sws.peerConsecutiveDontHaves, p)
+			sws.peerRspTrkr.cleanup(p)
 		}
 	}
-Original file line number
+Diff line change
@@ Expand Up @@
     			// Reset the count of consecutive DONT_HAVEs received from the
     			// peer
     			delete(sws.peerConsecutiveDontHaves, p)
+    			sws.peerRspTrkr.cleanup(p)
     		}
     	}
@@ Expand Down @@