挂起直到 StateFlow 达到所需状态之一并返回结果

如何解决挂起直到 StateFlow 达到所需状态之一并返回结果

考虑一个密封的类状态。

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
}