From ea61bc62e8f90e975fc8cf37950edae2d418e7bd Mon Sep 17 00:00:00 2001
From: Wondertan <hlibwondertan@gmail.com>
Date: Thu, 7 Nov 2024 13:38:07 +0100
Subject: [PATCH] cleanup peerresponsetracker

---
 bitswap/client/internal/session/peerresponsetracker.go | 5 +++++
 bitswap/client/internal/session/sessionwantsender.go   | 1 +
 2 files changed, 6 insertions(+)

diff --git a/bitswap/client/internal/session/peerresponsetracker.go b/bitswap/client/internal/session/peerresponsetracker.go
index f96424742..122317cdc 100644
--- 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
index e4ca3b1a5..905250b01 100644
--- 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)
 		}
 	}