如何等待主线程直到异步方法完成？

Question

我有 Service.class 和 start() 方法：

public void start()  {  
    for (int i = 0; i < companiesList.size(); i++) {
        asychronous.someAsynchronous(...);
    }
    log.info("Start method has finished");
}

我有 Asynchronous.class 和 someAsynchronous() 方法：

@Async("threadPoolTaskExecutor")
public CompletableFuture<Void> someAsynchronous(some_parameters) {
    //do some stuff
    return null;
}

log.info() 在 someAsynchronous() 方法完成之前出现。 如何强制它等待log.info() 直到循环中的someSynchronous() 方法完成？顺便说一句：完成循环后异步线程仍在运行。

Answer 1

CompletableFuture<Void> 调用被要求执行，但在它们都在单独的线程中启动之后，for 循环结束并且甚至在它们中的任何一个完成之前打印日志。这就是异步处理的优势——您不必关心它们的结果以及它们执行的时间。

为了实现你想要的，你必须定期检查它们是否所有都完成了，然后再进行日志输出。

// Adds executions to the List
List<CompletableFuture<Void>> futures = new ArrayList<>();
for (int i = 0; i < companiesList.size(); i++) {
    futures.add(asychronous.someAsynchronous(...));
}

// Periodical check
Iterator<CompletableFuture<Void>> iterator = futures.iterator();
while (iterator.hasNext()) {
    CompletableFuture<Void> future = iterator.next(); // get the next one
    if (future.isDone()) {                            // if finished...
        //...                                         // ... do an action
        iterator.remove();                            // ... and remove from the Iterator
    }
    if (!iterator.hasNext()) {                        // if you reach the end
        iterator = futures.iterator();                // ... repeat the remaining Futures
    }
}

log.info("Start method has finished");

请注意，在所有执行完成之前，此方法不会结束。

---

编辑：感谢 @Kayaman，他建议使用单个 method 专用于替换整个 Iterator 逻辑。 futures 必须是一个数组：

// Adds executions to the List
List<CompletableFuture<Void>> futures = new ArrayList<>();
for (int i = 0; i < companiesList.size(); i++) {
    futures.add(asychronous.someAsynchronous(...));
}

// Join the completion of all the threads
CompletableFuture.allOf(futures.toArray(new CompletableFuture[0])).join();

log.info("Start method has finished");