Kotlin中合并Room DAO查询的两个Flow失败，求解决方案

场景说明 #

现有两个Room DAO查询方法，分别根据语言和位置返回Plant列表的Flow。需要将这两个Flow的结果按顺序合并为一个统一的Flow，供UI收集展示，且两个查询返回的Plant实例需区分展示。

DAO定义 #

@Query("SELECT plantId FROM PlantView WHERE language = :language")
fun getPlantByLang(language: String): Flow<List<Plant>>

@Query("SELECT plantId FROM PlantView WHERE location = :location")
fun getPlantByLoc(location: String): Flow<List<Plant>>

需求 #

实现getPlants方法，返回合并后的Flow，使UI能通过collectAsState统一收集：

val plantList by viewModel.getPlants().collectAsState(emptyList())

尝试过的方法及问题 #

1. onCompletion合并：仅返回第一个Flow的结果，因为Room的Flow是热流，永不触发onCompletion回调。

   fun getPlants (lang: String, loc: String) = getPlantByLang(lang).onCompletion{emitAll(getPlantByLoc(loc))}
   

2. flattenConcat合并：与onCompletion结果一致，第一个Flow永不完成，无法切换到第二个Flow。

   fun getPlants (lang: String, loc: String) = flowOf( getPlantByLang(lang), getPlantByLoc(loc) ).flattenConcat()
   

3. merge合并：会导致UI无限循环刷新，因为merge会分别发射两个Flow的列表值，而非合并后的完整列表，且每次数据库变化都会触发重复发射。

   fun getPlants (lang: String, loc: String) = merge( getPlantByLang(lang), getPlantByLoc(loc) )
   

解决方案 #

使用combine操作符，它会监听两个Flow的变化，当任意一个Flow发射新数据时，将两个最新的列表按顺序合并后发射。

合并为Flow<List>（推荐，适配UI收集方式） #

fun getPlants(lang: String, loc: String): Flow<List<Plant>> = combine(
    getPlantByLang(lang),
    getPlantByLoc(loc)
) { plantsByLang, plantsByLoc ->
    // 按顺序合并：先语言查询结果，后位置查询结果
    plantsByLang + plantsByLoc
}

若需返回Flow（逐个发射Plant实例） #

fun getPlants(lang: String, loc: String): Flow<Plant> = combine(
    getPlantByLang(lang),
    getPlantByLoc(loc)
) { plantsByLang, plantsByLoc ->
    plantsByLang + plantsByLoc
}.flatMapConcat { combinedList ->
    combinedList.asFlow()
}

区分Plant来源（适配差异化展示需求） #

如果需要区分两种查询返回的Plant，可在合并时添加标记（假设Plant有source字段）：

fun getPlants(lang: String, loc: String): Flow<List<Plant>> = combine(
    getPlantByLang(lang),
    getPlantByLoc(loc)
) { plantsByLang, plantsByLoc ->
    val markedLangPlants = plantsByLang.map { it.copy(source = "language") }
    val markedLocPlants = plantsByLoc.map { it.copy(source = "location") }
    markedLangPlants + markedLocPlants
}

原理说明 #

Room查询返回的Flow是热流，会持续监听数据库变化并发射最新数据，永不完成。combine操作符正好适配这种场景：它不会依赖上游Flow完成，而是实时监听两个Flow的更新，每次任意一个Flow有新数据时，就基于两个Flow的最新值生成合并结果，完美满足顺序合并+实时更新的需求。

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