Java中如何从Repository向ViewModel返回新建文档的DocumentReference？

核心思路 #

Firestore的set()是异步操作，必须等待它执行完成后，才能确保文档已成功写入。虽然database.collection("Groups").document(id)会立即返回DocumentReference，但此时文档可能还未创建完成，导致后续无法获取数据。我们需要在set()操作成功后，再将该DocumentReference传递给ViewModel，同时严格遵循MVVM职责划分。

方案1：使用Firebase Task API #

让Repository返回Task<DocumentReference>，ViewModel监听任务完成状态，成功后将结果存入MutableLiveData。

Repository层代码

fun createGroup(id: String, group: Group): Task<DocumentReference> {
    val docRef = database.collection("Groups").document(id)
    // 执行set操作，完成后返回DocumentReference
    return docRef.set(group).continueWith { task ->
        if (task.isSuccessful) {
            docRef
        } else {
            throw task.exception ?: Exception("创建群组失败")
        }
    }
}

ViewModel层代码

// 定义MutableLiveData用于对外暴露DocumentReference
private val _groupDocRefLiveData = MutableLiveData<DocumentReference>()
val groupDocRefLiveData: LiveData<DocumentReference> = _groupDocRefLiveData

fun createGroup(id: String, group: Group) {
    repository.createGroup(id, group)
        .addOnSuccessListener { docRef ->
            _groupDocRefLiveData.value = docRef
        }
        .addOnFailureListener { exception ->
            // 处理创建失败的逻辑，比如通知UI显示错误
        }
}

方案2：使用Kotlin协程（推荐） #

通过协程将异步操作切换到IO线程，避免阻塞主线程，同时保持代码简洁。

Repository层代码

suspend fun createGroup(id: String, group: Group): DocumentReference {
    val docRef = database.collection("Groups").document(id)
    // 在IO线程执行set操作，await()会挂起直到操作完成
    withContext(Dispatchers.IO) {
        docRef.set(group).await()
    }
    return docRef
}

ViewModel层代码

private val _groupDocRefLiveData = MutableLiveData<DocumentReference>()
val groupDocRefLiveData: LiveData<DocumentReference> = _groupDocRefLiveData

fun createGroup(id: String, group: Group) {
    // 使用viewModelScope确保协程在ViewModel生命周期内执行
    viewModelScope.launch {
        try {
            val docRef = repository.createGroup(id, group)
            _groupDocRefLiveData.value = docRef
        } catch (e: Exception) {
            // 处理创建失败逻辑
        }
    }
}

对之前尝试方案的说明 #

• Repository返回LiveData：确实不符合职责划分，Repository应专注于数据操作，LiveData是UI层的观察工具，不应由Repository暴露。
• ViewModel操作Firestore：违反MVVM分层原则，ViewModel应依赖Repository获取数据，不直接操作数据库。
• Task.await()阻塞主线程：问题出在未切换线程，使用协程的withContext(Dispatchers.IO)或Task的addOnSuccessListener即可避免阻塞。

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