springboot整合websocket推送，实现指定用户推送

实现指定用户推送，可以通过给每个连接的 WebSocket Session 设置一个唯一标识符，然后在服务器端维护一个映射表，将每个标识符和对应的 WebSocket Session 关联起来。这样，当需要向某个用户发送消息时，只需要从映射表中找到对应的 WebSocket Session，然后使用该 Session 向该用户发送消息即可。

下面是一个简单的示例代码：

@Controller
public class WebSocketController {

    // 维护用户标识符和 WebSocket Session 的映射表
    private static final Map<String, WebSocketSession> sessions = new ConcurrentHashMap<>();

    @OnOpen
    public void onOpen(Session session) {
        // 获取用户标识符
        String userId = getUserId(session);
        // 将用户标识符和 WebSocket Session 关联起来
        sessions.put(userId, session);
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        // 处理消息
    }

    @OnClose
    public void onClose(Session session) {
        // 获取用户标识符
        String userId = getUserId(session);
        // 将用户标识符和 WebSocket Session 的关联移除
        sessions.remove(userId);
    }

    /**
     * 向指定用户发送消息
     * @param userId 用户标识符
     * @param message 消息内容
     */
    public void sendMessage(String userId, String message) throws IOException {
        WebSocketSession session = sessions.get(userId);
        if (session != null && session.isOpen()) {
            session.sendMessage(new TextMessage(message));
        }
    }

    /**
     * 获取用户标识符
     * @param session WebSocket Session
     * @return 用户标识符
     */
    private String getUserId(Session session) {
        // 从 WebSocket Session 的 HTTP 握手请求中获取用户标识符
        return ((ServletServerHttpRequest) session.getHandshakeRequest()).getServletRequest().getParameter("userId");
    }
}

在上面的代码中，我们通过 `@OnOpen` 注解来处理 WebSocket 连接建立事件，通过 `@OnMessage` 注解来处理消息事件，通过 `@OnClose` 注解来处理连接关闭事件。在连接建立时，我们从 WebSocket Session 的 HTTP 握手请求中获取用户标识符，并将该标识符和 WebSocket Session 关联起来，存储到映射表中。在需要向某个用户发送消息时，我们从映射表中找到对应的 WebSocket Session，然后使用该 Session 向该用户发送消息。注意，由于 WebSocket Session 是不安全的，因此需要进行线程安全处理。在上面的代码中，我们使用了 `ConcurrentHashMap` 来实现线程安全的映射表。