支持 js websocket

pkg/utils/resp.go

@@ -1,6 +1,12 @@
 package utils
 
-import "net/http"
+import (
+	"bufio"
+	"net"
+	"net/http"
+
+	"github.com/pkg/errors"
+)
 
 type WrittenResponseWriter struct {
 	write bool
@@ -18,6 +24,14 @@ func (w *WrittenResponseWriter) Header() http.Header {
 	return w.base.Header()
 }
 
+func (w *WrittenResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	w.write = true
+	if hijacker, ok := w.base.(http.Hijacker); ok {
+		return hijacker.Hijack()
+	}
+	return nil, nil, errors.New("not hijackable")
+}
+
 func (w *WrittenResponseWriter) Write(b []byte) (int, error) {
 	w.write = true
 	return w.base.Write(b)