diff --git a/server/websocket/internal/logic/datatransferlogic.go b/server/websocket/internal/logic/datatransferlogic.go index 92e264c1..545678bf 100644 --- a/server/websocket/internal/logic/datatransferlogic.go +++ b/server/websocket/internal/logic/datatransferlogic.go @@ -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} }