如何在Java中编写并发任务池？实例分析

原创逃离我推掉我的手 2025-03-16 17:42 17阅读 0赞

在Java中，我们可以使用`ExecutorService`和`ThreadPoolExecutor`来创建一个并发任务池。这里以`ThreadPoolExecutor`为例进行实例分析。

首先，我们需要导入相关的包：

```java
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
```

然后，我们定义一个需要并发执行的的任务类（例如`Task`）：

```java
public class Task implements Runnable {
    private String taskName;

public Task(String taskName) {
        this.taskName = taskName;
    }

@Override
    public void run() {
        System.out.println("Starting task: " + taskName);
        // 这里可以执行你的任务逻辑
        System.out.println("Task finished: " + taskName);
    }
}
```

现在，我们就可以创建一个`ThreadPoolExecutor`并使用它来管理并发任务了：

```java
public class ThreadPoolExample {
    private ExecutorService executor;

public ThreadPoolExample(int threadCount) {
        // 创建线程池，这里设置为固定大小的线程池
        this.executor = Executors.newFixedThreadPool(threadCount);
    }

public void submitTask(Task task) {
        // 提交任务到线程池执行
        executor.submit(task);
    }

// 关闭线程池，等待所有任务完成后再关闭
    public void shutdown() {
        executor.shutdown();
        try {
            // 等待所有任务完成
            executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS));
        } catch (InterruptedException e) {
            System.err.println("Error while shutting down ExecutorService: " + e);
        }
    }

public static void main(String[] args) {
        ThreadPoolExample example = new ThreadPoolExample(5); // 创建一个包含5个线程的线程池

Task task1 = new Task("Task 1"));
        Task task2 = new Task("Task 2"));
        Task task3 = new Task("Task 3");
        Task task4 = new Task("Task 4");
        Task task5 = new Task("Task 5");

// 提交任务到线程池执行
        example.submitTask(task1);
        example.submitTask(task2);
        example.submitTask(task3);
        example.submitTask(task4);
        example.submitTask(task5);

// 关闭线程池，等待所有任务完成后再关闭
        example.shutdown();
    }
}
```

这个例子展示了如何创建一个固定大小的`ThreadPoolExecutor`，然后提交任务到线程池进行并发执行。最后，当所有任务完成后，关闭线程池。