如何获取 Spring 5 WebSocket API 中的所有活动会话？

Question

我在经典 Spring 5 WebSocket API 的帮助下编写了 WebSocket 应用程序，即不使用 SockJS 和 STOMP。我在获取所有活动的 http 会话时遇到问题。我可以获取一个当前会话，但如何获取所有活动会话？如果我使用经典的 Java API（JSR356），我会使用一个方法：session.getOpenSessions() 来获取所有打开的会话。但我在 Spring 5 WebSocket API 中找不到类似的方法。如何获取所有活动会话？

//Configuration of WebSocket.

@Configuration
@EnableWebSocket
public class WebSocketConfig implements 
WebSocketConfigurer {

@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
  //Map endpoint URL to handler's method.
  //Add interceptor of HTTP-handshake to copy HTTP-session's attributes to WebSocket-session's attributes.
  registry.addHandler(getWsEndpoint(), "/endpoint").addInterceptors(new HttpSessionHandshakeInterceptor());
}

@Bean
public WebSocketHandler getWsEndpoint() {
    return new WsEndpoint();
}
}

我的 WebSocket 端点：

// WebSocket endpoint class. 

public class WsEndpoint extends TextWebSocketHandler {

 public WsEndpoint(){}

 public WebSocketMessage<String> wsMsg;                      

  // This method will be called after successful websocket connection.
  @Override
  public void afterConnectionEstablished(WebSocketSession session)                                     
   throws java.lang.Exception {

     wsMsg = new TextMessage(new String("Connection ok!"));        
   session.sendMessage(wsMsg);  
  }

    // This method is called if message was recieved. 
    @Override
    public void handleTextMessage(WebSocketSession session, TextMessage message) throws IOException {       

     wsMsg = new TextMessage(new String("Message recieved!")); 
     session.sendMessage(wsMsg);
    }
}

Answer 1

( SimpUserRegistry)DefaultSimpUserRegistry跟踪连接的 websocket 用户。

该方法getUsers()返回所有连接的用户及其会话。

这是示例代码：

@RestController
public class WebSocketController {

    private final SimpUserRegistry simpUserRegistry;

    public WebSocketController(SimpUserRegistry simpUserRegistry) {
        this.simpUserRegistry = simpUserRegistry;
    }

    @GetMapping("/ws/users")
    public List<String> connectedEquipments() {
        return this.simpUserRegistry
                .getUsers()
                .stream()
                .map(SimpUser::getName)
                .collect(Collectors.toList());
    }
}