Kotlin Flow如何合并两个流，仅在第一个流发射元素时输出组合结果

你要实现的逻辑等价于响应式编程中的withLatestFrom操作效果：仅当第一个流（FlowA）发射元素时，才组合两个流的最新值输出，FlowB的独立发射不会触发任何输出。

Kotlin 协程 Flow 实现 #

方案1：自定义实现（性能更优，无冗余计算） #

核心思路是将FlowB转为状态流缓存最新值，仅订阅FlowA的发射事件，每次FlowA发射时直接取FlowB的缓存值组合输出：

import kotlinx.coroutines.flow.*
import kotlinx.coroutines.currentCoroutineScope

fun <A, B> Flow<A>.withLatestFrom(other: Flow<B>): Flow<Pair<A, B>> = flow {
    // 缓存FlowB的最新值，随当前作用域生命周期自动销毁
    val otherState = other.stateIn(
        scope = currentCoroutineScope(),
        started = SharingStarted.WhileSubscribed(),
        initialValue = null
    )
    collect { valueA ->
        // 若要求FlowB至少发射过一次才输出，保留下面的判断；不需要可删除，改为直接取其他默认值
        val valueB = otherState.value ?: return@collect
        emit(valueA to valueB)
    }
}

方案2：基于内置combine实现（代码更简洁） #

用内置combine组合两个流后，过滤掉FlowA未发生变化的输出即可：

val combinedFlow = combine(flowA, flowB) { a, b -> a to b }
    .distinctUntilChanged { old, new -> old.first == new.first }

注意事项 #

• 两种方案都默认要求FlowB至少发射过一次元素才会输出组合结果，如果你允许FlowB未初始化时用默认值，直接修改stateIn的initialValue参数即可
• 如果你用的是其他响应式框架：
  • RxJava/ RxJS 直接调用内置的withLatestFrom操作符即可
  • Swift Combine 可以用combineLatest加过滤，或者使用第三方实现的withLatestFrom扩展

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