fix: use wait/notify pattern instead of Thread.sleep

Signed-off-by: Daniel Arteaga Barba <[email protected]>

Author: dani8art

util/src/main/java/io/kubernetes/client/util/WebSocketStreamHandler.java

@@ -22,6 +22,8 @@
 import java.io.PipedOutputStream;
 import java.io.Reader;
 import java.nio.charset.StandardCharsets;
+import java.time.Instant;
+import java.time.temporal.ChronoUnit;
 import java.util.HashMap;
 import java.util.Map;
 import okhttp3.WebSocket;
@@ -212,9 +214,9 @@ private class WebSocketOutputStream extends OutputStream {
 
     private static final long MAX_QUEUE_SIZE = 16L * 1024 * 1024;
 
-    private static final int MAX_QUEUE_SIZE_MAX_ATTEMPTS = 15;
+    private static final int MAX_WAIT_MILLIS = 10000;
 
-    private static final int MAX_QUEUE_SIZE_WAIT_MILLISECONDS = 1000;
+    private static final int WAIT_MILLIS = 10;
 
     private final byte stream;
 
@@ -276,19 +278,22 @@ public void write(byte[] b, int offset, int length) throws IOException {
         System.arraycopy(b, offset + bytesWritten, buffer, 1, bufferSize);
         ByteString byteString = ByteString.of(buffer);
 
-        int attempts = 0;
-        while (WebSocketStreamHandler.this.socket.queueSize() + byteString.size() > MAX_QUEUE_SIZE
-            && attempts < MAX_QUEUE_SIZE_MAX_ATTEMPTS) {
-          try {
-            Thread.sleep(MAX_QUEUE_SIZE_WAIT_MILLISECONDS);
-            attempts++;
-          } catch (InterruptedException e) {
-            throw new IOException("Error waiting web socket queue", e);
+        final Instant start = Instant.now();
+        synchronized (WebSocketOutputStream.this) {
+          while (WebSocketStreamHandler.this.socket.queueSize() + byteString.size() > MAX_QUEUE_SIZE
+              && Instant.now().isBefore(start.plus(MAX_WAIT_MILLIS, ChronoUnit.MILLIS))) {
+            try {
+              wait(WAIT_MILLIS);
+            } catch (InterruptedException e) {
+              throw new IOException("Error waiting web socket queue", e);
+            }
+          }
+
+          if (!WebSocketStreamHandler.this.socket.send(byteString)) {
+            throw new IOException("WebSocket has closed.");
           }
-        }
 
-        if (!WebSocketStreamHandler.this.socket.send(byteString)) {
-          throw new IOException("WebSocket has closed.");
+          notifyAll();
         }
 
         bytesWritten += bufferSize;