github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-03-25 15:00:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

[core] CoroutineOnDemandValueScope: fix emit when state is ProducerReady

Browse Source
This commit is contained in:
Him188 2023-04-16 10:28:09 +01:00
parent 1b122190f1
commit a1d8d744d0
No known key found for this signature in database
GPG Key ID: BA439CDDCF652375
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mirai-core/src/commonMain/kotlin/network/auth/CoroutineOnDemandValueScope.kt
View File

@ -73,6 +73,10 @@ internal class CoroutineOnDemandValueScope<T, V>(
}
}
is ProducerState.ProducerReady -> {
setStateProducing(state)
}
else -> throw IllegalProducerStateException(state)
}
}
@ -96,6 +100,14 @@ internal class CoroutineOnDemandValueScope<T, V>(
}
}
private fun setStateProducing(state: ProducerState.ProducerReady<T, V>) {
val deferred = CompletableDeferred<V>(coroutineScope.coroutineContext.job)
if (!compareAndSetState(state, ProducerState.Producing(state.producer, deferred))) {
deferred.cancel() // avoid leak
}
// loop again
}
private fun finishImpl(exception: Throwable?) {
state.loop { state ->
when (state) {
@ -173,11 +185,7 @@ internal class CoroutineOnDemandValueScope<T, V>(
}
is ProducerState.ProducerReady -> {
val deferred = CompletableDeferred<V>(coroutineScope.coroutineContext.job)
if (!compareAndSetState(state, ProducerState.Producing(state.producer, deferred))) {
deferred.cancel() // avoid leak
}
// loop again
setStateProducing(state)
}
is ProducerState.Producing -> return true // ok
@ -201,4 +209,4 @@ internal class CoroutineOnDemandValueScope<T, V>(
override fun finish() {
finishImpl(null)
}
}
}