Add WebSocket timeout and interval to JdkHttpClient

Author: iampopovich

java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java

@@ -80,13 +80,17 @@ public class JdkHttpClient implements HttpClient {
   private final ExecutorService executorService;
   private final Duration readTimeout;
   private final Duration connectTimeout;
+  private final Duration webSocketTimeout;
+  private final Duration webSocketInterval;
 
   JdkHttpClient(ClientConfig config) {
     Objects.requireNonNull(config, "Client config must be set");
 
     this.messages = new JdkHttpMessages(config);
     this.readTimeout = config.readTimeout();
     this.connectTimeout = config.connectionTimeout();
+    this.webSocketTimeout = config.webSocketTimeout();
+    this.webSocketInterval = config.webSocketInterval();
     this.websockets = new ArrayList<>();
     this.handler = config.filter().andFinally(this::execute0);
 
@@ -244,7 +248,7 @@ public void onError(java.net.http.WebSocket webSocket, Throwable error) {
 
     try {
       underlyingSocket =
-          webSocketCompletableFuture.get(readTimeout.toMillis(), TimeUnit.MILLISECONDS);
+          webSocketCompletableFuture.get(webSocketTimeout.toMillis(), TimeUnit.MILLISECONDS);
     } catch (CancellationException e) {
       throw new ConnectionFailedException("JdkWebSocket initial request canceled", e);
     } catch (ExecutionException e) {
@@ -314,7 +318,7 @@ public WebSocket send(Message message) {
               long start = System.currentTimeMillis();
               CompletableFuture<java.net.http.WebSocket> future = makeCall.get();
               try {
-                future.get(readTimeout.toMillis(), TimeUnit.MILLISECONDS);
+                future.get(webSocketTimeout.toMillis(), TimeUnit.MILLISECONDS);
               } catch (CancellationException e) {
                 throw new WebDriverException(e.getMessage(), e);
               } catch (ExecutionException e) {