Kotlin 协程障碍：等待所有协程完成

Question

我需要在 for 循环中启动许多协程，并在所有任务完成后在主线程中获取回调。

做这个的最好方式是什么？

//Main thread
fun foo(){
    messageRepo.getMessages().forEach {message->
       GlobalScope.launch {
            doHardWork(message)
       }
    }
   // here I need some callback to the Main thread that all work is done.
}

在 CoroutineScope 中没有迭代消息的变体。 迭代必须在主线程中完成。

Answer 1

您可以等到所有任务都使用awaitAll完成，然后使用withContext在主线程中执行您的回调

fun foo() {
    viewModelScope.launch {
        messageRepo.getMessages().map { message ->
            viewModelScope.async(Dispatchers.IO) {
                doHardWork(message)
            }
        }.awaitAll()
        withContext(Dispatchers.Main) {
            // here I need some callback that all work is done.
        }
    }
}