[java] Ensure Java 11 client can send multiple websocket requests. Collect all the received data before processing.

pujagani · pujagani · commit 7387bc94a39e · 2022-09-15T20:59:00.000+05:30
diff --git a/java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java b/java/src/org/openqa/selenium/remote/http/jdk/JdkHttpClient.java
@@ -123,10 +123,18 @@ public WebSocket openSocket(HttpRequest request, WebSocket.Listener listener) {
       client.newWebSocketBuilder().buildAsync(
         uri,
         new java.net.http.WebSocket.Listener() {
+          final StringBuilder builder = new StringBuilder();
 
           @Override
           public CompletionStage<?> onText(java.net.http.WebSocket webSocket, CharSequence data, boolean last) {
-            listener.onText(data);
+            builder.append(data);
+
+            if (last) {
+              listener.onText(builder.toString());
+              builder.setLength(0);
+            }
+
+            webSocket.request(1);
             return null;
           }
 
@@ -136,6 +144,7 @@ public CompletionStage<?> onBinary(java.net.http.WebSocket webSocket, ByteBuffer
             data.get(ary, 0, ary.length);
 
             listener.onBinary(ary);
+            webSocket.request(1);
             return null;
           }
 
@@ -148,6 +157,7 @@ public CompletionStage<?> onClose(java.net.http.WebSocket webSocket, int statusC
           @Override
           public void onError(java.net.http.WebSocket webSocket, Throwable error) {
             listener.onError(error);
+            webSocket.request(1);
           }
         });
 
