shuffle peers when limiting px set

libp2p · Dec 5, 2019 · 2f558ba · 2f558ba
1 parent a09bca2
commit 2f558ba
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/gossipsub.go b/gossipsub.go
@@ -287,6 +287,7 @@ func (gs *GossipSubRouter) addBackoff(p peer.ID, topic string) {
 
 func (gs *GossipSubRouter) pxConnect(peers []*pb.PeerInfo) {
 	if len(peers) > GossipSubPrunePeers {
+		shufflePeerInfo(peers)
 		peers = peers[:GossipSubPrunePeers]
 	}
 
@@ -896,3 +897,10 @@ func shufflePeers(peers []peer.ID) {
 		peers[i], peers[j] = peers[j], peers[i]
 	}
 }
+
+func shufflePeerInfo(peers []*pb.PeerInfo) {
+	for i := range peers {
+		j := rand.Intn(i + 1)
+		peers[i], peers[j] = peers[j], peers[i]
+	}
+}