同步获取 Firebase 用户令牌

Question

我正在尝试获取 Firebase 令牌来验证我对 Rest API 的调用。我可以使用以下代码异步生成令牌。

    FirebaseUser mUser = App.getFirebaseAuth().getCurrentUser();
    if (mUser!=null) {
        mUser.getIdToken(false)
                .addOnCompleteListener(new OnCompleteListener<GetTokenResult>() {
                    public void onComplete(@NonNull Task<GetTokenResult> task) {
                        if (task.isSuccessful()) {
                            ID_TOKEN = task.getResult().getToken();
                        } else {
                            Log.e(App.TAG, "Firebase Token task ended with error.");
                        }
                    }
                });
    } else {
        Log.i(App.TAG,"User is null, no Firebase Token available");
    }

ID_TOKEN 是一个保存结果的静态字符串变量。 问题是，我正在构建我的请求并添加身份验证标头。

        headers.put("Authentication",
                "Bearer  + ID_TOKEN);

问题是，由于 Firebase 令牌是异步检索的，因此 ID_TOKEN 变量有时为空。我尝试使用

强制线程等待任务

Tasks.await(任务)

但是我得到一个异常，说不能在主线程中调用 await。

有没有其他方法可以同步获取token，或者让线程等到任务完成？

Answer 1

我现在就是这样：

private suspend fun getTokenResult (firebaseUser: FirebaseUser) = suspendCoroutine<GetTokenResult?> { continuation ->
firebaseUser.getIdToken(true).addOnCompleteListener {
    if (it.isSuccessful) {
        continuation.resume(it.result)
    } else {
        continuation.resume(null)
    }
}

使用挂起函数和延续机制。所以如果你使用Coroutines，这可能是最简单的方法

Answer 2

我已经做到了，而且效果很好。 这是主类：

object FcmToken {
    @JvmStatic
    fun getToken(): String? {
        val task = FirebaseInstanceId.getInstance().instanceId
        try {
            val result = Tasks.await(task)
            return result.token
        } catch (e: ExecutionException) {
            throw IllegalArgumentException("")
        } catch (e: InterruptedException) {
            throw IllegalArgumentException("")
        }
    }
}

在视图模型中。

viewModelScope.launch(Dispatchers.IO) {
        val result = FcmToken.getToken(viewModelScope)
        Log.e("result", result.toString())
    }

Answer 3

我遇到了同样的问题，当我得到 407 时，我需要在 Retrofit Authenticator 中更新令牌。我使用 CountDownLatch

override fun authenticate(route: Route?, response: Response): Request? {
    val user = FirebaseAuth.getInstance().currentUser
    var token: String? = null
    val lock = CountDownLatch(1)

    user?.getIdToken(true)?.addOnCompleteListener { task ->
        if (task.isSuccessful) {
            token = task.result?.token
            if (token == null) {
                lock.countDown()            //<--unlock 
                return@addOnCompleteListener
            }
            //save token
        } else {
            lock.countDown()                //<--unlock 
            return@addOnCompleteListener
        }
    }
    lock.await()                            //<--wait unlock 

    return if (token != null)
        response.request().newBuilder().header(AUTHORIZATION_KEY, token).build()
    else null
}

Answer 4

如果可能，只需将您的所有后续操作（如 headers.put(...)）移至 onComplete 块中。这将保证执行顺序。

Answer 5

强制在主线程上同步运行是一个非常糟糕的主意。它可能会锁定您的应用程序并可能导致其 ANR。这就是为什么 Android 在您尝试运行 Tasks.await() 时会抱怨的原因 - 它知道可能存在问题。

学习如何进行异步编程以正确方式执行此操作要好得多。您必须在必须提供的侦听器中接收令牌。

Please read this blog 详细了解 Firebase API 为何是异步的。