Spring Boot无STOMP原生WebSocket动态端点及Spring Bean访问问题

完全理解你的感受——Java原生的WebSocket API明明简洁直观，Spring却把它包装得有点绕，而且文档一个劲推STOMP，确实有点过度设计了。不过好消息是，用@EnableWebSocket、WebSocketConfigurer和TextWebSocketHandler完全能实现带动态URI的端点，还能正常访问Spring Bean，下面给你一套可行的方案：

核心思路 #

我们通过自定义握手拦截器解析动态路径参数（比如/{username}），把参数存到WebSocket会话的属性中；同时让WebSocket Handler成为Spring管理的Bean，这样就能直接注入仓库、服务等组件。

1. 配置WebSocket端点与拦截器 #

首先创建WebSocket配置类，注册动态路径端点，并添加自定义拦截器来提取路径参数：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    private final ChatWebSocketHandler chatWebSocketHandler;
    private final PathMatcher pathMatcher;

    // 注入Spring管理的Handler和路径匹配器
    public WebSocketConfig(ChatWebSocketHandler chatWebSocketHandler, PathMatcher pathMatcher) {
        this.chatWebSocketHandler = chatWebSocketHandler;
        this.pathMatcher = pathMatcher;
    }

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(chatWebSocketHandler, "/chat/{username}")
                .setAllowedOrigins("http://your-frontend-domain.com") // 生产环境替换为实际域名
                .addInterceptors(new CustomHandshakeInterceptor(pathMatcher));
    }
}

2. 自定义握手拦截器提取路径参数 #

这个拦截器会在WebSocket握手阶段解析/{username}这样的动态参数，并存入会话属性：

public class CustomHandshakeInterceptor implements HandshakeInterceptor {

    private final PathMatcher pathMatcher;

    public CustomHandshakeInterceptor(PathMatcher pathMatcher) {
        this.pathMatcher = pathMatcher;
    }

    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
        if (request instanceof ServletServerHttpRequest servletRequest) {
            String requestUri = servletRequest.getServletRequest().getRequestURI();
            // 匹配动态路径模板，提取参数
            Map<String, String> pathVars = pathMatcher.extractUriTemplateVariables("/chat/{username}", requestUri);
            String username = pathVars.get("username");
            if (username != null) {
                attributes.put("username", username); // 把参数存入会话属性
            }
        }
        return true; // 允许握手继续
    }

    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
        // 握手完成后的操作（可选）
    }
}

3. 实现WebSocket Handler（可注入Spring Bean） #

把Handler标记为@Component，让Spring管理它，这样就能直接注入仓库、服务等组件，同时从会话属性中获取动态参数：

@Component
public class ChatWebSocketHandler extends TextWebSocketHandler {

    // 直接注入Spring Bean
    private final UserRepository userRepository;
    private final ChatService chatService;

    // 构造方法注入（推荐方式）
    public ChatWebSocketHandler(UserRepository userRepository, ChatService chatService) {
        this.userRepository = userRepository;
        this.chatService = chatService;
    }

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 从会话属性中获取动态参数
        String username = (String) session.getAttributes().get("username");
        if (username != null) {
            // 调用Spring Bean的方法，比如注册用户会话
            chatService.registerUserSession(username, session);
            System.out.println("用户 " + username + " 已连接");
        }
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        String username = (String) session.getAttributes().get("username");
        String content = message.getPayload();
        // 处理消息，比如调用服务转发给其他用户
        chatService.broadcastMessage(username, content);
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        String username = (String) session.getAttributes().get("username");
        if (username != null) {
            chatService.unregisterUserSession(username);
            System.out.println("用户 " + username + " 已断开连接");
        }
    }
}

关键注意事项 #

• 跨域配置：setAllowedOrigins在生产环境一定要指定具体域名，不要用*，避免安全风险。
• 线程安全：TextWebSocketHandler默认是单例，所以如果在Handler里维护会话映射（比如用户与session的对应关系），要使用ConcurrentHashMap这类线程安全的集合。
• 路径匹配：确保拦截器中extractUriTemplateVariables的路径模板和配置类中注册的端点路径完全一致，否则参数提取会失败。

这种方案完全基于Spring的基础WebSocket API，不需要引入STOMP或消息代理，同时保留了Spring Bean的注入能力，完美匹配你的需求。

内容的提问来源于stack exchange，提问作者Sidney de Moraes