Spring Boot STOMP端点隔离与消息路由策略

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfiguration implements WebSocketMessageBrokerConfigurer {

   @Override
   public void registerStompEndpoints(StompEndpointRegistry registry) {
      // 注册两个独立的STOMP端点
      registry.addEndpoint("/endpoint1", "/endpoint2")
              .setAllowedOriginPatterns("*") // 允许所有来源进行连接，生产环境应限制为特定域名
              .withSockJS(); // 启用SockJS支持，以便在浏览器不支持WebSocket时进行降级
   }

   // 其他WebSocket配置，例如配置消息代理等
   // @Override
   // public void configureMessageBroker(MessageBrokerRegistry config) {
   //    config.enableSimpleBroker("/topic", "/queue"); // 启用简单消息代理
   //    config.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客户端的消息
     * 消息发送到 /app/endpoint1/request
     * 响应发送给当前用户到 /user/endpoint1/queue/response
     */
    @MessageMapping("/endpoint1/request")
    @SendToUser("/endpoint1/queue/response")
    public MyResponse handleClient1Message(MyRequest request) {
        System.out.println("Received message from /endpoint1: " + request);
        // 这里可以包含处理/endpoint1客户端特定业务逻辑
        return new MyResponse("Response for endpoint1: " + request.getMessage());
    }

    /**
     * 处理来自/endpoint2客户端的消息
     * 消息发送到 /app/endpoint2/request
     * 响应发送给当前用户到 /user/endpoint2/queue/response
     */
    @MessageMapping("/endpoint2/request")
    @SendToUser("/endpoint2/queue/response")
    public MyResponse handleClient2Message(MyRequest request) {
        System.out.println("Received message from /endpoint2: " + request);
        // 这里可以包含处理/endpoint2客户端特定业务逻辑
        return new MyResponse("Response for endpoint2: " + request.getMessage());
    }
}

// 示例数据传输对象 (DTO)
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 + '\'' + '}'; }
}

class MyResponse {
    private String response;

    public MyResponse(String response) { this.response = response; }

    public String getResponse() { return response; }
    public void setResponse(String response) { this.response = response; }
}