等待Kotlin协程中的LiveData结果

Wait for LiveData result in a Kotlin coroutine(等待Kotlin协程中的LiveData结果)

问题描述

我有一个存储库类，其异步方法返回User包装成LiveData：

interface Repository {
    fun getUser(): LiveData<User>
}

在ViewModel的Coruotine作用域中，我希望等待getUser()方法的结果并使用User实例。

这就是我要找的：

private fun process() = viewModelScope.launch {
   val user = repository.getUser().await()
   // do something with a user instance
}

我找不到LiveData<>.await()扩展方法，也找不到任何实现它的尝试。 所以在我自己做之前，我想知道是否有更好的方法？

我找到的所有解决方案都是关于使getUser()成为suspend方法，但如果我无法更改Repository怎么办？

推荐答案

您应该能够使用suspendCancellableCoroutine()创建await()扩展函数。这可能并不完全正确，但以下几点应该是可行的：

public suspend fun <T> LiveData<T>.await(): T {
  return withContext(Dispatchers.Main.immediate) {
    suspendCancellableCoroutine { continuation ->
      val observer = object : Observer<T> {
        override fun onChanged(value: T) {
          removeObserver(this)
          continuation.resume(value)
        }
      }

      observeForever(observer)

      continuation.invokeOnCancellation {
        removeObserver(observer)
      }
    }
  }
}

这应返回LiveData发出的第一个值，而不会留下观察者。

这篇关于等待Kotlin协程中的LiveData结果的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！