集合遍历使用多线程并行执行

发布时间 2023-08-28 16:30:10作者: fchhk

在业务中常常有这样的需求,需要遍历集合,分别处理,相互之前单独处理,互不影响。
为了充分利用服务器资源,可以使用多线程来执行。

parallelStream创建一个并行流,
使用ConcurrentHashMap使多线程情况下线程安全
CompletableFuture.allOf是所有线程全部执行完获得结果
CompletableFuture.runAsync没有返回值
CompletableFuture.supplyAsync有返回值

CompletableFuture.get() 和 CompletableFuture.join() 这两个方法是获取异步守护线程的返回值的。
get() 方法会抛出经检查的异常,可被捕获,自定义处理或者直接抛出。而 join() 会抛出未经检查的异常。

ConcurrentHashMap<Integer, Object> map = new ConcurrentHashMap<>();
List<Demo> list = new ArrayList<>();
CompletableFuture.allOf(list.parallelStream()
						.map(d -> CompletableFuture.runAsync(() -> {
							// do something
							map.put(d.getId(), Object);
						}, threadPoolTaskExecutor)).toArray(CompletableFuture[]::new)).join();
// 对最终集合做处理,比如插入
insert(new ArrayList<>(map.values()));