使用Spring WebSocket的SimpMessageTemplate和多endpoint配置

Question

我将Spring 4.3.5和WebSocket与SockJS、STOMP和SimpleBrokerMessageHandler一起使用。

在我的应用程序中，我有三个独立的WebSocketendpoint在不同的地址上运行：/endPointA、/ednpointB、/endpointC更具体地说，我有三个独立的配置类，用@configuration@EnableWebSocketMessageBroker注释进行注释。

我还有一个类有@Autowired SimpMessageTemplate。

最后，我有三个客户端，每个客户端连接到一个不同的endpoint。然而，他们都订阅了“相同”的频道地址，即/topic/messages

• ClientOne与endpointA连接

当我使用SimpMessagingTemboard向 /topic/messages发送消息时，所有客户端都会收到此消息。

之后我有两个问题：

1. 是否有办法隔离Web Socketendpoint，使消息不会传播到所有endpoint？
2. 为什么这会发生在这里？

我做了一些调查（堆转储分析），发现对于我的配置，我有：

• SimpMessageTemplate的三个实例，但是我总是使用同一个实例来发送消息（因为@Autowire-另外，我正在打印SimpMessageTemplate.toString（）

所以我想知道，这种在所有endpoint上的消息传播是否是SimpleBrokerMessageHandler或SimpMessageTemplate的“功能”。

Answer 1

我在一个多租户应用程序中遇到了同样的问题，我通过以下线程解决了这个问题：

我的websocketendpoint是：ws://127.0.0.1/My context app/ws，java配置文件是：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

   @Override
   public void registerStompEndpoints(StompEndpointRegistry registry) {       
      registry.addEndpoint("/ws").setAllowedOrigins("*");
      registry.addEndpoint("/ws").setAllowedOrigins("*").withSockJS(); 
   }
}

我的webSocket队列url表单的前缀是tenant-id：/[tenant-id]/[url-of-队列]。

每个客户端都使用自己的租户id进行订阅。由于WebSocketSecurityConfig，它无法订阅另一个客户端的队列。configureInbound（MessageSecurityMetadataSourceRegistry）方法和一个自定义Springbean，它有一个“websocket队列订阅安全检查”方法：

@Configuration
public class WebSocketSecurityConfig extends AbstractSecurityWebSocketMessageBrokerConfigurer {

    protected void configureInbound(MessageSecurityMetadataSourceRegistry messages) {
        messages
          .simpSubscribeDestMatchers("/**")
          .access("@customWSSecurityCheck.isSubscriptionAllowed(authentication, message)");
    }

}

我的自定义bean命名为CustWSSecurityCheck，检查是否允许经过身份验证的用户在队列中订阅。请记住，自定义身份验证实现了带有附加tenantId属性的org.springframework.security.core.身份验证，该属性由自定义Spring安全筛选器/身份验证方法上没有提到的附加代码填充：

@Bean()
public class CustomWSSecurityCheck {

    public boolean isSubscriptionAllowed(CustomAuthentication authentication, Message message) {

       StompHeaderAccessor sha = StompHeaderAccessor.wrap(message);
       String url = sha.getDestination().substring(1);
       String tenantId = url.substring(0, url.indexOf("/"));

       return tenantId.equals(authentication.getTenantId());
    }
}

当然，它涉及到服务器发送的每条消息都应该以正确的租户id为前缀：MessagingService.convertAndSed（"[tenant-id]/[url-of-队列]"，MessagePayload）