问题描述
考虑一个密封的类状态。
sealed class State {
object UnkNown : State()
object Loading : State()
object Success : State()
data class Failure(val exception: Exception)
}
val state:State = MutableStateFlow(State.UnkNown)
现在,我还想有一个简单的挂起方法,它会等待状态达到 Success 或 Failure,这样只需要一次结果的消费者就不需要知道状态流。
如何实现这一目标?
解决方法
虽然您已经想出了一个可行的解决方案,但为了简单起见,您可能希望使用内置的 Flow.first { ... }
运算符。
suspend fun waitForResult(): State {
val resultStates = setOf(State.Success::class,State.Failure::class)
return state.first { it::class in resultStates }
}
,
我想出了以下看起来工作正常的扩展函数。
suspend fun waitForResult(): State {
val resultStates = setOf(State.Success::class,State.Failure::class)
return state.waitForStates(resultStates)
}
suspend fun <T : Any> StateFlow<T>.waitForStates(states: Set<KClass<out T>>): T = coroutineScope {
var currentValue = value
// not needed for correctness,just an optimisation
if (currentValue::class in states) {
return currentValue
}
coroutineScope {
collect {
if (it::class in states) {
currentValue = it
cancel()
}
}
}
return currentValue
}