Merge branch 'develop' into feature/auth

proxyserver/main.go

@@ -15,6 +15,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/gorilla/websocket"
 	"gopkg.in/yaml.v2"
 )
 
@@ -115,6 +116,7 @@ type Backend struct {
 	HttpAddress string
 	Client      *http.Client
 	Handler     http.HandlerFunc
+	Dialer      *websocket.Dialer
 }
 
 func NewBackend(mux *http.ServeMux, httpAddress string, muxPaths ...string) *Backend {
@@ -150,6 +152,60 @@ func NewBackend(mux *http.ServeMux, httpAddress string, muxPaths ...string) *Bac
 
 	// 创建处理请求的函数
 	handleRequest := func(w http.ResponseWriter, r *http.Request) {
+
+		if websocket.IsWebSocketUpgrade(r) {
+			//todo: 建立websocket的代理
+
+			target := url.URL{Scheme: "ws", Host: strings.Split(backend.HttpAddress, "//")[1], Path: r.URL.Path}
+
+			var transfer = func(src, dest *websocket.Conn) {
+				for {
+					mType, msg, err := src.ReadMessage()
+					if err != nil {
+						log.Println(err)
+						break
+					}
+
+					err = dest.WriteMessage(mType, msg)
+					if err != nil {
+						log.Println(err)
+						break
+					}
+				}
+
+				src.Close()
+				dest.Close()
+			}
+
+			header := r.Header.Clone()
+			// log.Println(target.String())
+			header.Del("Sec-Websocket-Extensions")
+			header.Del("Upgrade")
+			header.Del("Sec-Websocket-Key")
+			header.Del("Sec-Websocket-Version")
+			header.Del("Connection")
+			// header.Del("Origin")
+			proxyConn, _, err := backend.Dialer.Dial(target.String(), header)
+			if err != nil {
+				http.Error(w, err.Error(), http.StatusInternalServerError)
+				return
+			}
+			// defer proxyConn.Close()
+
+			upgrader := websocket.Upgrader{
+				CheckOrigin: func(r *http.Request) bool { return true },
+			}
+			conn, err := upgrader.Upgrade(w, r, nil)
+			if err != nil {
+				return
+			}
+			// defer conn.Close()
+
+			go transfer(proxyConn, conn)
+			// go transfer(conn, proxyConn)
+			return
+		}
+
 		// 解析目标URL，包含了查询参数
 		targetURL, err := url.Parse(httpAddress + r.URL.String())
 		if err != nil {