Java - 优雅地退出线程

Question

我有一堆执行计算的线程。它们使用 CyclicBarrier 进行“同步”。当任何线程的run()方法完成时，我希望所有其他线程在调用await()后也退出代码> 下次就在障碍物上。

到目前为止，我尝试过的所有操作要么在 await() 调用处挂起，要么导致屏障损坏。有什么建议吗？

编辑：这是（基本）代码：

public MyClass implements Runnable {
    public void run() {
        while (true) {
            if (someCondition) {
                // quit other threads when they call await()
                return;
            }
            barrier.await();
    }
}

Answer 1

reset() 将唤醒所有等待线程并抛出异常，

然后您可以使用await，这样

private static volatile boolean shouldStop=false;

public void run() {
    try{
        while (true) {
            if (someCondition) {
                // quit other threads when they call await()
                return;
            }
            try{
                if(shouldStop)return;
                barrier.await();
            }catch(BrokenBarrierException e){
                //someone stopped 
                return;
            }
       }
   }finally{
       shouldStop =true;
       barrier.reset();
   }
}

您还可以调用 if(shouldStop) 检查的方法

Answer 2

从听起来你可能想要一个 CountDownLatch< /a>.假设您知道线程/参与者的数量，您只需为那么多线程/参与者创建一个，然后当您的线程完成倒计时并等待锁存器时：

final int workers = …
final CountDownLatch latch = new CountDownLatch(workers);

void doSomething() throws InterruptedException {
  …
  latch.countDown();
  latch.await(); // blocks, throws InterruptedException
}

与 CyclicBarrier 相比，CountDownLatch不可重复使用，只能使用一次。然而，它确实将等待和释放问题分开，因此您可以拥有另一个允许线程通过的线程。

话虽如此，如果您确实需要 CyclicBarrier，上述代码的细微变化应该可以工作：

final int workers = …
final CyclicBarrier barrier = new CyclicBarrier(workers);

void doSomething() throws InterruptedException, BrokenBarrierException {
  …
  latch.await(); // blocks, throws InterruptedException, BrokenBarrierException
}

但是，如果任何线程被中断或 barrier.reset() 被调用，然后屏障被破坏并抛出异常。