change OpenStream to accept a context

libp2p · Dec 17, 2020 · ba56ac2 · ba56ac2
1 parent 3123af3
commit ba56ac2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/conn.go b/conn.go
@@ -37,8 +37,8 @@ func (c *conn) IsClosed() bool {
 }
 
 // OpenStream creates a new stream.
-func (c *conn) OpenStream() (mux.MuxedStream, error) {
-	qstr, err := c.sess.OpenStreamSync(context.Background())
+func (c *conn) OpenStream(ctx context.Context) (mux.MuxedStream, error) {
+	qstr, err := c.sess.OpenStreamSync(ctx)
 	return &stream{Stream: qstr}, err
 }
 

diff --git a/conn_test.go b/conn_test.go
@@ -130,7 +130,7 @@ var _ = Describe("Connection", func() {
 		Expect(err).ToNot(HaveOccurred())
 		defer serverConn.Close()
 
-		str, err := conn.OpenStream()
+		str, err := conn.OpenStream(context.Background())
 		Expect(err).ToNot(HaveOccurred())
 		_, err = str.Write([]byte("foobar"))
 		Expect(err).ToNot(HaveOccurred())
@@ -251,7 +251,7 @@ var _ = Describe("Connection", func() {
 			for _, c := range []tpt.CapableConn{serverConn1, serverConn2} {
 				go func(conn tpt.CapableConn) {
 					defer GinkgoRecover()
-					str, err := conn.OpenStream()
+					str, err := conn.OpenStream(context.Background())
 					Expect(err).ToNot(HaveOccurred())
 					defer str.Close()
 					_, err = str.Write(data)
@@ -315,7 +315,7 @@ var _ = Describe("Connection", func() {
 			defer GinkgoRecover()
 			conn, err := ln.Accept()
 			Expect(err).ToNot(HaveOccurred())
-			str, err := conn.OpenStream()
+			str, err := conn.OpenStream(context.Background())
 			Expect(err).ToNot(HaveOccurred())
 			str.Write([]byte("foobar"))
 		}()