Allow AbstractQueueVisitor implementations to introspect on active worker count and possibly use that value to make dynamic decisions around scheduling.

--
MOS_MIGRATED_REVID=116351222

1 files changed, 11 insertions, 0 deletions

diff --git a/src/test/java/com/google/devtools/build/lib/concurrent/AbstractQueueVisitorTest.java b/src/test/java/com/google/devtools/build/lib/concurrent/AbstractQueueVisitorTest.java
index f9cf3ac32d..9b4f8fe608 100644
--- a/src/test/java/com/google/devtools/build/lib/concurrent/AbstractQueueVisitorTest.java
+++ b/src/test/java/com/google/devtools/build/lib/concurrent/AbstractQueueVisitorTest.java
@@ -55,6 +55,8 @@ public class AbstractQueueVisitorTest {
     counter.enqueue();
     counter.awaitQuiescence(/*interruptWorkers=*/ false);
     assertSame(10, counter.getCount());
+    assertSame(0, counter.activeParallelTasks());
+    assertSame(1, counter.getMaxRunningConcurrently());
   }
 
   @Test
@@ -483,6 +485,7 @@ public class AbstractQueueVisitorTest {
     private final static String THREAD_NAME = "BlazeTest CountingQueueVisitor";
 
     private int theInt = 0;
+    private int maxRunningConcurrently = 0;
     private final Object lock = new Object();
 
     public CountingQueueVisitor() {
@@ -503,6 +506,10 @@ public class AbstractQueueVisitorTest {
                   theInt++;
                   enqueue();
                 }
+                int concurrentTasks = activeParallelTasks();
+                if (concurrentTasks > maxRunningConcurrently) {
+                  maxRunningConcurrently = concurrentTasks;
+                }
               }
             }
           });
@@ -511,6 +518,10 @@ public class AbstractQueueVisitorTest {
     public int getCount() {
       return theInt;
     }
+
+    public int getMaxRunningConcurrently() {
+      return maxRunningConcurrently;
+    }
   }
 
   private static class ConcreteQueueVisitor extends AbstractQueueVisitor {