teldufalsari/kmath
2
0
Fork 0
forked from kscience/kmath
Code Pull Requests 1 Activity

safe scopes

Browse Source
This commit is contained in:
Roland Grinis 2021-07-09 08:46:57 +01:00
parent 6384182593
commit 7744880ce7
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
kmath-noa/src/main/kotlin/space/kscience/kmath/noa/memory/NoaScope.kt
View File

@ -9,7 +9,7 @@ private typealias Disposable = () -> Unit
public class NoaScope { public class NoaScope {
private val disposables: ArrayDeque<Disposable> = ArrayDeque(0) internal val disposables: ArrayDeque<Disposable> = ArrayDeque(0)
public fun disposeAll() { public fun disposeAll() {
disposables.forEach(Disposable::invoke) disposables.forEach(Disposable::invoke)
@ -20,15 +20,30 @@ public class NoaScope {
disposables += { disposables += {
try { try {
disposable() disposable()
} catch (ignored: Throwable) { } catch (e: Throwable) {
} }
} }
} }
internal fun addAll(scope: NoaScope) {
disposables.addAll(scope.disposables)
}
} }
internal inline fun <R> withNoaScope(block: NoaScope.() -> R): R { internal fun <R> withNoaScope(block: NoaScope.() -> R): R? {
val noaScope = NoaScope() val noaScope = NoaScope()
val result = noaScope.block() val result = try { noaScope.block() } catch (e: Throwable) { null }
noaScope.disposeAll() noaScope.disposeAll()
return result return result
}
internal fun <R> withNoaScope(scope: NoaScope, block: NoaScope.() -> R): R? {
val noaScope = NoaScope()
val result = try { noaScope.block() } catch (e: Throwable) { null }
if (result == null){
noaScope.disposeAll()
} else {
scope.addAll(noaScope)
}
return result
} }