（代码）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;

    @OnOpen
    public 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());
        }
    }

    @OnClose
    public void onClose(@PathParam("uuid") String uuid, Session session) {
        // 当会话关闭时，从webSocketSet中移除该会话
        webSocketSet.remove(this);
        // 当会话关闭时，从webSocketMap中移除该会话
        webSocketMap.remove(uuid);
    }

    // 收到客户端消息后调用的方法
    @OnMessage
    public void onMessage(String message, Session session) {
        logger.info("Message from client: " + message);
    }

    // 发生错误时调用
    @OnError
    public 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 {
            //判断当前人员是否连接websocket
            if (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");
}