Browse Source

Adding method to get stream as net.Conn

Armon Dadgar 10 years ago
parent
commit
fa0f2a5f8d
2 changed files with 9 additions and 4 deletions
  1. 7 2
      session.go
  2. 2 2
      session_test.go

+ 7 - 2
session.go

@@ -101,8 +101,13 @@ func (s *Session) IsClosed() bool {
 	}
 }
 
-// Open is used to create a new stream
-func (s *Session) Open() (*Stream, error) {
+// Open is used to create a new stream as a net.Conn
+func (s *Session) Open() (net.Conn, error) {
+	return s.OpenStream()
+}
+
+// OpenStream is used to create a new stream
+func (s *Session) OpenStream() (*Stream, error) {
 	if s.IsClosed() {
 		return nil, ErrSessionShutdown
 	}

+ 2 - 2
session_test.go

@@ -107,7 +107,7 @@ func TestAccept(t *testing.T) {
 
 	go func() {
 		defer wg.Done()
-		stream, err := server.Open()
+		stream, err := server.OpenStream()
 		if err != nil {
 			t.Fatalf("err: %v", err)
 		}
@@ -121,7 +121,7 @@ func TestAccept(t *testing.T) {
 
 	go func() {
 		defer wg.Done()
-		stream, err := client.Open()
+		stream, err := client.OpenStream()
 		if err != nil {
 			t.Fatalf("err: %v", err)
 		}