等待Kotlin协程中的LiveData结果

本文介绍了等待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发出的第一个值，而不会留下观察者。