Provide opportunity to create a client WebSocket with a given vertx context to use.

Author: vietj

src/main/java/io/vertx/core/http/impl/ClientWebSocketImpl.java

@@ -27,7 +27,7 @@
 /**
   * Client WebSocket implementation
   */
-public class ClientWebSocketImpl implements ClientWebSocket {
+public class ClientWebSocketImpl implements ClientWebSocketInternal {
 
   private HttpClientBase client;
   private final AtomicReference<Promise<WebSocket>> connect = new AtomicReference<>();
@@ -48,7 +48,15 @@ public class ClientWebSocketImpl implements ClientWebSocket {
 
   @Override
   public Future<WebSocket> connect(WebSocketConnectOptions options) {
-    ContextInternal ctx = client.vertx().getOrCreateContext();
+    return connect(client.vertx().getOrCreateContext(), options);
+  }
+
+  @Override
+  public Future<WebSocket> connect(Context context, WebSocketConnectOptions options) {
+    return connect((ContextInternal) context, options);
+  }
+
+  private Future<WebSocket> connect(ContextInternal ctx, WebSocketConnectOptions options) {
     Promise<WebSocket> promise = ctx.promise();
     if (!connect.compareAndSet(null, promise)) {
       return ctx.failedFuture("Already connecting");

src/main/java/io/vertx/core/http/impl/ClientWebSocketInternal.java

@@ -0,0 +1,26 @@
+/*
+ * Copyright (c) 2011-2024 Contributors to the Eclipse Foundation
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License 2.0 which is available at
+ * http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
+ * which is available at https://www.apache.org/licenses/LICENSE-2.0.
+ *
+ * SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
+ */
+package io.vertx.core.http.impl;
+
+import io.vertx.core.Context;
+import io.vertx.core.Future;
+import io.vertx.core.http.ClientWebSocket;
+import io.vertx.core.http.WebSocket;
+import io.vertx.core.http.WebSocketConnectOptions;
+
+/**
+ * @author <a href="mailto:[email protected]">Julien Viet</a>
+ */
+public interface ClientWebSocketInternal extends ClientWebSocket {
+
+  Future<WebSocket> connect(Context context, WebSocketConnectOptions options);
+
+}

src/main/java/io/vertx/core/http/impl/HttpClientBase.java

@@ -172,16 +172,6 @@ public void webSocket(WebSocketConnectOptions connectOptions, Handler<AsyncResul
     webSocket(connectOptions, promise);
   }
 
-  Future<WebSocket> webSocket(ContextInternal ctx, WebSocketConnectOptions connectOptions) {
-    PromiseInternal<WebSocket> promise = ctx.promise();
-    webSocket(connectOptions, promise);
-    return promise.andThen(ar -> {
-      if (ar.succeeded()) {
-        ar.result().resume();
-      }
-    });
-  }
-
   private void webSocket(WebSocketConnectOptions connectOptions, PromiseInternal<WebSocket> promise) {
     ContextInternal ctx = promise.context();
     int port = getPort(connectOptions);