Java中的BlockingQueue-BlockingQueue代码示例

时间：2024-02-20 09:50:07

下面是一个简单的Java代码示例，展示了如何使用BlockingQueue来实现一个基本的生产者-消费者模型。

首先，我们定义一个Producer类，它负责向BlockingQueue中添加数据：

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 {
            for (int i = 0; i < 10; i++) {
                System.out.println("Produced: " + i);
                queue.put(i); // 如果队列满了，线程会阻塞
                Thread.sleep(1000); // 模拟生产耗时
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}

然后，我们定义一个Consumer类，它负责从BlockingQueue中取出数据：

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) {
                Integer item = queue.take(); // 如果队列为空，线程会阻塞
                System.out.println("Consumed: " + item);
                Thread.sleep(1500); // 模拟消费耗时
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}

最后，我们在主程序中创建BlockingQueue，并启动生产者和消费者线程：

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;

public class BlockingQueueExample {
    public static void main(String[] args) {
        BlockingQueue<Integer> queue = new LinkedBlockingQueue<>(5); // 创建一个有界队列，容量为5

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

        // 启动生产者线程
        new Thread(producer, "Producer").start();

        // 启动消费者线程
        new Thread(consumer, "Consumer").start();
    }
}

在这个例子中，Producer和Consumer类都实现了Runnable接口，并分别重写了run方法。我们创建了一个LinkedBlockingQueue实例，并指定了队列的容量为5。然后，我们创建了一个生产者线程和一个消费者线程，并启动它们。

生产者线程会向队列中添加整数（从0到9），而消费者线程会从队列中取出整数并处理。如果队列满了，生产者线程会阻塞，直到队列中有空间；如果队列为空，消费者线程会阻塞，直到队列中有元素可取。

注意，在实际应用中，你可能需要更复杂的逻辑来处理线程的生命周期和异常处理，以及确保资源的正确释放。此外，为了控制生产者和消费者的数量，你可能需要使用线程池（如ExecutorService）来管理这些线程。

相关文章