This commit is contained in:
laodaming 2023-09-05 14:07:33 +08:00
parent 290859eaa2
commit 1ba6a0ed64

View File

@ -117,13 +117,14 @@ type wsConnectItem struct {
// 请求建立连接升级websocket协议
func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) {
origin := r.Header.Get("Origin")
fmt.Println("*************:", origin)
//判断是不是允许的host
upgrader.CheckOrigin = func(r *http.Request) bool {
/*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是正常用户无则是白板用户也可以连接