This commit is contained in:
laodaming 2023-10-20 14:10:03 +08:00
parent 3b8773f4de
commit a58cccfff6
3 changed files with 13 additions and 1 deletions

View File

@ -9,6 +9,9 @@ const (
WEBSOCKET_CONNECT_SUCCESS Websocket = "WEBSOCKET_CONNECT_SUCCESS" //ws连接成功 1级消息单向通信 WEBSOCKET_CONNECT_SUCCESS Websocket = "WEBSOCKET_CONNECT_SUCCESS" //ws连接成功 1级消息单向通信
) )
// 心跳
const WEBSOCKET_HEARTBEAT Websocket = "WEBSOCKET_HEARTBEAT"
// websocket消息类型(通用通知类别) // websocket消息类型(通用通知类别)
const ( const (
WEBSOCKET_COMMON_NOTIFY Websocket = "WEBSOCKET_COMMON_NOTIFY" //通用回调通知1级消息单向通信 WEBSOCKET_COMMON_NOTIFY Websocket = "WEBSOCKET_COMMON_NOTIFY" //通用回调通知1级消息单向通信

View File

@ -322,7 +322,10 @@ func (w *wsConnectItem) heartbeat() {
w.debug = nil w.debug = nil
} }
//发送心跳信息 //发送心跳信息
if err := w.conn.WriteMessage(websocket.PongMessage, nil); err != nil { if err := w.conn.WriteMessage(websocket.PongMessage, w.respondDataFormat(constants.WEBSOCKET_HEARTBEAT, websocket_data.HeartBeatMsg{
WsCount: currentWebsocketConnectCount,
CombineCount: currentRequestCombineApiCount,
})); err != nil {
logx.Error("发送心跳信息异常,关闭连接:", w.uniqueId, err) logx.Error("发送心跳信息异常,关闭连接:", w.uniqueId, err)
w.close() w.close()
return return

View File

@ -23,3 +23,9 @@ type ConnectErrMsg struct {
type ConnectUnAuth struct { type ConnectUnAuth struct {
Message string `json:"message"` Message string `json:"message"`
} }
// 心跳数据
type HeartBeatMsg struct {
WsCount int `json:"ws_count"`
CombineCount int `json:"combine_count"`
}