From ba422e8a40f0fd1adfe749193c884c6b7cf12435 Mon Sep 17 00:00:00 2001
From: Zach Marcantel <zmarcantel@utexas.edu>
Date: Fri, 11 Jul 2014 18:56:45 -0500
Subject: [PATCH] add .Hijack() to ResponseWriter

---
 examples/websocket/example_websocket.go | 58 +++++++++++++++++++++++++
 response_writer.go                      | 12 +++++
 2 files changed, 70 insertions(+)
 create mode 100644 examples/websocket/example_websocket.go

diff --git a/examples/websocket/example_websocket.go b/examples/websocket/example_websocket.go
new file mode 100644
index 00000000..e164d4ad
--- /dev/null
+++ b/examples/websocket/example_websocket.go
@@ -0,0 +1,58 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"strconv"
+
+	"code.google.com/p/go.net/websocket"
+
+	"github.com/gin-gonic/gin"
+)
+
+func main() {
+	r := gin.New()
+	r.GET("/", func(c *gin.Context) {
+		handler := websocket.Handler(func(conn *websocket.Conn) {
+			// This is the simplest possible echoserver
+			//
+			// For a more advanced handler you can use the
+			// conn.Read and conn.Write methods as the
+			// websocket.Conn type conforms to io.Reader+io.Writer
+
+			io.Copy(conn, conn)
+		})
+		handler.ServeHTTP(&c.Writer, c.Req)
+	})
+
+	go r.Run(":8080")
+
+	lock := make(chan bool)
+	go testServer(100, lock)
+	<-lock
+}
+
+func testServer(count int, done chan bool) {
+	client, err := websocket.Dial("ws://localhost:8080", "", "http://localhost/")
+	if err != nil {
+		panic(err)
+	}
+
+	for i := 0; i < count; i++ {
+		out := []byte(strconv.Itoa(i))
+		_, err = client.Write(out)
+		if err != nil {
+			panic(err)
+		}
+		fmt.Printf("Sent: %s\n", out)
+
+		var in = make([]byte, 512)
+		_, err = client.Read(in)
+		if err != nil {
+			panic(err)
+		}
+		fmt.Printf("Received: %s\n\n", in)
+	}
+
+	done <- true
+}
diff --git a/response_writer.go b/response_writer.go
index ee6bbe8a..6cf7d968 100644
--- a/response_writer.go
+++ b/response_writer.go
@@ -1,6 +1,9 @@
 package gin
 
 import (
+	"bufio"
+	"errors"
+	"net"
 	"net/http"
 )
 
@@ -46,3 +49,12 @@ func (w *ResponseWriter) Status() int {
 func (w *ResponseWriter) WasWritten() bool {
 	return w.status == StatusUnset
 }
+
+// allow connection hijacking
+func (w *ResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	hijacker, ok := w.ResponseWriter.(http.Hijacker)
+	if !ok {
+		return nil, nil, errors.New("the ResponseWriter doesn't support the Hijacker interface")
+	}
+	return hijacker.Hijack()
+}