Merge pull request #167 from epickrram/master

[Java] Make synchronisation around shutdown latches more robust
real-logic · Feb 15, 2019 · bfdc6bd · bfdc6bd
2 parents dd098f3 + bc9d490
commit bfdc6bd
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/agrona/src/main/java/org/agrona/concurrent/ShutdownSignalBarrier.java b/agrona/src/main/java/org/agrona/concurrent/ShutdownSignalBarrier.java
@@ -47,7 +47,7 @@ public class ShutdownSignalBarrier
      */
     public ShutdownSignalBarrier()
     {
-        synchronized (this)
+        synchronized (LATCHES)
         {
             LATCHES.add(latch);
         }
@@ -58,19 +58,31 @@ public ShutdownSignalBarrier()
      */
     public void signal()
     {
-        synchronized (this)
+        synchronized (LATCHES)
         {
             LATCHES.remove(latch);
             latch.countDown();
         }
     }
 
+    /**
+     * Programmatically signal all awaiting threads.
+     */
+    public void signalAll()
+    {
+        synchronized (LATCHES)
+        {
+            LATCHES.forEach(CountDownLatch::countDown);
+            LATCHES.clear();
+        }
+    }
+
     /**
      * Remove the barrier from the shutdown signals.
      */
     public void remove()
     {
-        synchronized (this)
+        synchronized (LATCHES)
         {
             LATCHES.remove(latch);
         }