Spring Boot + WebSocket 实时监控异常

本文已经收录到Github仓库，该仓库包含计算机基础、Java基础、多线程、JVM、数据库、Redis、Spring、Mybatis、SpringMVC、SpringBoot、分布式、微服务、设计模式、架构、校招社招分享等核心知识点，欢迎star~

Github地址：https://github.com/Tyson0314/Java-learning

写在前面

此异常非彼异常，标题所说的异常是业务上的异常。

最近做了一个需求，消防的设备巡检，如果巡检发现异常，通过手机端提交，后台的实时监控页面实时获取到该设备的信息及位置，然后安排员工去处理。

因为需要服务端主动向客户端发送消息，所以很容易的就想到了用WebSocket来实现这一功能。

前端略微复杂，需要在一张位置分布图上进行鼠标描点定位各个设备和根据不同屏幕大小渲染，本文不做介绍，只是简单地用页面样式进行效果呈现。

绿色代表正常，红色代表异常

预期效果，未接收到请求前----->id为3的提交了异常，id为3的王五变成了红色

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SheQMM2i-1675344624892)(http://img.topjavaer.cn/img/springboot-websocket异常监控01.png)]

实现

前端：

直接贴代码

  1 <!DOCTYPEhtml>
  2 <html>
  3     <head>
  4         <metacharset="utf-8"/>
  5         <title>实时监控</title>
  6     </head>
  7     <style>8         .item{
  9             display: flex;
 10             border-bottom: 1px solid #000000;
 11             justify-content: space-between;
 12             width: 30%;
 13             line-height: 50px;
 14             height: 50px;
 15         }16         
 17         .item span:nth-child(2){
 18             margin-right: 10px;
 19             margin-top: 15px;
 20             width: 20px;
 21             height: 20px;
 22             border-radius: 50%;
 23             background: #55ff00;
 24         }25         .nowI{
 26             background: #ff0000 !important;
 27         }
 28     </style>
 29     <body>
 30         <divid="app">
 31             <divv-for="item in list"class="item">
 32                 <span>{{item.id}}.{{item.name}}</span>
 33                 <span:class='item.state==-1?"nowI":""'></span>
 34             </div>
 35         </div>
 36     </body>
 37     <scriptsrc="./js/vue.min.js"></script>
 38     <scripttype="text/javascript">39var vm =newVue({40el:"#app",41data:{42list:[{43id:1,44name:'张三',45state:146},47{48id:2,49name:'李四',50state:151},52{53id:3,54name:'王五',55state:156},57{58id:4,59name:'韩梅梅',60state:161},62{63id:5,64name:'李磊',65state:166},67]68}69})7071var webSocket =null;72if('WebSocket'in window){73//创建WebSocket对象74             webSocket =newWebSocket("ws://localhost:18801/webSocket/"+getUUID());7576//连接成功77             webSocket.onopen=function(){78                 console.log("已连接");79                 webSocket.send("消息发送测试")80}81//接收到消息82             webSocket.onmessage=function(msg){83//处理消息84var serverMsg = msg.data;85var t_id =parseInt(serverMsg)//服务端发过来的消息，ID，string需转化为int类型才能比较86for(var i =0; i < vm.list.length; i++){87var item = vm.list[i];88if(item.id == t_id){89                         item.state =-1;90                         vm.list.splice(i,1,item)91break;92}93}94};9596//关闭事件97             webSocket.onclose=function(){98                 console.log("websocket已关闭");99};100//发生了错误事件101             webSocket.onerror=function(){102                 console.log("websocket发生了错误");103}104}else{105alert("很遗憾，您的浏览器不支持WebSocket！")106}107108functiongetUUID(){//获取唯一的UUID109return'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g,function(c){110111var r = Math.random()*16|0,112                     v = c =='x'? r :(r &0x3|0x8);113return v.toString(16);114});115}116</script>
117 </html>
复制代码

后端：

项目结构是这样子的，后面的代码关键注释都有，就不重复描述了

1、新建SpringBoot工程，选择web和WebSocket依赖

2、配置application.yml

#端口
server:
  port: 18801

#密码，因为接口不需要权限，所以加了个密码做校验
mySocket:
  myPwd: jae_123
复制代码

3、WebSocketConfig配置类

1@Configuration2publicclassWebSocketConfig{34/**
 5      * 注入一个ServerEndpointExporter,该Bean会自动注册使用@ServerEndpoint注解申明的websocket endpoint
 6      */7@Bean8public ServerEndpointExporter serverEndpointExporter(){9returnnewServerEndpointExporter();10}11}
复制代码

4、WebSocketServer类，用来进行服务端和客户端之间的交互

 1 /**
 2  * @author jae
 3  * @ServerEndpoint("/webSocket/{uid}") 前端通过此URI与后端建立链接
 4  */
 5 
 6 @ServerEndpoint("/webSocket/{uid}")
 7 @Component
 8 public class WebSocketServer {
 9 
10     private static Logger log = LoggerFactory.getLogger(WebSocketServer.class);
11 
12     //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
13     private static final AtomicInteger onlineNum = new AtomicInteger(0);
14 
15     //concurrent包的线程安全Set，用来存放每个客户端对应的WebSocketServer对象。
16     private static CopyOnWriteArraySet<Session> sessionPools = new CopyOnWriteArraySet<Session>();
17 
18     /**
19      * 有客户端连接成功
20      */
21     @OnOpen
22     public void onOpen(Session session, @PathParam(value = "uid") String uid){
23         sessionPools.add(session);
24         onlineNum.incrementAndGet();
25         log.info(uid + "加入webSocket！当前人数为" + onlineNum);
26     }
27 
28     /**
29      * 连接关闭调用的方法
30      */
31     @OnClose
32     public void onClose(Session session) {
33         sessionPools.remove(session);
34         int cnt = onlineNum.decrementAndGet();
35         log.info("有连接关闭，当前连接数为：{}", cnt);
36     }
37 
38     /**
39      * 发送消息
40      */
41     public void sendMessage(Session session, String message) throws IOException {
42         if(session != null){
43             synchronized (session) {
44                 session.getBasicRemote().sendText(message);
45             }
46         }
47     }
48 
49     /**
50      * 群发消息
51      */
52     public void broadCastInfo(String message) throws IOException {
53         for (Session session : sessionPools) {
54             if(session.isOpen()){
55                 sendMessage(session, message);
56             }
57         }
58     }
59 
60     /**
61      * 发生错误
62      */
63     @OnError
64     public void onError(Session session, Throwable throwable){
65         log.error("发生错误");
66         throwable.printStackTrace();
67     }
68 
69 }
复制代码

5、WebSocketController类，用于进行接口测试

1@RestController2@RequestMapping("/open/socket")3publicclass WebSocketController {45@Value("${mySocket.myPwd}")6public String myPwd;78@Autowired9private WebSocketServer webSocketServer;1011/**
12      * 手机客户端请求接口
13      * @param id    发生异常的设备ID
14      * @param pwd   密码（实际开发记得加密）
15      * @throws IOException
16      */17@PostMapping(value ="/onReceive")18public void onReceive(String id,String pwd) throws IOException {19if(pwd.equals(myPwd)){//密码校验一致（这里举例，实际开发还要有个密码加密的校验的），则进行群发20             webSocketServer.broadCastInfo(id);21}22}2324}
复制代码