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

Question

I have a multithreading task with producer-consumer pattern. There could be many producers and one consumer. I use ArrayBlockingQueue as a shared resource.

run() method in Producer class:

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());
     }

run() method in Consumer class:

     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() method:

// 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 **

Now I have consumer ending condition when queue is empty. But there could be a moment when queue becomes empty for a moment, but some producer threads are still working. So I need to finish consumer thread only AFTER ALL producer threads are finished (but their count is not known beforehand).

What is the proper way to code it?

Answer 1

As you have a fixed number of producers, I would recommend setting a counter at the start with the number of producers. Each producer decrements the counter when it is finished and the consumer only terminates when it reaches zero. For the counter, you would want to use an Atomic Integer .