JAVA 等待并发任务结束的几种方式
2021/6/14 20:23:07
本文主要是介绍JAVA 等待并发任务结束的几种方式,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
有时候需要并发执行一系列任务,等待所有的任务结束后再进行一些操作,下面介绍几种实现方式。
假设需要执行n个任务,任务为:
Runnable r = () -> { try { Thread.sleep(new Random().nextInt(1000)); } catch (InterruptedException e) { e.printStackTrace(); } };
1 暴力方式
List<Thread> threads = IntStream.range(0, n).boxed() .map(a -> new Thread(r)).collect(Collectors.toList()); threads.forEach(Thread::start); for (Thread thread : threads) { try { thread.join(); } catch (InterruptedException e) { e.printStackTrace(); } }
这种方式比较原始,如果业务使用线程池,就没法操作了。
2 改写业务代码
如果使用线程池提交任务,则可以使用CountDownLatch,每个任务执行完时,CountDownLatch#countDown, 提交完任务后,CountDownLatch#await,等待任务执行完。不过这种方式入侵了业务逻辑,不够优雅。
@AllArgsConstructor private class MyRunnable implements Runnable { private CountDownLatch latch; @Override public void run() { try { Thread.sleep(new Random().nextInt(1000)); } catch (InterruptedException e) { e.printStackTrace(); } finally { latch.countDown(); } } }
CountDownLatch latch = new CountDownLatch(n); ExecutorService executorService = Executors.newCachedThreadPool(); IntStream.range(0, n).boxed() .forEach(a -> executorService.submit(new MyRunnable(latch))); try { latch.await(); } catch (InterruptedException e) { e.printStackTrace(); }
3 不改写业务代码
使用CompletableFuture,可以避免在业务逻辑中嵌入任务控制逻辑,使用CompletableFuture#whenComplete可以在任务执行完后回调传入的逻辑,这样可以避免任务控制逻辑入侵业务逻辑。
CountDownLatch latch = new CountDownLatch(n); ExecutorService executorService = Executors.newCachedThreadPool(); IntStream.range(0, n).boxed() .forEach(a -> CompletableFuture.runAsync(r, executorService) .whenComplete(((aVoid, throwable) -> latch.countDown()))); try { latch.await(); } catch (InterruptedException e) { e.printStackTrace(); }
4 继续优化
上面的方式虽然可行,但不够优雅,需要用CountDownLatch来做控制。实际上使用CompletableFuture提供的接口,完全可以去掉CountDownLatch。
ExecutorService executorService = Executors.newCachedThreadPool(); CompletableFuture[] futures = IntStream.range(0, n).boxed() .map(a -> CompletableFuture.runAsync(r, executorService)) .toArray(CompletableFuture[]::new); CompletableFuture.allOf(futures).join();
* 如果任务有返回值,使用supplyAsync即可。
这篇关于JAVA 等待并发任务结束的几种方式的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-04TiDB 资源管控的对撞测试以及最佳实践架构
- 2024-07-03万字长文聊聊Web3的组成架构
- 2024-07-02springboot项目无法注册到nacos-icode9专业技术文章分享
- 2024-06-26结对编程到底难不难?答案在这里
- 2024-06-19《2023版Java工程师》课程升级公告
- 2024-06-15matplotlib作图不显示3D图,怎么办?
- 2024-06-1503-Loki 日志监控
- 2024-06-1504-让LLM理解知识 -Prompt
- 2024-06-05做软件测试需要懂代码吗?
- 2024-06-0514-ShardingSphere的分布式主键实现