Spring Boot STOMP端点隔离：实现独立消息处理与路由

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfiguration implements WebSocketMessageBrokerConfigurer {
   @Override
   public void registerStompEndpoints(StompEndpointRegistry registry) {
      // 注册了两个STOMP端点
      registry.addEndpoint("/endpoint1", "/endpoint2")
              .setAllowedOriginPatterns("*")
              .withSockJS();
   }

   // ... 其他配置 ...
}

@Controller
public class WebSocketController {
   @MessageMapping("/request")
   @SendToUser("/queue/response")
   public MyResponse handleMessage(MyRequest request) {
      // 业务逻辑
      return new MyResponse("Processed: " + request.getMessage());
   }
}

import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendToUser;
import org.springframework.stereotype.Controller;

@Controller
public class WebSocketMessageRouter {

    /**
     * 处理来自 /endpoint1 客户端的消息
     * 客户端发送到 /app/endpoint1/request (如果配置了 /app 作为应用目的地前缀)
     * 响应发送到用户专属的 /endpoint1/queue/response 队列
     */
    @MessageMapping("/endpoint1/request")
    @SendToUser("/endpoint1/queue/response")
    public MyResponse handleClient1Message(MyRequest request) {
        System.out.println("Processing message from client 1: " + request.getMessage());
        // 处理来自客户端 1 的 STOMP 消息
        // 假设 MyRequest 和 MyResponse 是自定义的数据模型
        return new MyResponse("Client 1 processed: " + request.getMessage());
    }

    /**
     * 处理来自 /endpoint2 客户端的消息
     * 客户端发送到 /app/endpoint2/request
     * 响应发送到用户专属的 /endpoint2/queue/response 队列
     */
    @MessageMapping("/endpoint2/request")
    @SendToUser("/endpoint2/queue/response")
    public MyResponse handleClient2Message(MyRequest request) {
        System.out.println("Processing message from client 2: " + request.getMessage());
        // 处理来自客户端 2 的 STOMP 消息
        return new MyResponse("Client 2 processed: " + request.getMessage());
    }
}

// 连接到 /endpoint1
var socket1 = new SockJS('/endpoint1');
var stompClient1 = Stomp.over(socket1);

stompClient1.connect({}, function(frame) {
    console.log('Connected to /endpoint1: ' + frame);
    // 订阅 /endpoint1 的响应队列
    stompClient1.subscribe('/user/endpoint1/queue/response', function(message) {
        console.log('Client 1 received: ' + message.body);
    });
    // 发送消息到 /endpoint1 的请求目的地
    stompClient1.send("/app/endpoint1/request", {}, JSON.stringify({'message': 'Hello from Client 1'}));
});

// 连接到 /endpoint2
var socket2 = new SockJS('/endpoint2');
var stompClient2 = Stomp.over(socket2);

stompClient2.connect({}, function(frame) {
    console.log('Connected to /endpoint2: ' + frame);
    // 订阅 /endpoint2 的响应队列
    stompClient2.subscribe('/user/endpoint2/queue/response', function(message) {
        console.log('Client 2 received: ' + message.body);
    });
    // 发送消息到 /endpoint2 的请求目的地
    stompClient2.send("/app/endpoint2/request", {}, JSON.stringify({'message': 'Hello from Client 2'}));
});