only count an outbound connection if it has a pubsub stream

This commit is contained in:
vyzo 2020-05-18 18:38:45 +03:00
parent 09e6d9e1e3
commit 4207beae0d
1 changed files with 13 additions and 2 deletions

View File

@ -356,10 +356,21 @@ func (gs *GossipSubRouter) AddPeer(p peer.ID, proto protocol.ID) {
// track the connection direction
outbound := false
conns := gs.p.host.Network().ConnsToPeer(p)
loop:
for _, c := range conns {
if c.Stat().Direction == network.DirOutbound {
// only count the connection if it has a pubsub stream
for _, s := range c.GetStreams() {
switch s.Protocol() {
case FloodSubID:
fallthrough
case GossipSubID_v10:
fallthrough
case GossipSubID_v11:
outbound = true
break
break loop
}
}
}
}
gs.outbound[p] = outbound