当需要利用多线程快速处理大量的任务时,开发者常常面临如何判断所有线程执行完毕的问题。过早的判断会导致数据不完整,而阻塞主线程又不合理。
对于这种情况,有几种有效的方法:
countdownlatch
立即学习“Java免费学习笔记(深入)”;
使用 countdownlatch 可以实现计数器的功能。每次一个线程执行完毕,就递减计数器。当计数器为 0 时,表示所有线程都已执行完毕。
示例代码:
int requestcount = 1000; countdownlatch latch = new countdownlatch(requestcount); for (int i = 0; i < requestcount; i++) { new thread(() -> { // 调用第三方接口 // ... // 操作完成,递减计数器 latch.countdown(); }).start(); } // 新开一个线程等待所有请求完成后发送短信 new thread(() -> { try { latch.await(); // 所有请求完成,发送短信 // ... } catch (interruptedexception e) { e.printstacktrace(); } }).start();
completablefuture.allof
completablefuture.allof 方法可以将多个 completablefuture 组合在一起。当所有子任务都完成后,才执行指定的动作。
示例代码:
completablefuture[] futures = new completablefuture[requestcount]; for (int i = 0; i < requestcount; i++) { futures[i] = completablefuture.runasync(() -> { // 调用第三方接口 // ... }); } // 所有请求完成后发送短信 completablefuture.allof(futures).thenrun(() -> { // 发送短信 // ... });
自定义线程池
当需要更精细的控制线程并行度时,可以使用自定义线程池。completablefuture 的 runasync 方法可以指定线程池参数。
示例代码:
ExecutorService executor = Executors.newFixedThreadPool(100); CompletableFuture[] futures = new CompletableFuture[requestCount]; for (int i = 0; i < requestCount; i++) { futures[i] = CompletableFuture.runAsync(() -> { // 调用第三方接口 // ... }, executor); // 用自定义线程池 } // 所有请求完成后发送短信 CompletableFuture.allOf(futures).thenRun(() -> { // 发送短信 // ... }).thenRun(() -> executor.shutdown()); // 所有任务完成后关闭线程池