如何在Spring WebSocket STOMP与ActiveMQ中实现客户端订阅路由？

这个需求其实很常见，我们可以通过自定义客户端入站通道拦截器来拦截订阅请求，根据当前登录用户的角色动态修改订阅目标地址，从而实现路由到对应ActiveMQ主题的效果。具体实现步骤如下：

1. 自定义订阅拦截器 #

创建一个实现ChannelInterceptor的拦截器类，在订阅请求发送到消息 broker 之前，修改目标目的地：

import org.springframework.messaging.Message;
import org.springframework.messaging.MessageChannel;
import org.springframework.messaging.simp.stomp.StompCommand;
import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
import org.springframework.messaging.support.ChannelInterceptor;
import org.springframework.messaging.support.MessageHeaderAccessor;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Component;

@Component
public class RoleBasedSubscribeInterceptor implements ChannelInterceptor {

    @Override
    public Message<?> preSend(Message<?> message, MessageChannel channel) {
        StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class);
        
        // 仅处理SUBSCRIBE类型的命令
        if (StompCommand.SUBSCRIBE.equals(accessor.getCommand())) {
            String originalDestination = accessor.getDestination();
            
            // 匹配客户端统一订阅的/app/notificator
            if ("/app/notificator".equals(originalDestination)) {
                // 获取当前登录用户的认证信息
                Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
                String targetDestination;
                
                // 根据用户角色判断目标主题
                // 这里假设角色标识为ROLE_ADMIN和ROLE_USER，根据你的实际系统调整
                boolean isAdmin = authentication.getAuthorities().stream()
                        .anyMatch(auth -> auth.getAuthority().equals("ROLE_ADMIN"));
                
                if (isAdmin) {
                    targetDestination = "/topic/notificator/admin";
                } else {
                    targetDestination = "/topic/notificator/user";
                }
                
                // 修改订阅的目标地址
                accessor.setDestination(targetDestination);
            }
        }
        
        return message;
    }
}

2. 注册拦截器到WebSocket配置 #

修改你现有的WebSocket配置类，将自定义拦截器添加到客户端入站通道：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.ChannelRegistration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Autowired
    private RoleBasedSubscribeInterceptor roleBasedSubscribeInterceptor;

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        registry.enableStompBrokerRelay("/topic");
        registry.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws")
                .withSockJS();
    }

    // 将自定义拦截器注册到客户端入站通道
    @Override
    public void configureClientInboundChannel(ChannelRegistration registration) {
        registration.interceptors(roleBasedSubscribeInterceptor);
    }
}

3. 注意事项 #

• 用户角色验证：确保你的系统已经正确配置了Spring Security（或其他认证机制），当前登录用户的角色信息能通过SecurityContextHolder正常获取。如果角色标识不是ROLE_ADMIN/ROLE_USER，请修改拦截器中的判断逻辑。
• 灵活性扩展：如果后续需要添加更多角色或修改路由规则，可以把目的地映射关系放到配置文件中（比如用@Value读取），避免硬编码。
• 无主动发送需求：因为客户端只接收消息不主动发送，这个方案完全适配——拦截器只处理订阅请求，不会影响其他消息流程。

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