合并Kotlin流

时间:2019-08-28 13:17:41

标签: kotlin kotlin-coroutines

给出2个或更多相同类型的流,是否存在现有的Kotlin协程函数来合并它们,例如RX合并运算符?

目前我正在考虑:

fun <T> merge(vararg flows: Flow<T>): Flow<T> = channelFlow {
    val flowJobs = flows.map { flow ->
        GlobalScope.launch { flow.collect { send(it) } }
    }
    flowJobs.joinAll()
}

但看起来有些笨拙。

3 个答案:

答案 0 :(得分:11)

这是Coroutines库的一部分(撰写本文时,Coroutines版本1.3.5)。

您可以这样使用它:

val flowA = flow { emit(1) } 
val flowB = flow { emit(2) }

merge(flowA, flowB).collect{ println(it) } // Prints two integers
// or:
listOf(flowA, flowB).merge().collect { println(it) } // Prints two integers

您可以在the source code

中阅读更多内容

答案 1 :(得分:5)

我对流程还不太熟悉,所以这可能不是最理想的。无论如何,我认为您可以创建所有输入流的流,然后使用flattenMerge将它们再次压平为单个流。像这样:

fun <T> merge(vararg flows: Flow<T>): Flow<T> = flowOf(*flows).flattenMerge()

答案 2 :(得分:1)

可能已经晚了,但我相信这可能是一个可行的解决方案:

fun <T> combineMerge(vararg flows: Flow<T>) = flow {
    coroutineScope {
        flows.forEach {
            launch {
                it.collect {
                    emit(it)
                }
            }
        }
    }
}

fun <T> combineConcat(vararg flows: Flow<T>) = flow {
    flows.forEach {
        it.collect {
            emit(it)
        }
    }
}