Improve Node health checks by batch runs

Signed-off-by: Viet Nguyen Duc <[email protected]>

Author: VietND96

java/src/org/openqa/selenium/grid/distributor/local/LocalNodeRegistry.java

@@ -32,7 +32,6 @@
 import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
-import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.TimeUnit;
@@ -404,31 +403,22 @@ private void processBatchesInParallel(List<List<Runnable>> batches, int maxConcu
     if (batches.isEmpty()) {
       return;
     }
-    List<CompletableFuture<Void>> inFlight = new ArrayList<>();
-    for (List<Runnable> batch : batches) {
-      CompletableFuture<Void> fut =
-          CompletableFuture.runAsync(
-              () ->
-                  batch.parallelStream()
-                      .forEach(
-                          r -> {
-                            try {
-                              r.run();
-                            } catch (Throwable t) {
-                              LOG.log(
-                                  getDebugLogLevel(), "Health check execution failed in batch", t);
-                            }
-                          }),
-              nodeHealthCheckService);
-      inFlight.add(fut);
-      if (inFlight.size() >= maxConcurrentBatches) {
-        CompletableFuture.allOf(inFlight.toArray(new CompletableFuture[0])).join();
-        inFlight.clear();
-      }
-    }
-    if (!inFlight.isEmpty()) {
-      CompletableFuture.allOf(inFlight.toArray(new CompletableFuture[0])).join();
-    }
+
+    // Process batches with controlled parallelism
+    batches.parallelStream()
+        .limit(maxConcurrentBatches)
+        .forEach(
+            batch ->
+                batch.parallelStream()
+                    .forEach(
+                        r -> {
+                          try {
+                            r.run();
+                          } catch (Throwable t) {
+                            LOG.log(
+                                getDebugLogLevel(), "Health check execution failed in batch", t);
+                          }
+                        }));
   }
 
   private static List<List<Runnable>> partition(List<Runnable> list, int size) {