Spring Boot + WebSocket 实时监控异常

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

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

WebSocket就不做介绍了，上链接：developer.mozilla.org/zh-CN/docs/…

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

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

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

实现

前端：

直接贴代码

1   2   3       4           5           6       7      29      30         
 31             
 32                 {{item.id}}.{{item.name}} 33                  34             
 35 
 36  37  38 117 

后端：

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

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

2、配置application.yml

#端口server:  port: 18801
#密码，因为接口不需要权限，所以加了个密码做校验mySocket:  myPwd: jae_123复制代码3、WebSocketConfig配置类 1 @Configuration 2 public class WebSocketConfig { 3  4     /** 5      * 注入一个ServerEndpointExporter,该Bean会自动注册使用@ServerEndpoint注解申明的websocket endpoint 6      */ 7     @Bean 8     public ServerEndpointExporter serverEndpointExporter(){ 9         return new ServerEndpointExporter();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 sessionPools = new CopyOnWriteArraySet();17 18     /**19      * 有客户端连接成功20      */21     @OnOpen22     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     @OnClose32     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     @OnError64     public void onError(Session session, Throwable throwable){65         log.error("发生错误");66         throwable.printStackTrace();67     }68 69 }

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

1 @RestController 2 @RequestMapping("/open/socket") 3 public class WebSocketController { 4  5     @Value("${mySocket.myPwd}") 6     public String myPwd; 7  8     @Autowired 9     private WebSocketServer webSocketServer;10 11     /**12      * 手机客户端请求接口13      * @param id    发生异常的设备ID14      * @param pwd   密码（实际开发记得加密）15      * @throws IOException16      */17     @PostMapping(value = "/onReceive")18     public void onReceive(String id,String pwd) throws IOException {19         if(pwd.equals(myPwd)){  //密码校验一致（这里举例，实际开发还要有个密码加密的校验的），则进行群发20             webSocketServer.broadCastInfo(id);21         }22     }23 24 }

测试

1、打开前端页面，进行WebSocket连接

控制台输出，连接成功

2、因为是模拟数据，所以全部显示正常，没有异常提交时的页面呈现

3、接下来，我们用接口测试工具Postman提交一个异常

注意id为3的这个数据的状态变化

我们可以看到，id为3的王五状态已经变成异常的了，实时通讯成功。

最后

工作中有这方面关于实时监控的需求，可以参考一下哦。