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-08-23 12:29:59 +08:00
parent 41caa6db64
commit a72aa2feca
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -45,7 +45,7 @@ var (
}, },
} }
//升级websocket //升级websocket
upgrade = websocket.Upgrader{ upgrader = websocket.Upgrader{
//最大可读取大小 10M //最大可读取大小 10M
ReadBufferSize: 1024 * 10, ReadBufferSize: 1024 * 10,
//握手超时时间15s //握手超时时间15s
@ -86,9 +86,9 @@ func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request)
token := r.Header.Get("Sec-Websocket-Protocol") token := r.Header.Get("Sec-Websocket-Protocol")
r.Header.Set("Authorization", "Bearer "+token) r.Header.Set("Authorization", "Bearer "+token)
//设置Sec-Websocket-Protocol //设置Sec-Websocket-Protocol
upgrade.Subprotocols = []string{token} upgrader.Subprotocols = []string{token}
//升级websocket //升级websocket
conn, err := upgrade.Upgrade(w, r, nil) conn, err := upgrader.Upgrade(w, r, nil)
if err != nil { if err != nil {
logx.Error("http upgrade websocket err:", err) logx.Error("http upgrade websocket err:", err)
return return