Java消费者线程等待所有生产者线程完成

Question

我有一个生产者-消费者模式的多线程任务。可能有许多生产者和一个消费者。我使用ArrayBlockingQueue作为共享资源。

Producer类中的run（）方法：

public void run() {
         for (Order order : orderList) {
             try {
                 queue.put(order);
             } catch (InterruptedException e) {
                 e.printStackTrace();
             }
         }
         log.info("{} has placed all orders", Thread.currentThread().getName());
     }

Consumer类中的run（）方法：

     public void run() {
         while (!queue.isEmpty()) {
             try {
                 Order order = queue.take();
                 checkOrder(order);
             } catch (InterruptedException e) {
                 e.printStackTrace();
             }
         }
         log.info("All orders has been checked");
     }

main（）方法：

// creating N producers
         for (int i = 0; i < PRODUCERS_NUM; i++) {
             Producer producer = new Producer(orderQueue, orderList);
             new Thread(producer , "Producer " + i).start();
         }
 
         Thread consumerThread = new Thread(new Consumer(orderQueue, limitList), "Consumer");
         consumerThread.start();
         consumerThread.join();

**Printing results **

现在，当队列为空时，我有消费者结束条件。但是可能会有一段时间队列变成空的，但是一些生产者线程仍然在工作。所以我只需要在完成所有生产者线程之后才完成消费者线程（但它们的数量事先不知道）。

正确的编码方式是什么？

Answer 1

由于您有一个固定数量的生产者，我建议在开始时设置一个计数器与生产者的数量。每个生产者在计数器结束时递减计数器，而消费者只有在计数器达到零时才终止计数器。对于计数器，您希望使用原子整数。