Introduce getWaitCount() method for FiberLock

- Determines the number of fibers waiting on the lock
lancaster-university · Oct 12, 2020 · e0089a9 · e0089a9
1 parent 72d3175
commit e0089a9
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/inc/core/CodalFiber.h b/inc/core/CodalFiber.h
@@ -359,6 +359,11 @@ namespace codal
          * Release the lock, and signal to all waiting fibers to continue
          */
         void notifyAll();
+
+        /**
+         * Determine the number of fibers currently blocked on this lock
+         */
+        int getWaitCount();
     };
 }
 

diff --git a/source/core/CodalFiber.cpp b/source/core/CodalFiber.cpp
@@ -1067,4 +1067,22 @@ void FiberLock::notifyAll()
     }
 
     locked = false;
+}
+
+
+/**
+ * Determine the number of fibers currently blocked on this lock
+ */
+int FiberLock::getWaitCount()
+{
+    Fiber *f = queue;
+    int count = 0;
+
+    while (f)
+    {
+        count++;
+        f = f->qnext;
+    }
+
+    return count;
 }