Implement HTTP hijacker for switching from HTTP to WebSocket

middleware/hijack.go

@@ -0,0 +1,30 @@
+package middleware
+
+import (
+	"bufio"
+	"net"
+	"net/http"
+)
+
+type HijackWriter struct {
+	conn net.Conn
+}
+
+func NewHijackWriter(conn net.Conn) http.ResponseWriter {
+	return &HijackWriter{conn: conn}
+}
+
+func (rw *HijackWriter) Header() http.Header {
+	return http.Header{}
+}
+
+func (rw *HijackWriter) Write(bytes []byte) (int, error) {
+	return rw.conn.Write(bytes)
+}
+
+func (rw *HijackWriter) WriteHeader(statusCode int) {
+}
+
+func (rw *HijackWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	return rw.conn, bufio.NewReadWriter(bufio.NewReader(rw.conn), bufio.NewWriter(rw.conn)), nil
+}