fix

constants/websocket.go

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

server/websocket/internal/logic/datatransferlogic.go

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