位置: IT常识 - 正文
WebSocket开发（心跳监测）功能(websocket())
编辑：rootadmin推荐整理分享WebSocket开发（心跳监测）功能(websocket())，希望有所帮助，仅作参考，欢迎阅读内容。
文章相关热门搜索词:websocket开源库,websocket开源库,websocket 开发,websocket快速入门,websocket快速入门,websocket 开发,websocket 开发,websocket入门教程,内容如对您有帮助，希望把文章链接给更多的朋友！
在之前的文章中完成了客服对话的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@RestControllerpublic class CheckHealthController { @GetMapping("/health") public ResponeApi health() { log.info("健康检查chatroom-IM --> 检查成功！"); return ResponeApi.success(ResponeCodeEnum.SUCCESS,"SUCCESS"); }}效果如下：
上一篇：Window.location用法(windowlocation用法)
下一篇：Vue3.0 项目启动（打造企业级音乐App）(vue3.0启动命令)
