如何等待Spring WebSocketStompClient连接

// 创建WebSocket连接配置
WebSocketClient webSocketClient = new StandardWebSocketClient();
WebSocketStompClient stompClient = new WebSocketStompClient(webSocketClient);
stompClient.setDefaultUri("ws://localhost:8080/ws");

// 创建WebSocket连接处理器
WebSocketHandler webSocketHandler = new MyWebSocketHandler();

// 创建WebSocket连接监听器
WebSocketConnectionListener connectionListener = new MyWebSocketConnectionListener();

// 创建WebSocket连接
ListenableFuture<StompSession> future = stompClient.connect(webSocketHandler, connectionListener);

// 等待连接建立
StompSession stompSession = future.get();

// 连接建立后的操作
// ...

// WebSocket连接处理器的实现
class MyWebSocketHandler extends WebSocketHandlerAdapter {
    // 处理连接建立事件
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 连接建立后的处理逻辑
        // ...
    }
    
    // 处理消息接收事件
    @Override
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
        // 消息接收后的处理逻辑
        // ...
    }
    
    // 处理连接关闭事件
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        // 连接关闭后的处理逻辑
        // ...
    }
}

// WebSocket连接监听器的实现
class MyWebSocketConnectionListener implements WebSocketConnectionListener {
    // 处理连接建立事件
    @Override
    public void onConnect(StompSession session) {
        // 连接建立后的处理逻辑
        // ...
    }
    
    // 处理连接关闭事件
    @Override
    public void onClose(StompSession session, CloseStatus closeStatus) {
        // 连接关闭后的处理逻辑
        // ...
    }
}