Question

我正在构建一个将Firebase用于两件事的客户端应用程序：

用户身份验证
使用实时数据库

我已经设法在客户端和后端服务器上正确设置了所有内容（使用Firebase的Admin SDK），并且能够正确验证用户身份并允许他们读取/写入数据库。

我还使用Retrofit2将请求从客户端发送到后端。

作为允许用户访问数据库的一部分，需要将用户的令牌发送到后端，以便可以验证用户。

为此，我有以下逻辑：

val user = FirebaseAuth.getInstance().currentUser

    if (user != null) {
        user.getIdToken(false).addOnCompleteListener {
            if (it.isSuccessful) {
                val token = it.result?.token
                //retrofit logic to send request happens from here
            }
       }

如您所见，获取用户的ID令牌是一个异步调用，在我拥有的当前代码库中，对于后端（复制）的每个调用，我都有此代码块。

我想知道如何将该代码段导出到一个函数（也许是一个暂停方法？），以便可以在每次调用后端时重用它

我在网上进行搜索，看到了很多SO问题，但没有一个适合这种情况。我曾考虑过传递回调，但是我有几种与后端通信的方法，每个方法都需要一个不同的回调方法。

我正在寻找的解决方案看起来像这样：

fun fetchDataFromDB() {
  getIdTokenForUser()
  //wait till it finishes and then
  //perform request to DB
}

fun updateDataInDB() {
  getIdTokenForUser()
  //wait till it finishes and then
  //perform request to DB
}

//......

我曾尝试阅读和实现协程，但我缺乏正确执行协程的知识。

编辑

感谢@Doug Stevenson的回答和指导，我设法构造了以下内容：

private suspend fun getUserIdToken(user: FirebaseUser) = coroutineScope {

    val job = async {
        user.getIdToken(false).result?.token
    }
    job.await()
}

我以这种方式使用它：

fun updateDB(context: Context) = runBlocking {

    val user = FirebaseAuth.getInstance().currentUser

    if (user != null) {
        val token = getUserIdToken(user)
    }

  }

这是正确的方法吗？由于以下给出的答案提出了不同的实现方式。

Answer 1

getIdToken是异步返回Task对象的。如果要在Kotlin协程中使用Task对象，可以使用库kotlinx-coroutines-play-services向任务添加一个扩展方法await()，以使其在协程中可用。这样，您可以编写如下内容：

implementation "org.jetbrains.kotlinx:kotlinx-coroutines-play-services:1.3.9"

import kotlinx.coroutines.tasks.await

suspend fun getIdTokenForUser(user: FirebaseUser): GetTokenResult {
    return try {
        user.getIdToken(false).await()
    }
    catch (e: Exception) {
        // handle error
    }
}

您可能需要在这里更新类型-我没有尝试编译或测试它。

另请参阅：

Answer 2

为了使用基于回调的API，如下所示：


val myCallback = object : ServiceCallback() {
    override fun onResult(theobject: Something) {
        // your callback code here
    }
    override fun onFailure(ex: Throwable) {
         // error handling
    }

}

theService.enqueue(callback)

您可以使用suspendCoroutine

它的作用是中止执行，直到回调满足连续性为止。因此，您可以编写如下的KTX：

suspend fun Service.getSomething(): Something = suspendCoroutine{ cont ->
    val callback = object : ServiceCallback(){
         override fun onSuccess(data: Something): Unit = cont.resume(data)
         override fun onFailure(ex: Throwable): Unit = cont.resume(ex)
    }
    this.enqueue(callback)
}

如何在OnCompleteListener Firebase中使用异步/等待/协程

2 个答案: