如何根据条件限制Spring中订阅特定主题的stomp客户端的数量？

Question

我一直在研究一种方法来限制可以订阅特定跺脚主题但尚未理解的客户数量，这可能是根据我的需求的正确方法。

我的用例是一个游戏，我正在Angular（ng2 stompjs stomp客户端）和Spring Boot Websockets（目前正在使用Spring内存消息代理）中开发它。

其想法是，用户可以连接并订阅“/大厅”跺脚主题，在那里他可以看到打开的游戏室，这些游戏室可能处于不同的状态。例如，由于加入的玩家数量较少，正在比赛或尚未开始。我想截取并以编程方式将客户端的可能订阅限制为特定的“/room/{roomId}”主题，如果已达到最大玩家数，例如4。也可能有一些简单的客户端验证来限制这一点，但我认为只有客户端验证是不够的

因此，我的主要问题是：如何在Spring拦截特定的stomp主题订阅？是否可以向客户端请求者返回某种无法完成订阅的错误消息？

我真的很感激你的帮助，提前谢谢你！

Answer 1

您可以实现一个侦听订阅的StompEventListener，在这里，我们可以将目的地（房间号）与特定房间中的玩家数量进行映射。如果计数已达到最大值，则拒绝订阅。

@Service
class StompEventListener() {
   private Map<String, int> roomIdVsPlayerCount = new HashMap<>();
   
   @EventListener
   public void handleSubscription(SessionSubscribe event) {
     StompHeaderAccessor accessor = StompHeaderAccessor.wrap(event.getMessage());
     String destination = accessor.getDestination();

     String roomId = destination.substring(...); //Parsed RoomID
     if(roomIdVsPlayerCount.get(roomId) == MAX_ALLOWED_PLAYERS) {
       //Throw exception which will terminate that client connection 
         or, send an error message like so:
       simpMessagingTemplate.convertAndSend(<some_error_message>);
       return;
     }
     //So it is not at maximum do further logic to actually subscribe 
       user and
     roomIdVsPlayerCount.get(roomId) += 1; 
   }

   @EventListener
   public void handleUnsubscription(SessionUnsubscribe event) {
    ...
   }
}

有用的参考资料：

1. SessionSubscribeeEvent（用于处理订阅）

编辑

请尝试从通道侦听器发送异常，因为上面没有发送异常，所以它会传播到客户端。我们前面定义的映射可以定义为一个单独类中的bean，该类可以访问（使用@Autowired）事件处理程序（用于递增和递减）和TopicSubscriptionInterceptor（用于验证）。

@Component
class TopicSubscriptionInterceptor implements ChannelInterceptor {
    @Override
    public Message<?> preSend(Message<?> message, MessageChannel channel){
      StompHeaderAccessor accessor = StompHeaderAccessor.wrap(message);
      String destination = accessor.getDestination();

      String roomId = destination.substring(...); //Parsed RoomID
      if(roomIdVsPlayerCount.get(roomId) == MAX_ALLOWED_PLAYERS) {
        //Throw exception which will terminate that client connection 
      }
     
     //Since it is not at limit continue 
    
    }
}

实现TopicSubcriptionInterceptor的有用参考：TopicSubcriptionInterceptor