如何检测多线程执行完毕
为了提高数据处理效率,您希望并发调用第三方接口,并在所有调用完成后发送短信。但是,使用 future.get() 会阻塞主线程,因为您需要等待所有线程执行完毕。
解决这个问题的一种方法是使用 countdownlatch。countdownlatch 提供一个计数器,它跟踪尚未完成的任务数量。
在多线程处理中,每个线程完成其任务后,使用 countdown() 方法递减计数器。当计数器变为零时,它表示所有任务都已完成。
这里是如何使用 countdownlatch 检测多线程执行完毕:
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 实例,并在所有 completablefuture 完成时执行一个操作。
completablefuture[] futures = new completablefuture[requestcount]; for (int i = 0; i < requestcount; i++) { futures[i] = completablefuture.runasync(() -> { // 调用第三方接口 // ... }); } // 所有请求完成后发送短信 completablefuture.allof(futures).thenrun(() -> { // 发送短信 // ... });
为了进一步提高并发性,您可以使用自定义线程池来执行异步任务,如下所示:
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()); // 所有任务完成后关闭线程池
通过使用 countdownlatch 或 completablefuture.allof,您可以异步处理任务,并在所有任务完成时执行后续操作,而不会阻塞主线程。