use vthread exec for smoke

salesforce · May 26, 2024 · 4565a8d · 4565a8d
1 parent 54fa1f4
commit 4565a8d
Showing 1 changed file with 40 additions and 24 deletions.
diff --git a/model/src/test/java/com/salesforce/apollo/model/DomainTest.java b/model/src/test/java/com/salesforce/apollo/model/DomainTest.java
@@ -38,6 +38,7 @@
 import java.util.Arrays;
 import java.util.UUID;
 import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.Executors;
 import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
 import java.util.stream.IntStream;
@@ -90,30 +91,45 @@ public static void smoke(Oracle oracle) throws Exception {
 
         var countDown = new CountDownLatch(17);
 
-        retryNesting(() -> oracle.map(helpDeskMembers, adminMembers), 3).whenCompleteAsync(
-        (_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(ali, adminMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(ali, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(burcu, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(can, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(managerMembers, userMembers), 3).whenCompleteAsync(
-        (_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(technicianMembers, userMembers), 3).whenCompleteAsync(
-        (_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(demet, helpDeskMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(egin, helpDeskMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(egin, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(fuat, managerMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(gl, managerMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(hakan, technicianMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(irmak, technicianMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(abcTechMembers, technicianMembers), 3).whenCompleteAsync(
-        (_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(flaggedTechnicianMembers, technicianMembers), 3).whenCompleteAsync(
-        (_, _) -> countDown.countDown());
-        retryNesting(() -> oracle.map(jale, abcTechMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown());
-
-        countDown.await(120, TimeUnit.SECONDS);
+        try (var exec = Executors.newVirtualThreadPerTaskExecutor()) {
+
+            retryNesting(() -> oracle.map(helpDeskMembers, adminMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(ali, adminMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                   exec);
+            retryNesting(() -> oracle.map(ali, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                  exec);
+            retryNesting(() -> oracle.map(burcu, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                    exec);
+            retryNesting(() -> oracle.map(can, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                  exec);
+            retryNesting(() -> oracle.map(managerMembers, userMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(technicianMembers, userMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(demet, helpDeskMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                        exec);
+            retryNesting(() -> oracle.map(egin, helpDeskMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                       exec);
+            retryNesting(() -> oracle.map(egin, userMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                   exec);
+            retryNesting(() -> oracle.map(fuat, managerMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                      exec);
+            retryNesting(() -> oracle.map(gl, managerMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                    exec);
+            retryNesting(() -> oracle.map(hakan, technicianMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(irmak, technicianMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(abcTechMembers, technicianMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(flaggedTechnicianMembers, technicianMembers), 3).whenCompleteAsync(
+            (_, _) -> countDown.countDown(), exec);
+            retryNesting(() -> oracle.map(jale, abcTechMembers), 3).whenCompleteAsync((_, _) -> countDown.countDown(),
+                                                                                      exec);
+
+            countDown.await(120, TimeUnit.SECONDS);
+        }
 
         // Protected resource namespace
         var docNs = Oracle.namespace("Document");