Question

我正在使用this教程，我正在试图弄清楚如何获得当前会话的数量。

我的WebSocketConfig看起来像这样（从教程中复制并粘贴）：

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic");
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/gs-guide-websocket").withSockJS();
    }

}

我想知道这个课程内部的课程数量（再次复制和粘贴）：

@Controller
public class GreetingController {


    @MessageMapping("/hello")
    @SendTo("/topic/greetings")
    public Greeting greeting(HelloMessage message) throws Exception {
        Thread.sleep(1000); // simulated delay
        return new Greeting("Hello, " + message.getName() + "!");
    }

}

是否有一种简单的方法可以获得当前会话（用户，连接）到websocket的数量？

编辑：

以下是我的解决方案：

Set<String> mySet = Collections.newSetFromMap(new ConcurrentHashMap<String, Boolean>());

@EventListener
private void onSessionConnectedEvent(SessionConnectedEvent event) {
    StompHeaderAccessor sha = StompHeaderAccessor.wrap(event.getMessage());
    mySet.add(sha.getSessionId());
}

@EventListener
private void onSessionDisconnectEvent(SessionDisconnectEvent event) {
    StompHeaderAccessor sha = StompHeaderAccessor.wrap(event.getMessage());
    mySet.remove(sha.getSessionId());
}

我现在可以使用mySet.size()获取会话数。

Answer 1

您可以使用SimpUserRegistry及其getUserCount()方法，而不是手动处理连接。

示例：

@Autowired
private SimpUserRegistry simpUserRegistry;

public int getNumberOfSessions() {
    return simpUserRegistry.getUserCount();
}

Answer 2

您可以使用 ApplicationContext 事件。每个连接，订阅或其他操作都会触发一个特殊事件： SessionConnectEvent ， SessionConnectedEvent ， SessionSubscribeEvent 等等。

完整文档是here。当其中一个事件触发时，您可以使用自己的逻辑来处理它。

示例代码供参考：

@EventListener(SessionConnectEvent.class)
public void handleWebsocketConnectListner(SessionConnectEvent event) {
    logger.info("Received a new web socket connection : " + now());
}

@EventListener(SessionDisconnectEvent.class)
public void handleWebsocketDisconnectListner(SessionDisconnectEventevent) {
    logger.info("session closed : " + now());
}

Spring websocket - 如何获得会话数

2 个答案: