（代码）springboot webscoket 定时发送消息

本文主要是介绍（代码）springboot webscoket 定时发送消息，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

代码

创建WebSocket端点

@ServerEndpoint(value = "/websocket/{uuid}")
@Component
public class MessageHandleController{private static final Logger logger = LoggerFactory.getLogger(MessageHandleController.class);//concurrent包的线程安全Set，用来存放每个客户端对应的WebSocketController对象。public static CopyOnWriteArraySet<MessageHandleController> webSocketSet = new CopyOnWriteArraySet<>();// 使用ConcurrentHashMap来存储用户ID和WebSocket会话对象的映射。private static ConcurrentHashMap<String, MessageHandleController> webSocketMap = new ConcurrentHashMap<>();//与某个客户端的连接会话，需要通过它来给客户端发送数据private Session session;private String uuid;@OnOpenpublic void onOpen(@PathParam("uuid") String uuid, Session session) {logger.info("uuid: {}, sessionId: {}", uuid, session.getId());// 将新建立的会话添加到webSocketMap中try {if (webSocketMap.containsKey(uuid)) {//如果有旧连接就先断开webSocketMap.get(uuid).session.close();webSocketSet.remove(webSocketMap.get(uuid));}this.session = session;this.uuid = uuid;webSocketSet.add(this); //加入set中webSocketMap.put(uuid, this); //加入map中// 其他代码...} catch (Exception e) {logger.error("error:" + e.getMessage());}}@OnClosepublic void onClose(@PathParam("uuid") String uuid, Session session) {// 当会话关闭时，从webSocketSet中移除该会话webSocketSet.remove(this);// 当会话关闭时，从webSocketMap中移除该会话webSocketMap.remove(uuid);}// 收到客户端消息后调用的方法@OnMessagepublic void onMessage(String message, Session session) {logger.info("Message from client: " + message);}// 发生错误时调用@OnErrorpublic void onError(Session session, Throwable error) {logger.error("error:" + error.getMessage());try {session.close();// 当会话关闭时，从webSocketSet中移除该会话webSocketSet.remove(this);// 当会话关闭时，从webSocketMap中移除该会话webSocketMap.remove(this.uuid);} catch (IOException e) {logger.error("error:" + e.getMessage());}}public void sendMessage(Session session, String msg) {logger.info("发送消息");try {//判断当前人员是否连接websocketif (session.isOpen()) {session.getAsyncRemote().sendText(msg);} else {session.close();webSocketSet.remove(this); //从set中删除webSocketMap.remove(this.uuid); //从map中删除}} catch (IOException e) {e.printStackTrace();}}public static CopyOnWriteArraySet<MessageHandleController> getWebSocketSet() {return webSocketSet;}public static void setWebSocketSet(CopyOnWriteArraySet<MessageHandleController> webSocketSet) {MessageHandleController.webSocketSet = webSocketSet;}public static ConcurrentHashMap<String, MessageHandleController> getWebSocketMap() {return webSocketMap;}public static void setWebSocketMap(ConcurrentHashMap<String, MessageHandleController> webSocketMap) {MessageHandleController.webSocketMap = webSocketMap;}public Session getSession() {return session;}public void setSession(Session session) {this.session = session;}public String getUuid() {return uuid;}public void setUuid(String uuid) {this.uuid = uuid;}
}

创建定时任务

@Scheduled(cron = "0 * * * * ?")
public void sendMsg() {logger.info("定时任务,start");CopyOnWriteArraySet<MessageHandleController> webSocketSet;try {webSocketSet = MessageHandleController.getWebSocketSet();webSocketSet.forEach(obj -> {logger.info("uuid:" + obj.getUuid());logger.info("session:" + obj.getSession().getId());obj.sendMessage(obj.getSession(), "time:" + LocalDateTime.now());});} catch (Exception e) {logger.error("error:" + e.getMessage());}logger.info("定时任务,end");
}