java队列多线程

原创

mob64ca12d78ba3 2024-08-15 06:31:09 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12d78ba3的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java队列与多线程编程指南

在多线程编程中，使用队列是一种常见且有效的方式来管理线程间的任务调度。本文将以实例的方式指导你如何在Java中实现一个简单的多线程队列。我们将使用BlockingQueue类，它支持线程安全的添加和移除元素。

流程概述

我们将按照以下步骤进行实现：

步骤	描述
1. 创建队列	使用`BlockingQueue`创建一个任务队列
2. 创建工作线程	创建多个消费者线程，负责处理队列中的任务
3. 创建生产者线程	创建多个生产者线程，负责将任务放入队列中
4. 启动线程	启动生产者和消费者线程
5. 停止程序	在适当的时机终止程序

接下来，让我们详细了解每个步骤及其实现代码。

1. 创建队列

首先，我们需要创建一个队列用于存储任务。我们将使用ArrayBlockingQueue作为我们的任务队列。

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

// 创建一个阻塞队列，队列大小为10
BlockingQueue<String> queue = new ArrayBlockingQueue<>(10);

BlockingQueue<String>：表示我们将队列中存储字符串类型的任务。

2. 创建工作线程

我们需要定义一个消费者类，该类将实现Runnable接口。消费者线程将从队列中获取任务并执行。

class Consumer implements Runnable {
    private BlockingQueue<String> queue;

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

    @Override
    public void run() {
        try {
            while (true) {
                String task = queue.take(); // 从队列中获取任务
                System.out.println("Consumed: " + task);
                // 模拟任务处理时间
                Thread.sleep(1000);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt(); // 恢复中断状态
        }
    }
}

queue.take()：该方法会阻塞直到有任务可供消费。
Thread.sleep(1000)：模拟任务处理时间。

3. 创建生产者线程

生产者线程将不断地向队列中添加任务。我们同样需要实现一个Runnable接口。

class Producer implements Runnable {
    private BlockingQueue<String> queue;

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

    @Override
    public void run() {
        try {
            for (int i = 0; i < 10; i++) {
                String task = "Task " + i;
                queue.put(task); // 向队列中添加任务
                System.out.println("Produced: " + task);
                // 模拟生产任务的时间
                Thread.sleep(500);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt(); // 恢复中断状态
        }
    }
}

queue.put(task)：该方法会阻塞直到队列中有空间可供添加任务。

4. 启动线程

接下来，我们将创建线程并启动它们。我们会创建一个生产者和多个消费者。

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

        // 创建并启动生产者线程
        Thread producerThread = new Thread(new Producer(queue));
        producerThread.start();

        // 创建并启动消费者线程
        for (int i = 0; i < 3; i++) {
            Thread consumerThread = new Thread(new Consumer(queue));
            consumerThread.start();
        }
    }
}

Thread producerThread... start()：启动生产者线程。
Thread consumerThread... start()：启动多个消费者线程。

5. 停止程序

在实际应用中，我们应该有一些机制来优雅地停止程序，但为了简化，我们将不在此实现完全的停止逻辑。

状态图

在我们的程序执行时，状态转换可以用以下状态图表示：

stateDiagram
    [*] --> Idle
    Idle --> Producing : Start Producer
    Producing --> Consuming : Produce Task
    Consuming --> Idle : Task Completed
    Idle --> [*]

任务分配饼状图

我们可以用饼状图来表示生产者和消费者之间的任务分配情况。在这里，假设生产者生产了10个任务，3个消费者均匀处理这些任务。

pie
    title Task Distribution
    "Produced Tasks": 10
    "Consumed by Consumer 1": 3
    "Consumed by Consumer 2": 3
    "Consumed by Consumer 3": 4