Spring Boot WebSocket：如何知道客户端何时取消订阅？

  • Z时代
  • 2024-01-10
  • 分类：问答

我目前有一个具有STOMP设置的简单WebSocket，其中客户端连接到主题（具有ID）。控制器立即响应要求的内容，并设置了一个变量来指示客户端已订阅的内容。@Scheduled现在，带有注释的方法每隔几秒钟就会向客户端发送一次请求。

在客户端第一次连接之前，计划的方法不会执行任何操作。但是，在第一次订阅后，无论是否订阅了客户端，它将继续发布。

@Controller
public class ServiceWebSocketController {
    @Autowired
    private ServiceService serviceService;
    @Autowired
    WebSocketSessionController webSocketSessionController;
    @Autowired
    private SimpMessagingTemplate simpMessagingTemplate;
    private Set<Long> services = new HashSet<>();
    @SubscribeMapping("/service/{serviceId}")
    public ServiceDTO subscribe(@DestinationVariable("serviceId") final Long serviceId) throws SQLException {
        System.out.println("Subscribed to Service with ID: " + serviceId);
        services.add(serviceId);
        return serviceService.getServiceWithProperties(serviceId).orElseThrow(() -> new ResourceNotFoundException("Service", "id", serviceId));
    }
    @Scheduled(fixedDelay = 2000)
    public void service() throws SQLException {
        services.removeIf(serviceId -> !webSocketSessionController.hasSubscriptionTo("/topic/service/" + serviceId));
        // Publish specified Service data to each anonymously subscribed client.
        services.forEach(serviceId -> {
            try {
                System.out.println("Publishing Service with ID: " + serviceId);
                // We don't use .convertAndSendToUser here, because all our clients are anonymous.
                simpMessagingTemplate.convertAndSend("/topic/service/" + serviceId, serviceService.getServiceWithProperties(serviceId));
            } catch (SQLException e) {
                e.printStackTrace();
            }
        });
    }
}

回答：

您可以SessionUnsubscribeEvent像这样收听事件：

@Controller
public class SessionUnsubscribeListener implements ApplicationListener<SessionUnsubscribeEvent> {
   @Override
   public void onApplicationEvent(SessionUnsubscribeEvent event) {
       GenericMessage message = (GenericMessage) event.getMessage();
       String simpDestination = (String) message.getHeaders().get("simpDestination");
       if ("/topic/service".equals(simpDestination)) {
           // do stuff
       }
   }
}

以上是 Spring Boot WebSocket：如何知道客户端何时取消订阅？ 的全部内容， 来源链接： utcz.com/qa/431371.html

回到顶部