logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-04 05:43:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

shuffle peers when limiting px set

Browse Source
This commit is contained in:
vyzo 2019-12-05 21:01:37 +02:00
parent e4667b6073
commit efb09f2984
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
gossipsub.go
View File

@ -289,6 +289,7 @@ func (gs *GossipSubRouter) addBackoff(p peer.ID, topic string) {
func (gs *GossipSubRouter) pxConnect(peers []*pb.PeerInfo) { func (gs *GossipSubRouter) pxConnect(peers []*pb.PeerInfo) {
if len(peers) > GossipSubPrunePeers { if len(peers) > GossipSubPrunePeers {
shufflePeerInfo(peers)
peers = peers[:GossipSubPrunePeers] peers = peers[:GossipSubPrunePeers]
} }
@ -898,3 +899,10 @@ func shufflePeers(peers []peer.ID) {
peers[i], peers[j] = peers[j], peers[i] 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]
}
}