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

Add peers back to gossipsub table, slow down heartbeat (#256) 

* Add peers back to gossipsub table, slow down heartbeat

* exclude on unsub from mesh and fanout
This commit is contained in:
Giovanni Petrantoni 2020-07-09 02:06:26 +09:00 committed by GitHub
parent a52763cc6d
commit fec507e755
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -168,6 +168,7 @@ proc rebalanceMesh(g: GossipSub, topic: string) {.async.} =
# send a graft message to the peer # send a graft message to the peer
await p.sendPrune(@[topic]) await p.sendPrune(@[topic])
g.mesh[topic].excl(id) g.mesh[topic].excl(id)
g.gossipsub[topic].incl(id)
libp2p_gossipsub_peers_per_topic_gossipsub libp2p_gossipsub_peers_per_topic_gossipsub
.set(g.gossipsub.getOrDefault(topic).len.int64, .set(g.gossipsub.getOrDefault(topic).len.int64,
@ -274,7 +275,7 @@ proc heartbeat(g: GossipSub) {.async.} =
except CatchableError as exc: except CatchableError as exc:
trace "exception ocurred in gossipsub heartbeat", exc = exc.msg trace "exception ocurred in gossipsub heartbeat", exc = exc.msg
await sleepAsync(1.seconds) await sleepAsync(5.seconds)
method handleDisconnect*(g: GossipSub, peer: PubSubPeer) = method handleDisconnect*(g: GossipSub, peer: PubSubPeer) =
## handle peer disconnects ## handle peer disconnects
@ -323,6 +324,10 @@ method subscribeTopic*(g: GossipSub,
trace "removing subscription for topic", peer = peerId, name = topic trace "removing subscription for topic", peer = peerId, name = topic
# unsubscribe remote peer from the topic # unsubscribe remote peer from the topic
g.gossipsub[topic].excl(peerId) g.gossipsub[topic].excl(peerId)
if peerId in g.mesh.getOrDefault(topic):
g.mesh[topic].excl(peerId)
if peerId in g.fanout.getOrDefault(topic):
g.fanout[topic].excl(peerId)
libp2p_gossipsub_peers_per_topic_gossipsub libp2p_gossipsub_peers_per_topic_gossipsub
.set(g.gossipsub[topic].len.int64, labelValues = [topic]) .set(g.gossipsub[topic].len.int64, labelValues = [topic])
@ -362,6 +367,7 @@ proc handlePrune(g: GossipSub, peer: PubSubPeer, prunes: seq[ControlPrune]) =
if prune.topicID in g.mesh: if prune.topicID in g.mesh:
g.mesh[prune.topicID].excl(peer.id) g.mesh[prune.topicID].excl(peer.id)
g.gossipsub[prune.topicID].incl(peer.id)
libp2p_gossipsub_peers_per_topic_mesh libp2p_gossipsub_peers_per_topic_mesh
.set(g.mesh[prune.topicID].len.int64, labelValues = [prune.topicID]) .set(g.mesh[prune.topicID].len.int64, labelValues = [prune.topicID])