如何用StateFlow将Room返回的Flow<List>关联至UiState并自动更新

修改后的完整代码 #

class HomeViewModel(
    savedStateHandle: SavedStateHandle,
    private val categoryRepository: CategoryRepository,
    private val expenseRepository: ExpenseRepository
): ViewModel() {
    private val month: Int = savedStateHandle[HomeDestination.monthArgs] ?: Calendar.MONTH
    private val year: Int = savedStateHandle[HomeDestination.yearArgs] ?: Calendar.YEAR

    private val _categories: Flow<List<Category>> = categoryRepository.getAllCategoriesStream()
    private val _expenses: Flow<List<Expense>> = expenseRepository.getAllExpensesBetweenStream(
        LocalDate.of(year, month, 1).toEpochDay(),
        LocalDate.of(year, if (month + 1 == 13) 1 else month + 1, 1).toEpochDay()
    )

    private var _uiState = MutableStateFlow(UiState())
    val uiState: StateFlow<UiState> = _uiState.asStateFlow()

    init {
        viewModelScope.launch {
            // 同时监听分类和支出的Flow变化，合并后生成新UiState
            combine(_categories, _expenses) { categories, expenses ->
                UiState(
                    categories = categories.map { it.toCategoryUiState() },
                    expenses = expenses.map { it.toExpenseUiState() }
                )
            }.collect { newUiState ->
                _uiState.value = newUiState
            }
        }
    }

    data class UiState(
        val categories: List<CategoryUiState> = listOf(),
        val expenses: List<ExpenseUiState> = listOf()
    )
}

// 扩展函数：将数据模型转换为UI状态模型
private fun Category.toCategoryUiState(): CategoryUiState {
    // 根据实际的Category和CategoryUiState结构填充字段
    return CategoryUiState(
        id = this.id,
        name = this.name
        // 其他字段按需添加
    )
}

private fun Expense.toExpenseUiState(): ExpenseUiState {
    // 根据实际的Expense和ExpenseUiState结构填充字段
    return ExpenseUiState(
        id = this.id,
        amount = this.amount,
        date = this.date
        // 其他字段按需添加
    )
}

核心实现要点 #

• 协程作用域：使用viewModelScope.launch启动协程，协程会随ViewModel销毁自动取消，避免内存泄漏。
• Flow监听与更新：
  • 若需同时同步分类和支出数据，用combine合并两个Flow，任一Flow有更新时都会生成新的UiState。
  • 仅需同步支出数据时，可单独收集_expenses并更新状态：

    viewModelScope.launch {
        _expenses.collect { expenses ->
            _uiState.value = _uiState.value.copy(
                expenses = expenses.map { it.toExpenseUiState() }
            )
        }
    }
    

• 模型转换：必须将Room返回的Expense/Category数据模型转换为UI层专用的ExpenseUiState/CategoryUiState，隔离数据层与UI层，保证架构解耦。
• 状态更新：通过_uiState.value赋值或copy方法更新状态，确保StateFlow能正确推送新状态到UI层，实现数据同步。

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