Supply an Executor

Signed-off-by: Andrew Carbonetto <[email protected]>
Bit-Quill · Nov 24, 2023 · 0db4061 · 0db4061
1 parent 3e6feed
commit 0db4061
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/java/benchmarks/src/main/java/javababushka/benchmarks/utils/Benchmarking.java b/java/benchmarks/src/main/java/javababushka/benchmarks/utils/Benchmarking.java
@@ -8,6 +8,8 @@
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 import java.util.concurrent.atomic.AtomicInteger;
@@ -152,6 +154,8 @@ public static void printResults(
 
   public static void testClientSetGet(
       Supplier<Client> clientCreator, BenchmarkingApp.RunConfiguration config, boolean async) {
+    ExecutorService asyncExecutor = Executors.newFixedThreadPool(
+        Runtime.getRuntime().availableProcessors() + 1);
     for (int concurrentNum : config.concurrentTasks) {
       int iterations = Math.min(Math.max(100000, concurrentNum * 10000), 10000000);
       for (int clientCount : config.clientCount) {
@@ -212,7 +216,7 @@ public static void testClientSetGet(
                         clientIndex = iterationIncrement % clients.size();
                       }
                       return taskActionResults;
-                    }));
+                    }, asyncExecutor));
           }
           if (config.debugLogging) {
             System.out.printf("%s client Benchmarking: %n", clientName);