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.
        }
    }
}