Spring多同用户连接场景下，如何通过STOMP WebSocket向特定会话发消息？

核心思路 #

STOMP协议支持向特定WebSocket会话定向发送消息，无需修改前端订阅逻辑。后端可通过SessionSubscribeEvent获取当前订阅的会话ID，结合用户认证信息拉取历史消息，最终通过指定会话ID的方式仅发送给当前订阅的会话。

具体实现步骤 #

1. 从订阅事件中提取关键信息
   在EventSubscribeListener的onApplicationEvent方法中，过滤目标订阅事件，提取会话ID、用户ID，并拉取历史消息：

   @Override
   public void onApplicationEvent(SessionSubscribeEvent event) {
       log.info("Stomp subscribed");
       
       // 获取Stomp头信息，过滤目标订阅路径
       StompHeaderAccessor headerAccessor = StompHeaderAccessor.wrap(event.getMessage());
       String destination = headerAccessor.getDestination();
       if (destination == null || !destination.contains("/private-message")) {
           return;
       }
       
       // 获取当前会话ID
       String sessionId = event.getSessionId();
       
       // 获取当前订阅用户的ID（需确保WebSocket连接已完成用户认证）
       WebSocketSession session = event.getSession();
       Principal principal = session.getPrincipal();
       if (principal == null) {
           log.warn("Unauthorized subscription attempt");
           return;
       }
       String userId = principal.getName();
       
       // 从数据库拉取该用户的历史消息
       List<Message> historyMessages = messageRepository.findByReceiverId(Long.valueOf(userId)); // 替换为你的实际查询逻辑
       
       // 向当前会话发送历史消息
       sendMessagesToSpecificSession(sessionId, userId, historyMessages);
   }
   

2. 实现定向发送到特定会话的方法
   通过设置消息头的sessionId属性，确保消息仅投递到目标会话：

   private void sendMessagesToSpecificSession(String sessionId, String userId, List<Message> messages) {
       // 创建消息头，指定目标会话ID
       SimpMessageHeaderAccessor headerAccessor = SimpMessageHeaderAccessor.create(SimpMessageType.MESSAGE);
       headerAccessor.setSessionId(sessionId);
       headerAccessor.setLeaveMutable(true);
       
       // 发送到用户订阅的路径，但仅目标会话可见
       String targetDestination = "/user/" + userId + "/private-message";
       for (Message msg : messages) {
           simpMessagingTemplate.convertAndSend(targetDestination, msg, headerAccessor.getMessageHeaders());
       }
   }
   

关键说明 #

• 前端无感知：前端保持原有的/user/{userId}/private-message订阅逻辑即可，无需处理会话ID相关逻辑。
• 会话隔离：通过消息头指定sessionId后，Spring WebSocket会自动过滤会话，仅将消息发送到目标连接。
• 认证依赖：需确保WebSocket连接完成用户认证，否则无法从Principal中获取用户ID（未认证场景可从自定义请求头或参数中提取用户标识）。

问题根源补充 #

你之前若直接调用simpMessagingTemplate.convertAndSendToUser(userId, "/private-message", messages)，会将消息广播到/user/{userId}/private-message路径下的所有订阅会话，这就是同用户多设备重复接收的原因。而指定会话ID的方式，相当于给消息添加了会话级过滤条件，仅匹配当前订阅的会话。

内容的提问来源于stack exchange，提问作者Kaarel Vomm