logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-08-11 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

heartbeat doPX is a more complex decision

Browse Source
This commit is contained in:
vyzo 2020-03-02 15:20:23 +02:00
parent f9ce661bda
commit 487bbaf09f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
gossipsub.go
View File

@ -649,7 +649,7 @@ func (gs *GossipSubRouter) heartbeat() {
tograft := make(map[peer.ID][]string) tograft := make(map[peer.ID][]string)
toprune := make(map[peer.ID][]string) toprune := make(map[peer.ID][]string)
doPX := true doPX := make(map[peer.ID]bool)
// clean up expired backoffs // clean up expired backoffs
gs.clearBackoff() gs.clearBackoff()
@ -758,7 +758,7 @@ func (gs *GossipSubRouter) clearBackoff() {
} }
} }
func (gs *GossipSubRouter) sendGraftPrune(tograft, toprune map[peer.ID][]string, doPX bool) { func (gs *GossipSubRouter) sendGraftPrune(tograft, toprune map[peer.ID][]string, doPX map[peer.ID]bool) {
for p, topics := range tograft { for p, topics := range tograft {
graft := make([]*pb.ControlGraft, 0, len(topics)) graft := make([]*pb.ControlGraft, 0, len(topics))
for _, topic := range topics { for _, topic := range topics {
@ -771,7 +771,7 @@ func (gs *GossipSubRouter) sendGraftPrune(tograft, toprune map[peer.ID][]string,
delete(toprune, p) delete(toprune, p)
prune = make([]*pb.ControlPrune, 0, len(pruning)) prune = make([]*pb.ControlPrune, 0, len(pruning))
for _, topic := range pruning { for _, topic := range pruning {
prune = append(prune, gs.makePrune(p, topic, doPX)) prune = append(prune, gs.makePrune(p, topic, doPX[p]))
} }
} }
@ -782,7 +782,7 @@ func (gs *GossipSubRouter) sendGraftPrune(tograft, toprune map[peer.ID][]string,
for p, topics := range toprune { for p, topics := range toprune {
prune := make([]*pb.ControlPrune, 0, len(topics)) prune := make([]*pb.ControlPrune, 0, len(topics))
for _, topic := range topics { for _, topic := range topics {
prune = append(prune, gs.makePrune(p, topic, doPX)) prune = append(prune, gs.makePrune(p, topic, doPX[p]))
} }
out := rpcWithControl(nil, nil, nil, nil, prune) out := rpcWithControl(nil, nil, nil, nil, prune)