add a context to OpenStream and NewStream (#172)

* add a context to MuxedConn.OpenStream * add a context to Conn.NewStream
libp2p · Dec 19, 2020 · 83ac1d3 · 83ac1d3
1 parent a39b84e
commit 83ac1d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mux/mux.go b/mux/mux.go
@@ -5,6 +5,7 @@
 package mux
 
 import (
+	"context"
 	"errors"
 	"io"
 	"net"
@@ -74,7 +75,7 @@ type MuxedConn interface {
 	IsClosed() bool
 
 	// OpenStream creates a new stream.
-	OpenStream() (MuxedStream, error)
+	OpenStream(context.Context) (MuxedStream, error)
 
 	// AcceptStream accepts a stream opened by the other side.
 	AcceptStream() (MuxedStream, error)

diff --git a/network/conn.go b/network/conn.go
@@ -1,6 +1,7 @@
 package network
 
 import (
+	"context"
 	"io"
 
 	ic "github.com/libp2p/go-libp2p-core/crypto"
@@ -24,7 +25,7 @@ type Conn interface {
 	ID() string
 
 	// NewStream constructs a new Stream over this conn.
-	NewStream() (Stream, error)
+	NewStream(context.Context) (Stream, error)
 
 	// GetStreams returns all open streams over this conn.
 	GetStreams() []Stream