fix: complete client websocket future on open (#20587)

* fix: complete client websocket future on open

Completing the websocket future in onOpen event prevents the connection to hang
indefintely when application run on low resources.

Fixes #20586

* apply review suggestions

vaadin-dev-server/src/main/java/com/vaadin/base/devserver/viteproxy/ViteWebsocketConnection.java

@@ -42,7 +42,7 @@ public class ViteWebsocketConnection implements Listener {
 
     private final Consumer<String> onMessage;
     private final Runnable onClose;
-    private final CompletableFuture<WebSocket> clientWebsocket;
+    private final CompletableFuture<WebSocket> clientWebsocket = new CompletableFuture<>();
     private final List<CharSequence> parts = new ArrayList<>();
 
     private static Logger getLogger() {
@@ -72,15 +72,20 @@ public ViteWebsocketConnection(int port, String path, String subProtocol,
         this.onClose = onClose;
         String wsHost = ViteHandler.DEV_SERVER_HOST.replace("http://", "ws://");
         URI uri = URI.create(wsHost + ":" + port + path);
-        clientWebsocket = HttpClient.newHttpClient().newWebSocketBuilder()
+        HttpClient.newHttpClient().newWebSocketBuilder()
                 .subprotocols(subProtocol).buildAsync(uri, this)
                 .whenComplete(((webSocket, failure) -> {
                     if (failure == null) {
                         getLogger().debug(
                                 "Connection to {} using the {} protocol established",
                                 uri, webSocket.getSubprotocol());
+                        if (clientWebsocket.complete(webSocket)) {
+                            getLogger().trace(
+                                    "Websocket future completed in client build completion");
+                        }
                     } else {
                         getLogger().debug("Failed to connect to {}", uri);
+                        clientWebsocket.completeExceptionally(failure);
                         onConnectionFailure.accept(failure);
                     }
                 }));
@@ -90,6 +95,9 @@ public ViteWebsocketConnection(int port, String path, String subProtocol,
     public void onOpen(WebSocket webSocket) {
         getLogger().debug("Connected using the {} protocol",
                 webSocket.getSubprotocol());
+        if (clientWebsocket.complete(webSocket)) {
+            getLogger().trace("Websocket future completed in onOpen");
+        }
         Listener.super.onOpen(webSocket);
     }