如何在Text Composable加载前从Room数据库取值避免显示Null

核心问题 #

RoomViewModel中的email是普通变量，flowOf(email)只会发射初始的空字符串，后续email更新时不会触发Flow重新获取数据，导致energyState一直停留在初始值null。此外，Room的查询Flow在无匹配数据时不会发射任何值，也会导致UI收不到更新。

步骤1：重构RoomViewModel的邮箱状态 #

将普通email变量改为MutableStateFlow，让邮箱变化能被Flow感知：

class RoomViewModel(private val repository: Repository) : ViewModel() {
    // 用MutableStateFlow存储邮箱，外部通过setEmail更新
    private val _email = MutableStateFlow("")
    val email: StateFlow<String> = _email.asStateFlow()

    // 基于邮箱的StateFlow构建energyState，邮箱变化自动重新查询
    val energyState: StateFlow<Int?> = _email
        .flatMapLatest { currentEmail ->
            if (currentEmail.isEmpty()) {
                flowOf(null)
            } else {
                repository.getEnergy(currentEmail)
                    .onStart { emit(null) } // 无数据时先发射null
            }
        }
        .stateIn(
            scope = viewModelScope,
            started = SharingStarted.WhileSubscribed(5_000),
            initialValue = null,
        )

    // 外部调用此方法更新邮箱
    fun setEmail(newEmail: String) {
        _email.value = newEmail
    }

    fun upsertInfo(info: Info) {
        viewModelScope.launch {
            repository.upsertInfo(info)
        }
    }

    fun updateEnergy(energy: Int, email: String) {
        viewModelScope.launch {
            repository.updateEnergy(energy, email)
            // Room会自动监听数据库变化，无需手动更新energyState
        }
    }
}

步骤2：调整Dao与Repository的返回类型 #

修改Dao的查询方法，让无匹配数据时返回null：

@Query("SELECT energy FROM info WHERE email = :email LIMIT 1")
fun getEnergy(email: String): Flow<Int?>

同步更新Repository的返回类型：

fun getEnergy(email: String): Flow<Int?> {
    return db.dao.getEnergy(email)
}

步骤3：修正外部设置邮箱的方式 #

将直接赋值roomViewModel.email改为调用setEmail方法：

认证成功时的赋值：

LaunchedEffect(authState.value) {
    Log.i("tag", "authState in signInwithEmailScreen = ${authState}")
    when (authState.value) {
        is AuthState.Authenticated -> {
            navController.navigate(route = "Intro_Screen")
            roomViewModel.setEmail(signInViewModel.email.value ?: "")
        }
        // 其他状态逻辑...
    }
}

用户输入邮箱后的赋值：

Button(onClick = {
    roomViewModel.setEmail(email) // email为用户输入的字符串
}) {
    Text(text = "确认邮箱")
}

步骤4：确保数据库存在对应记录 #

如果是新用户，需先调用upsertInfo插入包含邮箱和初始energy的记录，否则查询会返回null。例如在注册成功后：

roomViewModel.upsertInfo(Info(email = userEmail, energy = 100))

步骤5：UI中正确收集状态 #

在Composable中收集energyState并处理null值：

val energy by roomViewModel.energyState.collectAsState()

Text(text = "当前能量：${energy ?: 0}") // 用0替代null，或根据需求自定义显示

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