From 83ac1d370dfe3f04aee38a904f73e49070be4863 Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Sat, 19 Dec 2020 09:36:59 +0700
Subject: [PATCH] add a context to OpenStream and NewStream (#172)

* add a context to MuxedConn.OpenStream

* add a context to Conn.NewStream
---
 mux/mux.go      | 3 ++-
 network/conn.go | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/mux/mux.go b/mux/mux.go
index 736da16..63d83b4 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 b78f53c..25f0868 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
-- 
GitLab