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-07-26 18:00:46 +08:00
parent 9a8bdcb5ba
commit fdaddf9184
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
constants/websocket.go
View File

@ -8,8 +8,6 @@ const (
WEBSOCKET_UNAUTH = "WEBSOCKET_UNAUTH" WEBSOCKET_UNAUTH = "WEBSOCKET_UNAUTH"
//ws连接成功 //ws连接成功
WEBSOCKET_CONNECT_SUCCESS = "WEBSOCKET_CONNECT_SUCCESS" WEBSOCKET_CONNECT_SUCCESS = "WEBSOCKET_CONNECT_SUCCESS"
//心跳信息
WEBSOCKET_HEARTBEAT = "WEBSOCKET_HEARTBEAT"
//图片渲染 //图片渲染
WEBSOCKET_RENDER_IMAGE = "WEBSOCKET_RENDER_IMAGE" WEBSOCKET_RENDER_IMAGE = "WEBSOCKET_RENDER_IMAGE"
//数据格式错误 //数据格式错误

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

@ -142,12 +142,12 @@ func (l *DataTransferLogic) checkAuth(svcCtx *svc.ServiceContext, r *http.Reques
// 心跳 // 心跳
func (w *wsConnectItem) heartbeat() { func (w *wsConnectItem) heartbeat() {
tick := time.Tick(time.Second * 10)
for { for {
time.Sleep(time.Second * 10)
select { select {
case <-w.closeChan: case <-w.closeChan:
return return
default: case <-tick:
//发送心跳信息 //发送心跳信息
if err := w.conn.WriteMessage(websocket.PongMessage, nil); err != nil { if err := w.conn.WriteMessage(websocket.PongMessage, nil); err != nil {
logx.Error("发送心跳信息异常，关闭连接:", w.flag, err) logx.Error("发送心跳信息异常，关闭连接:", w.flag, err)