MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

[fleet, resources] support cancellation in shared resource

Browse Source 
GitOrigin-RevId: eced67e045e983ecf064cf9b2384217f5140dec1
This commit is contained in:
Alexander Zolotov
2025-04-12 13:29:29 +00:00
committed by intellij-monorepo-bot
parent e87dde430e
commit 59d9e4e4af
1 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fleet/util/core/srcCommonMain/fleet/util/async/Resource.kt
+24 -7
View File
@@ -223,25 +223,30 @@ fun <T> Resource<T>.useOn(coroutineScope: CoroutineScope): Deferred<T> {
return deferred
}
enum class StopMode {
NOPE, STOP, CANCEL
}
sealed class SharingMode(
val runImmediately: Boolean,
val stopWithoutConsumers: Boolean,
val stopWithoutConsumersMode: StopMode,
) {
/**
* The resource starts immediately and remains active until the scope is canceled.
*/
data object Eager : SharingMode(runImmediately = true, stopWithoutConsumers = false)
data object Eager : SharingMode(runImmediately = true, stopWithoutConsumersMode = StopMode.NOPE)
/**
* The resource starts when a consumer appears and remains active until the scope is canceled.
*/
data object Lazy : SharingMode(runImmediately = false, stopWithoutConsumers = false)
data object Lazy : SharingMode(runImmediately = false, stopWithoutConsumersMode = StopMode.NOPE)
/**
* The resource starts when a consumer appears and is gracefully shut down when the last consumer leaves.
* This cycle may repeat multiple times.
*/
data object WhileUsed : SharingMode(runImmediately = false, stopWithoutConsumers = true)
data class WhileUsed(val graceful: Boolean = true) : SharingMode(runImmediately = false,
stopWithoutConsumersMode = if (graceful) StopMode.STOP else StopMode.CANCEL)
}
/**
@@ -291,9 +296,21 @@ fun <T> Resource<T>.shareIn(coroutineScope: CoroutineScope, sharing: SharingMode
when (val s = state) {
is SharedResourceState.Stopping<*>, is SharedResourceState.NotRunning<*> -> error("we are not yet done with the resource, yet it is not running")
is SharedResourceState.Running<T> -> {
state = if (s.refCount == 1 && sharing.stopWithoutConsumers) {
s.runnning.termination.complete()
SharedResourceState.Stopping(s.runnning.job)
state = if (s.refCount == 1) {
when (sharing.stopWithoutConsumersMode) {
StopMode.STOP -> {
s.runnning.termination.complete()
SharedResourceState.Stopping(s.runnning.job)
}
StopMode.CANCEL -> {
s.runnning.job.cancel()
s.runnning.termination.complete()
SharedResourceState.Stopping(s.runnning.job)
}
StopMode.NOPE -> {
s.copy(refCount = 0)
}
}
}
else {
s.copy(refCount = s.refCount - 1)