如何正确等待Flow收集完成？ViewModel加载Flow数据遇初始化问题

原写法的核心问题 #

1. 协程异步导致未初始化异常：init块里启动协程调用dataLoader()是异步执行的，紧接着就调用doSomething(loadedData[0])，这时候loadedData根本还没被赋值，必然会抛出UninitializedPropertyAccessException。
2. 状态管理冗余且易不一致：单独维护loadedData和state两个状态，后续数据更新时容易出现两者不同步的情况，违背了单一状态源的设计原则。
3. Flow使用不当：数据库返回的Flow（比如Room的Flow）是冷流+持续发射的，只要数据有更新就会重新发射值，用collect持续赋值给loadedData没问题，但没必要单独存，应该直接同步到StateFlow里。

修正后的实现方式 #

方案1：监听数据变化并同步更新状态（推荐，支持数据实时刷新） #

直接在协程里收集数据库Flow，每次拿到数据就更新StateFlow，同时完成后续处理：

class MyViewModel: ViewModel() {
    private var _state = MutableStateFlow(MyState())
    val state = _state.asStateFlow()
    private val myDataSource: MyDataSource

    init {
        viewModelScope.launch {
            // 收集数据库Flow，数据变化时自动更新状态
            myDataSource.getData().collect { dataList ->
                // 处理数据并更新状态
                val targetValue = dataList.firstOrNull() ?: defaultValue // 替换成你的默认值
                _state.value = _state.value.copy(thing = targetValue)
            }
        }
    }
}

方案2：仅获取首次加载的数据（不需要监听后续变化） #

如果只需要数据库的初始数据，不需要监听后续更新，可以用first()获取Flow的第一个值：

class MyViewModel: ViewModel() {
    private var _state = MutableStateFlow(MyState())
    val state = _state.asStateFlow()
    private val myDataSource: MyDataSource

    init {
        viewModelScope.launch {
            // 获取首次加载的数据
            val loadedData = myDataSource.getData().first()
            // 执行后续处理
            doSomething(loadedData)
        }
    }

    private fun doSomething(loadedData: List<DataType>){
        val targetValue = loadedData.firstOrNull() ?: defaultValue // 替换成你的默认值
        _state.value = _state.value.copy(thing = targetValue)
    }
}

关键注意点 #

• 永远不要在协程外直接访问协程内初始化的lateinit变量，因为你无法保证协程执行的时机。
• ViewModel的状态应该统一通过StateFlow管理，避免分散的状态变量，减少数据不一致的风险。
• 如果数据库Flow可能返回空列表，一定要用firstOrNull()代替[0]，避免数组越界异常。

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