在Spring Boot中为不同STOMP端点实现消息隔离与定制化路由

@Controller
public class WebSocketController {
   @MessageMapping("/request")
   @SendToUser("/queue/response")
   public MyResponse handleMessage(MyRequest request) {
      // 业务逻辑处理
   }
}

import org.springframework.context.annotation.Configuration;
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 WebSocketConfiguration implements WebSocketMessageBrokerConfigurer {

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

   @Override
   public void configureMessageBroker(MessageBrokerRegistry config) {
      // 配置消息代理，这里使用简单的内存消息代理
      // 客户端发送到/app前缀的会路由到@MessageMapping
      config.setApplicationDestinationPrefixes("/app");
      // 客户端订阅/topic或/queue前缀的会路由到消息代理
      config.enableSimpleBroker("/topic", "/queue");
   }
}

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

@Controller
public class WebSocketController {

    // 假设MyRequest和MyResponse是你的消息POJO
    static class MyRequest {
        private String message;
        // getters and setters
        public String getMessage() { return message; }
        public void setMessage(String message) { this.message = message; }
    }

    static class MyResponse {
        private String response;
        // getters and setters
        public String getResponse() { return response; }
        public void setResponse(String response) { this.response = response; }
    }

    /**
     * 处理来自 /endpoint1 客户端的消息
     * 客户端发送到 /app/endpoint1/request
     * 响应发送到 /user/endpoint1/queue/response
     */
    @MessageMapping("/endpoint1/request")
    @SendToUser("/endpoint1/queue/response")
    public MyResponse handleClient1Message(MyRequest request) {
        System.out.println("Received from endpoint1: " + request.getMessage());
        // 处理来自客户端1的STOMP消息
        MyResponse response = new MyResponse();
        response.setResponse("Response from Endpoint1: " + request.getMessage());
        return response;
    }

    /**
     * 处理来自 /endpoint2 客户端的消息
     * 客户端发送到 /app/endpoint2/request
     * 响应发送到 /user/endpoint2/queue/response
     */
    @MessageMapping("/endpoint2/request")
    @SendToUser("/endpoint2/queue/response")
    public MyResponse handleClient2Message(MyRequest request) {
        System.out.println("Received from endpoint2: " + request.getMessage());
        // 处理来自客户端2的STOMP消息
        MyResponse response = new MyResponse();
        response.setResponse("Response from Endpoint2: " + request.getMessage());
        return response;
    }
}

// 假设使用SockJS和Stomp.js
var socket1 = new SockJS('/endpoint1');
var stompClient1 = Stomp.over(socket1);

stompClient1.connect({}, function(frame) {
    console.log('Connected to endpoint1: ' + frame);

    // 订阅 /user/endpoint1/queue/response
    stompClient1.subscribe('/user/endpoint1/queue/response', function(message) {
        console.log('Response from endpoint1: ' + message.body);
    });

    // 发送消息到 /app/endpoint1/request
    stompClient1.send("/app/endpoint1/request", {}, JSON.stringify({'message': 'Hello from Client1'}));
});

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

stompClient2.connect({}, function(frame) {
    console.log('Connected to endpoint2: ' + frame);

    // 订阅 /user/endpoint2/queue/response
    stompClient2.subscribe('/user/endpoint2/queue/response', function(message) {
        console.log('Response from endpoint2: ' + message.body);
    });

    // 发送消息到 /app/endpoint2/request
    stompClient2.send("/app/endpoint2/request", {}, JSON.stringify({'message': 'Hello from Client2'}));
});