treat transient conns as opt-in for opening new streams

0cf68a57 · vyzo · 746f908d · 0cf68a57
Commit 0cf68a57 authored Feb 04, 2021 by vyzo
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

swarm.go swarm.go +7 -0

No files found.
--- a/swarm.go
+++ b/swarm.go
@@ -351,6 +351,13 @@ func (s *Swarm) NewStream(ctx context.Context, p peer.ID) (network.Stream, error
 				return nil, err
 			}
 		}
+
+		if c.Stat().Transient {
+			if !network.GetUseTransient(ctx) {
+				return nil, network.ErrTransientConn
+			}
+		}
+
 		s, err := c.NewStream(ctx)
 		if err != nil {
 			if c.conn.IsClosed() {