继续挂起回调函数

Question

我使用 firestore 作为我的后端数据库，保存数据如下所示：

suspend fun uploadDataToFirestore() {
  val firestore = Firebase.firestore
  var batch = firestore.batch
 
 -- fill batch with data --

  batch.commit().addOnCompleteListener {
    if (it.isSuccessful) {
      Timber.d("Successfully saved data")
     
      doAdditionalSuspendingStuff()
            
    } else {
      Timber.d("error at saving data: ${it.exception}")     
    }
}
 

问题在于onCompleteListener，因为我无法调用其他挂起函数。有没有办法从 onCompleteListener 中调用挂起函数，但它们仍然附加到同一范围，因为我不希望函数 uploadDataToFirestore 在执行 doAdditionalSuspendingStuff() 之前完成。

Answer 1

您可以使用 kotlinx-coroutines-play-services artifact，其中包含一小组用于在协程和 Tasks API 之间进行转换的实用程序（在其他 Play 相关库中也很常见）。

您应该能够将基于回调的 API (addOnCompleteListener()) 替换为暂停 Task.await() 扩展函数：

suspend fun uploadDataToFirestore() {
    val firestore = Firebase.firestore
    val batch = firestore.batch

    try {
        batch.commit().await() // Suspend the coroutine while uploading data.

        Timber.d("Successfully saved data")
        doAdditionalSuspendingStuff()
    } catch (exception: Exception) {
        Timber.d("error at saving data: $exception")     
    }
}

await() 还返回一个解包结果（Task<T> 中的T）。

在后台，它使用suspendCancellableCoroutine 将Task<T>.addCompleteListener() 转换为挂起函数。源代码可在here获取。