基于STOMP的Spring WebSocket多控制器映射及客户端连接问题

嘿，刚好这些都是Spring WebSocket + STOMP开发里的常见场景，我来给你一步步解答：

STOMP是一个标准化的消息协议，不管你的客户端是不是Spring技术栈，只要支持STOMP就能和Spring WebSocket服务端通信。比如前端可以用官方的stomp.js，Java客户端可以用Spring提供的StompClient，甚至其他语言的STOMP客户端都没问题——底层的WebSocket连接只是传输通道，STOMP负责上层的消息路由逻辑。

其实这里要区分下：底层的WebSocket连接是一个持久化的TCP连接，但STOMP是基于**消息目的地（Destination）**来路由的，不是直接绑定到某个控制器。Spring允许你在多个不同的控制器类里，通过@MessageMapping注解定义不同的消息目的地，这样同一个WebSocket连接发送的不同消息，会根据目的地被转发到对应的控制器方法处理。

核心就是利用STOMP的目的地路由规则，结合Spring的注解配置，具体步骤如下：

第一步：配置Spring WebSocket的STOMP基础 #

先写一个配置类，开启消息代理并注册WebSocket端点：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        // 配置广播消息的前缀（如果需要推送结果给客户端的话）
        config.enableSimpleBroker("/topic");
        // 配置客户端发送消息到服务端的前缀，比如客户端发/app/xxx，Spring会匹配对应的@MessageMapping
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        // 注册WebSocket端点，客户端通过这个地址连接，同时启用SockJS作为降级方案
        registry.addEndpoint("/ws").withSockJS();
    }
}

第二步：编写多个控制器处理不同消息 #

创建两个控制器，分别对应消息X和消息Y，用@MessageMapping指定各自的目的地：

处理消息X的控制器 #

@Controller
public class MessageXController {

    // 匹配客户端发送到/app/handle-message-x的消息
    @MessageMapping("/handle-message-x")
    // 可选：处理完成后把结果广播到/topic/message-x-result，客户端可以订阅这个地址接收结果
    @SendTo("/topic/message-x-result")
    public MessageXResult processMessageX(MessageX message) {
        // 这里写处理MessageX的业务逻辑
        return new MessageXResult("消息X处理成功", message.getContent());
    }
}

处理消息Y的控制器 #

@Controller
public class MessageYController {

    @MessageMapping("/handle-message-y")
    @SendTo("/topic/message-y-result")
    public MessageYResult processMessageY(MessageY message) {
        // 处理MessageY的业务逻辑
        return new MessageYResult("消息Y处理完成", message.getData());
    }
}

第三步：客户端通过单一连接发送不同消息 #

客户端只需要建立一次WebSocket连接，然后发送消息时指定对应的目的地即可。这里用前端stomp.js举个例子：

// 连接到Spring的WebSocket端点
const socket = new SockJS('/ws');
const stompClient = Stomp.over(socket);

// 建立连接
stompClient.connect({}, (frame) => {
    console.log('连接成功:', frame);

    // 可选：订阅处理结果的广播地址
    stompClient.subscribe('/topic/message-x-result', (result) => {
        console.log('消息X的处理结果:', JSON.parse(result.body));
    });
    stompClient.subscribe('/topic/message-y-result', (result) => {
        console.log('消息Y的处理结果:', JSON.parse(result.body));
    });

    // 发送消息X到对应的控制器
    stompClient.send(
        "/app/handle-message-x", 
        {}, 
        JSON.stringify({ content: "这是来自客户端的消息X" })
    );

    // 发送消息Y到对应的控制器
    stompClient.send(
        "/app/handle-message-y", 
        {}, 
        JSON.stringify({ data: "这是来自客户端的消息Y" })
    );
});

这样一来，客户端只用一个WebSocket连接，发送的不同消息会被Spring自动路由到对应的控制器方法处理，完全满足你的需求～

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