add a context to OpenStream and NewStream

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)

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