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 16:05:18 +08:00
parent 6c94bc29c9
commit 292f2f1ca3
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -118,12 +118,16 @@ func (l *DataTransferLogic) DataTransfer(req *types.DataTransferReq, w http.Resp
case 2:
token = strings.Trim(tokenSlice[0], " ")
debugToken = strings.Trim(tokenSlice[1], " ")
r.Header.Set("Debug-Token", debugToken)
default:
logx.Error("invalid ws token:", tokens)
return
}
r.Header.Set("Authorization", "Bearer "+token)
if token != "empty_token" && token != "" {
r.Header.Set("Authorization", "Bearer "+token)
}
if debugToken != "empty_debug_token" && debugToken != "" {
r.Header.Set("Debug-Token", debugToken)
}
//设置Sec-Websocket-Protocol
upgrader.Subprotocols = []string{tokens}
}