java 集合遍历多线程

原创

mob64ca12d61d6b 2023-10-20 05:09:06 ©著作权

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

Java集合遍历多线程实现指南

引言

在Java开发中，我们经常需要对集合进行遍历操作。而当集合较大时，使用单线程进行遍历会导致性能下降，因此多线程遍历集合成为了一种常用的优化手段。本文将指导一位刚入行的小白如何实现Java集合的多线程遍历。

流程图

flowchart TD
    A(开始)
    B(创建线程池)
    C(分割集合)
    D(创建任务)
    E(提交任务到线程池)
    F(等待任务执行完成)
    G(合并结果)
    H(结束)
    A --> B
    B --> C
    C --> D
    D --> E
    E --> F
    F --> G
    G --> H

步骤说明

创建线程池：使用Executors.newFixedThreadPool()方法创建一个固定大小的线程池，线程池的大小根据实际情况进行调整。
```
// 创建线程池
ExecutorService executorService = Executors.newFixedThreadPool(4);
```

分割集合：将要遍历的集合按照线程数平均分割成若干子集合。

// 分割集合
List<List<T>> subLists = new ArrayList<>();
int size = collection.size();
int batchSize = size / threadCount;
int remainder = size % threadCount;
Iterator<T> iterator = collection.iterator();
for (int i = 0; i < threadCount; i++) {
    List<T> subList = new ArrayList<>();
    int count = batchSize + (i < remainder ? 1 : 0);
    for (int j = 0; j < count && iterator.hasNext(); j++) {
        subList.add(iterator.next());
    }
    subLists.add(subList);
}

创建任务：将每个子集合作为一个任务，并实现Runnable接口的run()方法。

// 创建任务
List<Runnable> tasks = new ArrayList<>();
for (List<T> subList : subLists) {
    Runnable task = () -> {
        for (T item : subList) {
            // 处理每个元素的逻辑
        }
    };
    tasks.add(task);
}

提交任务到线程池：使用线程池的submit()方法提交每个任务到线程池中执行。

// 提交任务到线程池
List<Future<?>> futures = new ArrayList<>();
for (Runnable task : tasks) {
    Future<?> future = executorService.submit(task);
    futures.add(future);
}

等待任务执行完成：使用Future对象的get()方法等待每个任务执行完成。

// 等待任务执行完成
for (Future<?> future : futures) {
    try {
        future.get();
    } catch (InterruptedException | ExecutionException e) {
        // 处理异常情况
    }
}

合并结果：根据实际需求，将每个子任务的执行结果进行合并。
```
// 合并结果
// 这里略去合并结果的具体实现
```
关闭线程池：使用executorService.shutdown()方法关闭线程池。
```
// 关闭线程池
executorService.shutdown();
```