Spring Boot STOMP 多端点隔离：实现客户端消息路由与封装

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfiguration implements WebSocketMessageBrokerConfigurer {

   @Override
   public void registerStompEndpoints(StompEndpointRegistry registry) {
      registry.addEndpoint("/endpoint1", "/endpoint2")
              .setAllowedOriginPatterns("*") // 允许所有来源，生产环境应限制
              .withSockJS(); // 启用SockJS支持
   }

   // 其他配置，如消息代理、消息转换器等
   // @Override
   // public void configureMessageBroker(MessageBrokerRegistry registry) {
   //    registry.enableSimpleBroker("/topic", "/queue"); // 启用简单的消息代理
   //    registry.setApplicationDestinationPrefixes("/app"); // 设置应用目的地前缀
   // }
}

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

@Controller
public class WebSocketController {

    /**
     * 处理来自 /endpoint1 客户端的请求
     * 消息目的地应为 /endpoint1/request
     */
    @MessageMapping("/endpoint1/request")
    @SendToUser("/queue/response") // 或 /endpoint1/queue/response 以保持完全隔离
    public MyResponse handleClient1Message(MyRequest request) {
        System.out.println("收到来自 /endpoint1 的请求: " + request);
        // 处理来自客户端1的STOMP消息
        // ... 业务逻辑 ...
        return new MyResponse("Client1 processed: " + request.getMessage());
    }

    /**
     * 处理来自 /endpoint2 客户端的请求
     * 消息目的地应为 /endpoint2/request
     */
    @MessageMapping("/endpoint2/request")
    @SendToUser("/queue/response") // 或 /endpoint2/queue/response 以保持完全隔离
    public MyResponse handleClient2Message(MyRequest request) {
        System.out.println("收到来自 /endpoint2 的请求: " + request);
        // 处理来自客户端2的STOMP消息
        // ... 业务逻辑 ...
        return new MyResponse("Client2 processed: " + request.getMessage());
    }

    // 假设 MyRequest 和 MyResponse 是自定义的数据传输对象
    static class MyRequest {
        private String message;
        public String getMessage() { return message; }
        public void setMessage(String message) { this.message = message; }
        @Override public String toString() { return "MyRequest{message='" + message + "'}"; }
    }

    static class MyResponse {
        private String responseMessage;
        public MyResponse(String responseMessage) { this.responseMessage = responseMessage; }
        public String getResponseMessage() { return responseMessage; }
        public void setResponseMessage(String responseMessage) { this.responseMessage = responseMessage; }
        @Override public String toString() { return "MyResponse{responseMessage='" + responseMessage + "'}"; }
    }
}

// 连接到 /endpoint1
var socket1 = new SockJS('/endpoint1');
var stompClient1 = Stomp.over(socket1);
stompClient1.connect({}, function(frame) {
    console.log('Connected to /endpoint1: ' + frame);
    // 订阅自己的响应队列
    stompClient1.subscribe('/user/queue/response', function(message) {
        console.log('Client1 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);
    // 订阅自己的响应队列
    stompClient2.subscribe('/user/queue/response', function(message) {
        console.log('Client2 received: ' + message.body);
    });
    // 向 /endpoint2 的请求目的地发送消息
    stompClient2.send("/app/endpoint2/request", {}, JSON.stringify({'message': 'Hello from Client 2'}));
});