灵活的CountDownLatch？

Question

我现在两次遇到问题，生产者线程会产生N个工作项，将它们提交给an ExecutorService，然后需要等待，直到所有N个项都已处理完毕。

注意事项

• N事先未知 。如果是这样，我将简单地创建一个CountDownLatch然后具有生产者线程，await()直到完成所有工作。
• 使用a CompletionService是不合适的，因为尽管我的生产者线程需要阻塞（即，通过调用take()），但 无法表示所有工作都已完成 ，从而导致生产者线程停止等待。

我当前喜欢的解决方案是使用整数计数器，并在提交工作项时将其 递增 ，而在处理工作项时将其 递减
。提交所有N个任务之后，我的生产者线程将需要等待锁，检查是否counter == 0每当收到通知时。消费者线程将需要通知生产者，如果它已减少计数器且新值为0。

有没有更好的方法来解决这个问题，或者java.util.concurrent我应该使用合适的结构而不是“自己动手”？

提前致谢。

Answer 1

java.util.concurrent.Phaser看起来对您来说效果很好。它计划以Java
7发行，但是最稳定的版本可以在jsr166的兴趣小组网站上找到。

移相器是美化的循环屏障。您可以注册N个参与方，当您准备好在特定阶段等待其参加时。

关于它如何工作的简单示例：

final Phaser phaser = new Phaser();

public Runnable getRunnable(){
    return new Runnable(){
        public void run(){
            ..do stuff...
            phaser.arriveAndDeregister();
        }
    };
}
public void doWork(){
    phaser.register();//register self
    for(int i=0 ; i < N; i++){
        phaser.register(); // register this task prior to execution 
        executor.submit( getRunnable());
    }
    phaser.arriveAndAwaitAdvance();
}