Added -timeout <ms> option to PinnedVThreads

cfredri4 · Mar 3, 2025 · 3d693c8 · 3d693c8
1 parent bca00a5
commit 3d693c8
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/tests/other/org/jgroups/tests/PinnedVThreads.java b/tests/other/org/jgroups/tests/PinnedVThreads.java
@@ -22,12 +22,16 @@
  * @since  5.4.4
  */
 public class PinnedVThreads {
-    protected int cores, pinned;
+    protected int  cores, pinned;
+    protected long timeout=5000;
 
     public int  cores()            {return cores;}
     public void cores(int cores)   {this.cores=cores;}
     public int  pinned()           {return pinned;}
     public void pinned(int pinned) {this.pinned=pinned;}
+    public long timeout()          {return timeout;}
+    public void timeout(long t)    {timeout=t;}
+
 
     public void start() throws IOException, InterruptedException {
         int available_cores=Runtime.getRuntime().availableProcessors();
@@ -63,7 +67,7 @@ public void start() throws IOException, InterruptedException {
             t.start();
         }
 
-        Util.waitUntilTrue(6_000, 500, () -> count.get() == pinned);
+        Util.waitUntilTrue(timeout, 500, () -> count.get() == pinned);
         if(count.get() == pinned)
             System.out.printf("\n** SUCCESS: %d worker threads completed (expected: %d)\n", count.get(), pinned);
         else
@@ -89,7 +93,12 @@ public static void main(String[] args) throws Exception {
                 test.pinned(Integer.parseInt(args[++i]));
                 continue;
             }
-            System.out.printf("%s [-cores <number of cores>] [-pinned <number of virtual threads pinned> ]\n",
+            if("-timeout".equals(args[i])) {
+                test.timeout(Long.parseLong(args[++i]));
+                continue;
+            }
+            System.out.printf("%s [-cores <number of cores>] [-pinned <number of virtual threads pinned>] " +
+                                "[-timeout <msecs>]\n",
                               PinnedVThreads.class.getSimpleName());
             return;
         }