Android中如何在ViewModel内使用StateIn挂起函数？

问题根源 #

你报错的核心原因是：employeeRepository.fetch()是挂起函数，而非Flow类型，而.map()、.stateIn()都是Flow的扩展函数，只有Flow对象才能调用这些方法。

正确实现方式 #

我们需要先把挂起函数的调用包装成Flow，再通过stateIn转换成StateFlow，分两种场景处理：

场景1：修改仓库层（推荐）

把仓库的fetch方法改成返回Flow，将API调用和异常处理统一放在仓库层：

// 仓库代码
fun fetch(): Flow<List<Employee>> = flow {
    try {
        val result = employeesApiService.fetchEmployees()
        emit(result)
    } catch (e: Exception) {
        // 这里可以选择emit空列表、异常状态，或者抛出异常让ViewModel处理
        emit(emptyList())
        // 或者：throw e
    }
}

然后ViewModel中直接使用：

// ViewModel代码
val employeeState: StateFlow<EmployeeDataState> =
    employeeRepository.fetch()
        .map(EmployeeDataState::Success)
        .catch { emit(EmployeeDataState.Error("加载失败：${it.message}")) }
        .stateIn(
            scope = viewModelScope,
            started = SharingStarted.WhileSubscribed(5_000),
            initialValue = EmployeeDataState.Loading
        )

场景2：不修改仓库层，在ViewModel中转换

如果不想改动仓库，用flow构建器包裹挂起函数调用，将其转换为Flow：

// ViewModel代码
val employeeState: StateFlow<EmployeeDataState> = flow {
    // flow构建器是挂起函数，可直接调用挂起的fetch方法
    val employees = employeeRepository.fetch()
    emit(employees)
}
    .map(EmployeeDataState::Success)
    .catch { emit(EmployeeDataState.Error("加载失败：${it.message}")) }
    .stateIn(
        scope = viewModelScope,
        started = SharingStarted.WhileSubscribed(5_000),
        initialValue = EmployeeDataState.Loading
    )

关于stateIn的挂起重载 #

stateIn的挂起重载(suspend fun Flow<T>.stateIn(...))不需要指定初始值，它会挂起直到Flow发出第一个值，再将该值作为初始值创建StateFlow。但这种方式不适合ViewModel初始化，因为ViewModel初始化阶段不能挂起，若要使用可结合lazyDeferred延迟初始化：

// ViewModel代码
private fun <T> lazyDeferred(block: suspend () -> T) = lazy {
    viewModelScope.async { block() }
}

val employeeState: StateFlow<EmployeeDataState> by lazyDeferred {
    flow {
        val employees = employeeRepository.fetch()
        emit(EmployeeDataState.Success(employees))
    }
    .catch { emit(EmployeeDataState.Error(it.message)) }
    .stateIn(viewModelScope, SharingStarted.WhileSubscribed(5_000))
}

不过这种方式需要UI层调用await()获取StateFlow，使用起来不如带初始值的方式便捷，一般不推荐在ViewModel中使用。

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