WebSocket开发（心跳监测）功能

news2025/2/24 1:08:45

前言

在之前的文章中完成了客服对话的Demo功能，但是现在的连接是无限制的长时间连接没有做心跳、失活、超时断连等功能，心跳的实现方法有很多种，并且WebSocket就提供了ping/pong类型的消息。

心跳的触发方式也分两种：

客户端触发：如果是前端发送心跳，后端需要返回心跳，也就是ping pong的过程会有两次数据传递。
服务端触发：后端来发送心跳的话，就只需要发送ping，前端不需要回应。

这两种后续的处理方式也有各自优缺点。

客户端触发：
- 优点
  - 灵活控制
  - 无需设置主动超时
  - 逻辑清晰
  - 服务端简单
- 缺点
  - 两次消息传递
  - 消息内容容易篡改
服务端触发：
- 优点
  - 节省宽带
  - 服务端控制频率
  - 消息体固定
- 缺点
  - 处理逻辑复杂
  - 需要添加定时任务
  - 考虑稳定性

两种方式各有利弊，看具体的应用场景选择心跳方式是最好的，这里使用客户端触发心跳进行Demo实验，前端变更比较容易，服务端也不需要写定时等处理复杂的业务，只需要在收到固定消息后返回对应消息即可。

1. WebSocket心跳

客户端触发心跳的话就是在服务端的OnMessage事件里进行截获处理，如果是接受参数为String，就在之前的逻辑之上加上判断健康检查的逻辑，功能很简单，客户端发送了特点消息直接返回对应的消息即可。

1.1 字符串消息

WebSocket已经设计了心跳，也就是Ping/Pong，这个功能可以到达检测链接是否可用，但是如果要携带数据还是需要自己用字符串、对象的消息类型进行实现。

代码如下：

	@OnMessage
    public void onMessage(String message, Session session,@PathParam("clientId") String clientId){
        /**
         * 持久化
         */
        baseWebSocketService.saveClientSendMsg(clientId,message,new Date());
        /**
         * 处理消息
         */
        UserMessageModel userMessageModel = JSONObject.parseObject(message, UserMessageModel.class);
        if (userMessageModel == null){
            this.sendMessage(BaseResponseMessage.error(null,"传递参数结构异常"));
        }
        userMessageModel.setSendId(clientId);
        /**
         * 健康检查
         */
        if ("HEALTH".equals(userMessageModel.getMessage())){
            this.sendText(WebSocketHealthEnum.HEALTH.result);
            return;
        }

        /**
         * 发送消息
         */
        HashMap<String,WebSocketClient> hashMap = webSocketClientMap.get(WebSocketTypeEnum.getAcceptType(this.type));
        if (!CollectionUtils.isEmpty(hashMap)){
            if (StringUtils.isEmpty(bindKfClients.get(this.clientId))){
                List<UserMessageModel> list = new ArrayList();
                list.addAll(baseWebSocketService.queryClientSendMsg(clientId));
                list.forEach(model-> {
                    this.toCSucceed(model);
                });
            }else{
                this.toCSucceed(userMessageModel);
            }
        }else{
            baseWebSocketService.saveClientCompensateMsg(userMessageModel.getAcceptId(),message,(byte) 0);
            log.info("客户端:{} 发送消息到接受端:{} 不在线，放置到代发送列表，当前待发送列表:{}条",clientId,userMessageModel.getAcceptId());
            this.sendMessage(BaseResponseMessage.error(null,"接收端不在线"));
        }
    }

如果客户端发送了内容HEALTH则回复对应消息，我这里回复了SUCCESS

在这里插入图片描述

但是这样有个问题，用户发送了HEALTH这个字符串服务端会将这个消息当作健康检查进行处理，而不是消息，这样影响了用户端的使用。

还记得之前预留了一个发送类型字段sendType吗，这时候这个类型就起作用了，如果要做健康检查的操作就将这个sendType设置为HEALTH，服务端根据sendType字段进行判断业务处理，修改一下代码：

        /**
         * 健康检查
         */
        if (WebSocketHealthEnum.HEALTH.msg.equals(userMessageModel.getSendType())){
            this.sendText(WebSocketHealthEnum.HEALTH.result);
            return;
        }

1.2 Ping/Pong消息

Ping的协议头是0x9，Pong的协议头是0xA
控制帧最大载荷为125bytes且不能拆分

服务端可以主动发生Ping/Pong消息，之前文章中写过WebSocket发送消息的四种类型，这里将上面发送Text文本类型换成发送Ping类型的消息，当然也可以发送Pong类型的消息。

代码如下：

        if (WebSocketHealthEnum.HEALTH.msg.equals(userMessageModel.getSendType())){
            try {
                session.getBasicRemote().sendPing(ByteBuffer.wrap("SUCCESS".getBytes()));
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            return;
        }

Ping消息是不会被我们的OnMessage事件接收的，所以不需要特殊处理，如果是Pong消息在服务的接收是可以的。

代码如下：

    @OnMessage
    public void onPong(PongMessage pongMessage) {
        ByteBuffer byteBuffer = pongMessage.getApplicationData();
    }

具体的业务可以二次处理

2. 服务心跳

上面的心跳是对每个客户端的心跳监测，服务的心跳也要做，服务的心跳就简单了，前端定时请求HTTP/HTTPS协议接口。

代码如下：

@Slf4j
@RestController
public class CheckHealthController {

    @GetMapping("/health")
    public ResponeApi health() {
        log.info("健康检查chatroom-IM --> 检查成功！");
        return ResponeApi.success(ResponeCodeEnum.SUCCESS,"SUCCESS");
    }
    
}