SpringBoot WebSocket 实现离线消息推送 - 详细教程和示例代码

WebSocket 是一种通信协议，可以在 Web 浏览器和服务器之间实现双向通信。Spring Boot 是一种快速开发框架，可以通过简单的配置和自动化流程来创建 Spring 应用程序。在这个背景下，离线消息是一种在用户断开连接后仍然能够接收到的消息。

实现离线消息需要以下步骤：

1. 在 WebSocket 会话中，将未发送的消息存储在队列中。
2. 当用户重新连接时，从队列中检索所有未发送的消息，并将它们发送给用户。

以下是实现离线消息的示例代码：

@Component
public class WebSocketHandler extends TextWebSocketHandler {
    private static final Map<String, Session> sessions = new ConcurrentHashMap<>();

    private static final Map<String, Queue<String>> offlineMessages = new ConcurrentHashMap<>();

    @Override
    public void afterConnectionEstablished(Session session) throws Exception {
        String userId = getUserIdFromSession(session);
        sessions.put(userId, session);

        Queue<String> queue = offlineMessages.get(userId);
        if (queue != null && !queue.isEmpty()) {
            while (!queue.isEmpty()) {
                session.sendMessage(new TextMessage(queue.poll()));
            }
        }
    }

    @Override
    public void afterConnectionClosed(Session session, CloseStatus status) throws Exception {
        String userId = getUserIdFromSession(session);
        sessions.remove(userId);
    }

    public void sendMessageToUser(String userId, String message) throws IOException {
        Session session = sessions.get(userId);
        if (session != null && session.isOpen()) {
            session.sendMessage(new TextMessage(message));
        } else {
            Queue<String> queue = offlineMessages.get(userId);
            if (queue == null) {
                queue = new ConcurrentLinkedQueue<>();
                offlineMessages.put(userId, queue);
            }
            queue.offer(message);
        }
    }

    private String getUserIdFromSession(Session session) {
        return (String) session.getAttributes().get('userId');
    }
}

在这个示例中，我们维护了两个映射，一个用于存储 WebSocket 会话，另一个用于存储离线消息。在 afterConnectionEstablished 方法中，我们将 WebSocket 会话存储在映射中，并从离线消息队列中检索未发送的消息。在 afterConnectionClosed 方法中，我们从 WebSocket 会话映射中删除会话。最后，我们提供了一个 sendMessageToUser 方法，用于将消息发送给用户。如果会话已经存在，则直接发送消息。否则，我们将消息存储在离线消息队列中，以便在用户重新连接时发送。