DEV Community

Anh Trần Tuấn
Anh Trần Tuấn

Posted on • Originally published at tuanh.net on

Methods to Solve the Producer-Consumer Problem in Java

1. Understanding the Producer-Consumer Problem

Before diving into solutions, let's break down the core concepts.

1.1 What is the Producer-Consumer Problem?

The Producer-Consumer problem is about managing a shared resource (a buffer) between two types of processes. Producers add items to the buffer, while consumers remove items. Proper synchronization is essential to avoid issues like buffer overflow or underflow.

1.2 Why is It Important?

Efficiently solving the Producer-Consumer problem is crucial for applications that involve tasks like data processing, networking, and multi-threaded operations. Proper handling ensures smooth and reliable operations without resource wastage or contention.

2. Common Solutions in Java

Java provides several mechanisms to address the Producer-Consumer problem, each with its own advantages and scenarios for use.

2.1 Using wait() and notify()

Java's wait() and notify() methods are traditional tools for managing synchronization. Here’s how you can use them:

Producer Class

import java.util.LinkedList;

public class Producer implements Runnable {
    private final LinkedList<Integer> buffer;
    private final int BUFFER_SIZE;

    public Producer(LinkedList<Integer> buffer, int size) {
        this.buffer = buffer;
        this.BUFFER_SIZE = size;
    }

    @Override
    public void run() {
        try {
            while (true) {
                synchronized (buffer) {
                    while (buffer.size() == BUFFER_SIZE) {
                        buffer.wait();
                    }
                    int item = produceItem();
                    buffer.add(item);
                    System.out.println("Produced: " + item);
                    buffer.notifyAll();
                }
                Thread.sleep(1000);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }

    private int produceItem() {
        return (int) (Math.random() * 100);
    }
}
Enter fullscreen mode Exit fullscreen mode

Consumer Class

import java.util.LinkedList;

public class Consumer implements Runnable {
    private final LinkedList<Integer> buffer;

    public Consumer(LinkedList<Integer> buffer) {
        this.buffer = buffer;
    }

    @Override
    public void run() {
        try {
            while (true) {
                synchronized (buffer) {
                    while (buffer.isEmpty()) {
                        buffer.wait();
                    }
                    int item = buffer.removeFirst();
                    System.out.println("Consumed: " + item);
                    buffer.notifyAll();
                }
                Thread.sleep(1000);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}
Enter fullscreen mode Exit fullscreen mode

Main Class

import java.util.LinkedList;

public class Main {
    public static void main(String[] args) {
        LinkedList<Integer> buffer = new LinkedList<>();
        int bufferSize = 10;

        Producer producer = new Producer(buffer, bufferSize);
        Consumer consumer = new Consumer(buffer);

        new Thread(producer).start();
        new Thread(consumer).start();
    }
}
Enter fullscreen mode Exit fullscreen mode

Demo Results

In this setup, the producer and consumer operate on a shared buffer. The producer will add items to the buffer while the consumer removes them. The buffer size is controlled to prevent overflow and underflow, ensuring smooth operation.

2.2 Using BlockingQueue

Java’s BlockingQueue interface provides a more robust solution by handling synchronization internally.

Producer Class

import java.util.concurrent.BlockingQueue;

public class Producer implements Runnable {
    private final BlockingQueue<Integer> queue;

    public Producer(BlockingQueue<Integer> queue) {
        this.queue = queue;
    }

    @Override
    public void run() {
        try {
            while (true) {
                int item = produceItem();
                queue.put(item);
                System.out.println("Produced: " + item);
                Thread.sleep(1000);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }

    private int produceItem() {
        return (int) (Math.random() * 100);
    }
}
Enter fullscreen mode Exit fullscreen mode

Consumer Class

import java.util.concurrent.BlockingQueue;

public class Consumer implements Runnable {
    private final BlockingQueue<Integer> queue;

    public Consumer(BlockingQueue<Integer> queue) {
        this.queue = queue;
    }

    @Override
    public void run() {
        try {
            while (true) {
                int item = queue.take();
                System.out.println("Consumed: " + item);
                Thread.sleep(1000);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}
Enter fullscreen mode Exit fullscreen mode

Main Class

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

public class Main {
    public static void main(String[] args) {
        BlockingQueue<Integer> queue = new ArrayBlockingQueue<>(10);

        Producer producer = new Producer(queue);
        Consumer consumer = new Consumer(queue);

        new Thread(producer).start();
        new Thread(consumer).start();
    }
}
Enter fullscreen mode Exit fullscreen mode

In this approach, the BlockingQueue handles the buffer size and synchronization automatically. The producer and consumer interact with the queue without needing explicit synchronization.

3. Conclusion

Understanding and solving the Producer-Consumer problem is essential for effective multi-threaded programming. Java provides various tools, from manual synchronization with wait() and notify(), to the more streamlined BlockingQueue. Choose the method that best fits your application's needs.

If you have any questions or need further clarification, feel free to leave a comment below!

Read posts more at : Methods to Solve the Producer-Consumer Problem in Java

Top comments (0)