diff --git a/mux/mux.go b/mux/mux.go index 736da160e500ede2d0601f743f2e5d839a85e012..63d83b4992e5d936f67caaa7273ca38639be3a37 100644 --- 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 index b78f53cda58f43c669b9b9bf8cf5f169e7af4993..25f0868467103191de0efb7ae14073363aba3d19 100644 --- 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