use the proto we already have in AddPeer to determine whether the connection houses the stream

31c9b219 · vyzo · 4207beae · 31c9b219
Commit 31c9b219 authored May 18, 2020 by vyzo
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 6 deletions

gossipsub.go gossipsub.go +1 -6

No files found.
--- a/gossipsub.go
+++ b/gossipsub.go
@@ -361,12 +361,7 @@ loop:
 		if c.Stat().Direction == network.DirOutbound {
 			// only count the connection if it has a pubsub stream
 			for _, s := range c.GetStreams() {
-				switch s.Protocol() {
+				if s.Protocol() == proto {
-				case FloodSubID:
-					fallthrough
-				case GossipSubID_v10:
-					fallthrough
-				case GossipSubID_v11:
 					outbound = true
 					break loop
 				}