关于Kotlin Flow中unsafeFlow的collect函数调用逻辑的疑问

先把你代码里的小笔误修正，完整代码如下：

internal inline fun <T> unsafeFlow(crossinline block: suspend FlowCollector<T>.() -> Unit): Flow<T> {
    return object : Flow<T> {
        override suspend fun collect(collector: FlowCollector<T>) {
            collector.block()
        }
    }
}

suspend fun FlowCollector<Int>.loopFunc() {
    for (i in 1..5) {
        emit(i)
    }
}

val flow: Flow<Int> = unsafeFlow<Int> {
    loopFunc()
}

// 修正lambda箭头的笔误
flow.collect { value ->
    println(value)
}

下面一步步拆解整个调用链：

先明确两个核心类型的作用 #

• Flow<T>：数据流的抽象接口，只有一个suspend fun collect(collector: FlowCollector<T>)方法——这是触发数据流发射的唯一入口。
• FlowCollector<T>：数据收集器的抽象，核心方法是suspend fun emit(value: T)，用来接收并处理数据流发射的值。

从flow.collect { ... }开始追踪 #

当你调用flow.collect { value -> println(value) }时：

1. Kotlin会自动把这个lambda包装成一个FlowCollector<Int>的实现类，这个类的emit方法逻辑就是：收到值value后立即执行println(value)。
2. 这个自动生成的收集器实例，会被传入到unsafeFlow返回的匿名Flow对象的collect方法中。

进入unsafeFlow返回的Flow的collect方法 #

这个匿名Flow的collect方法只有一行代码：collector.block()。这里的block就是你传给unsafeFlow的那个{ loopFunc() }lambda。

关键在于block的类型是**suspend FlowCollector<T>.() -> Unit**——这是带接收者的挂起函数类型，意味着这个lambda必须依附在某个FlowCollector实例上执行。所以collector.block()本质就是把刚才那个负责打印的收集器，作为lambda的this上下文，去执行lambda里的代码。

执行lambda中的loopFunc()触发打印 #

loopFunc是FlowCollector<Int>的扩展挂起函数，所以当在lambda里调用loopFunc()时，就是以那个打印收集器作为this，执行循环里的emit(i)。

每调用一次emit(i)，就会触发收集器的emit方法，也就是执行println(i)——最终会依次打印1到5。

完整流程链路 #

flow.collect { println(it) } → 生成带打印逻辑的FlowCollector → 传入匿名Flow的collect方法 → 以收集器为上下文执行unsafeFlow的block → 调用loopFunc循环调用emit(i) → 每次emit触发收集器执行打印。

内容的提问来源于stack exchange，提问作者lannyf