fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
laodaming 2023-10-18 15:28:47 +08:00
parent 6de13a61f7
commit 22f9bf7165
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
server/websocket/internal/logic/datatransferlogic.go
View File

@ -104,14 +104,28 @@ type wsConnectItem struct {
// 请求建立连接，升级websocket协议 // 请求建立连接，升级websocket协议
func (l *DataTransferLogic) DataTransfer(req *types.DataTransferReq, w http.ResponseWriter, r *http.Request) { func (l *DataTransferLogic) DataTransfer(req *types.DataTransferReq, w http.ResponseWriter, r *http.Request) {
//把子协议携带的token设置到标准token头信息中 //把子协议携带的token设置到标准token头信息中
token := r.Header.Get("Sec-Websocket-Protocol") tokens := r.Header.Get("Sec-Websocket-Protocol")
oldWid := req.Wid oldWid := req.Wid
oldWid = strings.Trim(oldWid, " ") oldWid = strings.Trim(oldWid, " ")
//有token是正常用户，无则是白板用户，也可以连接 //有token是正常用户，无则是白板用户，也可以连接
if token != "" { if tokens != "" {
token := ""
debugToken := ""
tokenSlice := strings.Split(tokens, ",")
switch len(tokenSlice) {
case 1:
token = strings.Trim(tokenSlice[0], " ")
case 2:
token = strings.Trim(tokenSlice[0], " ")
debugToken = strings.Trim(tokenSlice[1], " ")
r.Header.Set("Debug-Token", "Bearer "+debugToken)
default:
logx.Error("invalid ws token:", tokens)
return
}
r.Header.Set("Authorization", "Bearer "+token) r.Header.Set("Authorization", "Bearer "+token)
//设置Sec-Websocket-Protocol //设置Sec-Websocket-Protocol
upgrader.Subprotocols = []string{token} upgrader.Subprotocols = []string{tokens}
} }
//判断下是否火狐浏览器(获取浏览器第一条消息返回有收不到的bug需要延迟1秒) //判断下是否火狐浏览器(获取浏览器第一条消息返回有收不到的bug需要延迟1秒)
userAgent := r.Header.Get("User-Agent") userAgent := r.Header.Get("User-Agent")