fix

server/websocket/internal/logic/datatransferlogic.go

@@ -116,12 +116,14 @@ type wsConnectItem struct {
 
 // 请求建立连接，升级websocket协议
 func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) {
-	//判断是不是允许的origin
-	/*if _, ok := mapAllowOrigin[r.Header.Get("Origin")]; !ok {
-		upgrader.CheckOrigin = func(r *http.Request) bool {
+	origin := r.Header.Get("Origin")
+	//判断是不是允许的host
+	upgrader.CheckOrigin = func(r *http.Request) bool {
+		if _, ok := mapAllowOrigin[origin]; !ok {
 			return false
 		}
-	}*/
+		return true
+	}
 	//把子协议携带的token设置到标准token头信息中
 	token := r.Header.Get("Sec-Websocket-Protocol")
 	//有token是正常用户，无则是白板用户，也可以连接