Java Spring Boot中如何实现用户订阅WebSocket频道？

你在Django Channels中使用的路径维度频道隔离能力，在Spring生态中对应STOMP协议的目的地（Destination） 机制：客户端订阅指定路径的目的地，服务端向该路径推送消息时，框架会自动把消息下发给所有订阅了该路径的客户端，和你举例的/payment/1、/payment/2频道场景完全匹配，不需要手动维护用户和频道的订阅映射关系。

在Spring Boot项目中直接引入官方WebSocket starter即可，以Maven配置为例：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

编写配置类开启STOMP消息代理，定义连接入口和频道路径规则：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        // 定义WebSocket连接入口，作用和Django Channels中的websocket路由入口一致
        registry.addEndpoint("/ws")
                .setAllowedOriginPatterns("*") // 生产环境替换为实际前端域名，避免跨域问题
                .withSockJS(); // 兼容低版本浏览器的降级方案，不需要可去掉
    }

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        // 开启内置内存消息代理，所有以/payment开头的路径都将被代理接管作为独立频道
        registry.enableSimpleBroker("/payment");
        // 定义客户端向服务端发送消息的路径前缀，和订阅频道逻辑无强关联，可按需调整
        registry.setApplicationDestinationPrefixes("/app");
    }
}

内置的简单消息代理会自动维护所有路径和客户端订阅会话的映射关系，往某个路径发消息时只会推送给对应订阅者，和Django Channels的group_send逻辑效果一致。

前端连接WebSocket后，直接传入对应频道路径完成订阅即可，和Django Channels中调用group_add加入频道的效果一致：

// 支付单1的相关用户（用户1、用户2）订阅/payment/1频道
stompClient.subscribe('/payment/1', (res) => {
  const paymentMsg = JSON.parse(res.body);
  console.log('收到支付单1的状态更新', paymentMsg);
})

// 支付单2的相关用户（用户3、用户4）订阅/payment/2频道
stompClient.subscribe('/payment/2', (res) => {
  const paymentMsg = JSON.parse(res.body);
  console.log('收到支付单2的状态更新', paymentMsg);
})

直接注入框架提供的SimpMessagingTemplate，调用推送方法传入目标频道路径即可，不需要手动遍历订阅session：

@Service
public class PaymentService {
    @Resource
    private SimpMessagingTemplate messagingTemplate;

    /**
     * 支付状态变更时调用，向对应支付单的频道推送消息
     * @param paymentId 支付单ID
     * @param updateMsg 要推送的消息内容
     */
    public void pushPaymentUpdate(Long paymentId, Object updateMsg) {
        String targetChannel = "/payment/" + paymentId;
        // 推送消息，仅订阅了targetChannel的客户端能收到
        messagingTemplate.convertAndSend(targetChannel, updateMsg);
    }
}

• 频道权限控制：如果需要限制用户只能订阅和自己相关的支付单频道，可以实现ChannelInterceptor接口，在客户端发起订阅请求时拦截，校验当前登录用户和路径中paymentId的关联关系，非法订阅直接拒绝即可。
• 集群部署适配：内置的内存消息代理仅支持单节点使用，如果是多节点集群，只需要把内置代理替换为对接Redis/RabbitMQ的外部STOMP代理即可，上层业务推送、订阅的代码不需要做任何修改，逻辑和Django Channels对接通道层实现跨节点消息同步一致。
• 一对一推送：如果需要给单个用户推送私信类消息，可以使用messagingTemplate.convertAndSendToUser()方法，框架会自动路由到对应用户的会话，不需要额外定义独立频道。

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