如何在ViewModel中用LiveData处理Firebase登录用户的数据库查询

你遇到的重复插入问题，核心原因是直接读取LiveData的value属性时，Room的异步查询还未完成。LiveData初始值为null，你在查询结果返回前就做了判断，导致每次都误判为用户不存在，进而执行插入逻辑。

以下两种方案均符合你要求的Room+LiveData+ViewModel技术栈，推荐第一种更简洁的实现方式。

方案一：仓库提供挂起函数实现同步查询 #

既然需求是一次性判断用户是否存在（而非持续监听用户数据变化），直接让仓库返回挂起函数，利用协程等待数据库查询完成，是最直观的方案。

1. 修改仓库接口与Room DAO #

// UsersRepository
interface UsersRepository {
    // 保留原LiveData方法供UI层监听用户变化（如果需要）
    fun getUser(id: String): LiveData<User?>
    // 新增挂起函数，用于一次性查询
    suspend fun getUserSync(id: String): User?
}

// Room DAO层对应方法
@Query("SELECT * FROM user WHERE id = :id")
suspend fun getUser(id: String): User?

2. 调整ViewModel逻辑 #

在协程中调用挂起函数，等待查询结果返回后再判断：

class UserViewModel(
    private val usersRepository: UsersRepository
) : ViewModel() {

    // 暴露不可变的LiveData给UI层
    val currentUser: LiveData<User> get() = _currentUser
    private val _currentUser = MutableLiveData<User>()

    fun setFirebaseUser(fireBaseUser: FirebaseUser) = viewModelScope.launch {
        with(fireBaseUser) {
            val existingUser = usersRepository.getUserSync(uid)
            if (existingUser != null) {
                Log.i("UserViewModel", "User Found")
                _currentUser.value = existingUser
            } else {
                Log.i("UserViewModel", "Insert new user")
                val newUser = User(uid, email, displayName)
                usersRepository.insertUser(newUser)
                _currentUser.value = newUser
            }
        }
    }
}

3. Fragment中可选优化（确保数据准备完成后导航） #

如果HomeFragment依赖当前用户数据，可以等待ViewModel处理完成后再导航：

private fun onSignInResult(result: FirebaseAuthUIAuthenticationResult) {
    if (result.resultCode == AppCompatActivity.RESULT_OK) {
        FirebaseAuth.getInstance().currentUser?.let { user ->
            viewModel.setFirebaseUser(user)
            viewModel.currentUser.observe(viewLifecycleOwner) {
                findNavController().navigate(SignInFragmentDirections.navSignInToHome())
                // 移除观察者避免重复触发导航
                viewModel.currentUser.removeObservers(viewLifecycleOwner)
            }
        }
    }
}

方案二：ViewModel中正确观察LiveData（不推荐） #

如果坚持用LiveData查询，需要在ViewModel中通过observeForever监听结果，同时必须手动移除观察者避免内存泄漏：

fun setFirebaseUser(fireBaseUser: FirebaseUser) {
    with(fireBaseUser) {
        val repoUserLiveData = usersRepository.getUser(uid)
        val observer = Observer<User?> { user ->
            if (user != null) {
                Log.i("UserViewModel", "User Found")
                _currentUser.value = user
            } else {
                viewModelScope.launch {
                    val newUser = User(uid, email, displayName)
                    usersRepository.insertUser(newUser)
                    _currentUser.value = newUser
                }
            }
            // 拿到结果后立即移除观察者
            repoUserLiveData.removeObserver(observer)
        }
        repoUserLiveData.observeForever(observer)
    }
}

这种方式需要额外管理观察者的生命周期，代码冗余度更高，仅适合需要持续监听用户数据变化的场景。

优先选择方案一：

• 符合协程异步处理的设计思想，代码简洁直观
• 避免了LiveData在非UI层的不当使用（LiveData本质是为UI层生命周期感知设计的）
• 完美解决重复插入的问题，确保查询结果真实可靠

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