fix swarm transient conn (#241)

3563ed1f · Aarsh Shah · GitHub · f82c4250 · 3563ed1f · 3563ed1f
Unverified Commit 3563ed1f authored Feb 19, 2021 by Aarsh Shah Committed by GitHub Feb 19, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 6 deletions

swarm.go swarm.go +0 -6

swarm_conn.go swarm_conn.go +6 -0

No files found.
--- a/swarm.go
+++ b/swarm.go
@@ -360,12 +360,6 @@ func (s *Swarm) NewStream(ctx context.Context, p peer.ID) (network.Stream, error
 			}
 		}

-		if c.Stat().Transient {
-			if useTransient, _ := network.GetUseTransient(ctx); !useTransient {
-				return nil, network.ErrTransientConn
-			}
-		}
-
 		s, err := c.NewStream(ctx)
 		if err != nil {
 			if c.conn.IsClosed() {

--- a/swarm_conn.go
+++ b/swarm_conn.go
@@ -177,6 +177,12 @@ func (c *Conn) Stat() network.Stat {

 // NewStream returns a new Stream from this connection
 func (c *Conn) NewStream(ctx context.Context) (network.Stream, error) {
+	if c.Stat().Transient {
+		if useTransient, _ := network.GetUseTransient(ctx); !useTransient {
+			return nil, network.ErrTransientConn
+		}
+	}
+
 	ts, err := c.conn.OpenStream(ctx)

 	if err != nil {