如何合并两个Android Room DAO查询函数以获取带事件数的日期列表

问题根因 #

你当前写法的问题是LiveData的观察者回调是异步执行的：你调用observe()方法后并不会立刻执行回调里的代码，而是会等数据库查询完成后才会回调赋值。所以你在observe()调用结束后立刻读取dateCount，此时回调还没触发，拿到的自然是初始值0。
另外你在循环里反复注册Observer也会引发内存泄露、重复观察的问题，非常不推荐。

解决方案1：直接修改Room查询语句（最优） #

你完全可以用SQL的GROUP BY语法，直接一次性查询出所有日期对应的事件数量，不需要分两次查，也不需要手动合并LiveData。
首先新增一个数据类用来接收查询结果：

data class DateWithEventCount(
    val date: String,
    val count: Int
)

然后在Dao里新增这个查询方法：

@Query("SELECT date, COUNT(*) as count FROM event GROUP BY date")
fun getDatesWithEventCount(): LiveData<List<DateWithEventCount>>

之后你只需要在ViewModel里暴露这个LiveData，在Fragment里观察这一个LiveData就能直接拿到所有日期和对应的数量，直接传给Adapter使用即可，不需要额外做数据合并。

解决方案2：合并多个LiveData（适用必须拆分查询的场景） #

如果你有特殊需求不能修改SQL，可以用MediatorLiveData来合并多个LiveData的结果，逻辑统一放到ViewModel里处理，不要放在Fragment中。
ViewModel示例代码：

class EventsListViewModel : ViewModel() {
    // 合并后的结果LiveData
    val datesWithCountLiveData = MediatorLiveData<List<DateForEvents>>()

    init {
        // 先添加日期列表的数据源
        datesWithCountLiveData.addSource(getDates()) { dateList ->
            dateList ?: return@addSource
            // 清理之前绑定的计数数据源，避免重复回调
            val sourcesToRemove = datesWithCountLiveData.sources.filterIsInstance<LiveData<Int?>>()
            sourcesToRemove.forEach { datesWithCountLiveData.removeSource(it) }

            val resultMap = mutableMapOf<String, Int>()
            val validDates = dateList.filterNotNull()
            validDates.forEach { date ->
                // 为每个日期的计数LiveData添加数据源
                val countLiveData = getEventsCount(date)
                datesWithCountLiveData.addSource(countLiveData) { count ->
                    count ?: return@addSource
                    resultMap[date] = count
                    // 所有日期的计数都返回后再更新合并结果
                    if (resultMap.size == validDates.size) {
                        val finalList = resultMap.map { (date, count) ->
                            DateForEvents(Date(date), count)
                        }
                        datesWithCountLiveData.postValue(finalList)
                    }
                }
            }
        }
    }

    // 原有Dao调用封装
    fun getDates(): LiveData<List<String?>> = eventDao.getDates()
    fun getEventsCount(date: String): LiveData<Int?> = eventDao.getEventsCount(date)
}

之后在Fragment里只需要观察datesWithCountLiveData这一个LiveData，拿到结果直接给Adapter即可。

额外优化建议 #

• 不要在Fragment中处理数据合并逻辑，所有数据处理逻辑都应该放到ViewModel中，Fragment只负责观察数据更新UI
• 避免在循环中注册Observer，重复注册Observer会导致内存泄露和重复回调的问题
• 优先用SQL语句完成数据聚合，性能远高于在应用层合并数据

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