Kotlinx.coroutines API 中的 collect 函数

collect 函数是 Kotlinx.coroutines 库中的一个内部 API，主要用于协程中的流收集操作。这个函数会阻塞当前协程，直到流产生一个新的值或者流结束。

语法

suspend fun <T> Flow<T>.collect(action: suspend (value: T) -> Unit): Unit

参数说明

• action: 接收一个 T 类型的参数，用于对每个流发送的值进行处理。

使用示例

import kotlinx.coroutines.*
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.flow.flow

fun main() = runBlocking<Unit> {
    // 创建一个计数器流
    val counter: Flow<Int> = flow {
        repeat(5) {
            emit(it)
        }
    }
    // 在协程中收集流中的值
    counter.collect { value ->
        println("The current value is $value")
    }
    println("Flow completed")
}

输出结果：

The current value is 0
The current value is 1
The current value is 2
The current value is 3
The current value is 4
Flow completed

注意事项

• collect 函数会在当前协程中持续阻塞，直到流产生新的值或者结束。因此，应该将它放在一个独立的协程中执行，以避免阻塞主线程或其他协程。
• 在流中的值被消费之前，流不会进行任何计算，因此流可能会阻塞整个应用程序。因此，在编写流时应该考虑减少这种情况发生的可能性。