vacp2p
/
nim-libp2p
mirror of https://github.com/vacp2p/nim-libp2p.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

avoid creating prune message unless we're pruning (#487)

This commit is contained in:
Jacek Sieka 2020-12-15 22:46:03 +01:00 committed by GitHub
parent ea6988d380
commit 9e5ba64c48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libp2p/protocols/pubsub/gossipsub.nim
View File

@ -812,7 +812,7 @@ proc heartbeat(g: GossipSub) {.async.} =
let gossipPeers = g.gossipsub.getOrDefault(t) let gossipPeers = g.gossipsub.getOrDefault(t)
# this will be changed by rebalance but does not matter # this will be changed by rebalance but does not matter
totalMeshPeers += meshPeers.len totalMeshPeers += meshPeers.len
totalGossipPeers += gossipPeers.len totalGossipPeers += g.gossipsub.peers(t)
var prunes: seq[PubSubPeer] var prunes: seq[PubSubPeer]
for peer in meshPeers: for peer in meshPeers:
if peer.score < 0.0: if peer.score < 0.0:
@ -820,12 +820,13 @@ proc heartbeat(g: GossipSub) {.async.} =
g.pruned(peer, t) g.pruned(peer, t)
g.mesh.removePeer(t, peer) g.mesh.removePeer(t, peer)
prunes &= peer prunes &= peer
let prune = RPCMsg(control: some(ControlMessage( if prunes.len > 0:
prune: @[ControlPrune( let prune = RPCMsg(control: some(ControlMessage(
topicID: t, prune: @[ControlPrune(
peers: g.peerExchangeList(t), topicID: t,
backoff: g.parameters.pruneBackoff.seconds.uint64)]))) peers: g.peerExchangeList(t),
g.broadcast(prunes, prune) backoff: g.parameters.pruneBackoff.seconds.uint64)])))
g.broadcast(prunes, prune)
g.rebalanceMesh(t) g.rebalanceMesh(t)