From 0b784474b4a47cefa2eed211262ff33e2f5fd665 Mon Sep 17 00:00:00 2001
From: Iaroslav Postovalov <postovalovya@gmail.com>
Date: Sat, 16 Jan 2021 20:00:00 +0700
Subject: [PATCH] Add draft of multiplatform DeferScope

---
 .../kscience/kmath/memory/DeferScope.kt       |  7 +++++
 .../kscience/kmath/memory/DeferScopeJS.kt     | 30 +++++++++++++++++++
 .../kscience/kmath/memory/DeferScopeJvm.kt    | 30 +++++++++++++++++++
 .../kscience/kmath/memory/DeferScopeNative.kt |  7 +++++
 4 files changed, 74 insertions(+)
 create mode 100644 kmath-memory/src/commonMain/kotlin/kscience/kmath/memory/DeferScope.kt
 create mode 100644 kmath-memory/src/jsMain/kotlin/kscience/kmath/memory/DeferScopeJS.kt
 create mode 100644 kmath-memory/src/jvmMain/kotlin/kscience/kmath/memory/DeferScopeJvm.kt
 create mode 100644 kmath-memory/src/nativeMain/kotlin/kscience/kmath/memory/DeferScopeNative.kt

diff --git a/kmath-memory/src/commonMain/kotlin/kscience/kmath/memory/DeferScope.kt b/kmath-memory/src/commonMain/kotlin/kscience/kmath/memory/DeferScope.kt
new file mode 100644
index 000000000..df5040ee0
--- /dev/null
+++ b/kmath-memory/src/commonMain/kotlin/kscience/kmath/memory/DeferScope.kt
@@ -0,0 +1,7 @@
+package kscience.kmath.memory
+
+public expect class DeferScope {
+    public inline fun defer(crossinline block: () -> Unit)
+}
+
+public expect inline fun <R> withDeferScope(block: DeferScope.() -> R): R
diff --git a/kmath-memory/src/jsMain/kotlin/kscience/kmath/memory/DeferScopeJS.kt b/kmath-memory/src/jsMain/kotlin/kscience/kmath/memory/DeferScopeJS.kt
new file mode 100644
index 000000000..aeb699a39
--- /dev/null
+++ b/kmath-memory/src/jsMain/kotlin/kscience/kmath/memory/DeferScopeJS.kt
@@ -0,0 +1,30 @@
+package kscience.kmath.memory
+
+private typealias Deferred = () -> Unit
+
+public actual class DeferScope {
+    @PublishedApi
+    internal val deferred: MutableList<Deferred> = mutableListOf()
+
+    @PublishedApi
+    internal fun executeAllDeferred() {
+        deferred.forEach(Deferred::invoke)
+        deferred.clear()
+    }
+
+    public actual inline fun defer(crossinline block: () -> Unit) {
+        deferred += {
+            try {
+                block()
+            } catch (ignored: Throwable) {
+            }
+        }
+    }
+}
+
+public actual inline fun <R> withDeferScope(block: DeferScope.() -> R): R {
+    val ds = DeferScope()
+    val r = ds.block()
+    ds.executeAllDeferred()
+    return r
+}
diff --git a/kmath-memory/src/jvmMain/kotlin/kscience/kmath/memory/DeferScopeJvm.kt b/kmath-memory/src/jvmMain/kotlin/kscience/kmath/memory/DeferScopeJvm.kt
new file mode 100644
index 000000000..aeb699a39
--- /dev/null
+++ b/kmath-memory/src/jvmMain/kotlin/kscience/kmath/memory/DeferScopeJvm.kt
@@ -0,0 +1,30 @@
+package kscience.kmath.memory
+
+private typealias Deferred = () -> Unit
+
+public actual class DeferScope {
+    @PublishedApi
+    internal val deferred: MutableList<Deferred> = mutableListOf()
+
+    @PublishedApi
+    internal fun executeAllDeferred() {
+        deferred.forEach(Deferred::invoke)
+        deferred.clear()
+    }
+
+    public actual inline fun defer(crossinline block: () -> Unit) {
+        deferred += {
+            try {
+                block()
+            } catch (ignored: Throwable) {
+            }
+        }
+    }
+}
+
+public actual inline fun <R> withDeferScope(block: DeferScope.() -> R): R {
+    val ds = DeferScope()
+    val r = ds.block()
+    ds.executeAllDeferred()
+    return r
+}
diff --git a/kmath-memory/src/nativeMain/kotlin/kscience/kmath/memory/DeferScopeNative.kt b/kmath-memory/src/nativeMain/kotlin/kscience/kmath/memory/DeferScopeNative.kt
new file mode 100644
index 000000000..6809b18b3
--- /dev/null
+++ b/kmath-memory/src/nativeMain/kotlin/kscience/kmath/memory/DeferScopeNative.kt
@@ -0,0 +1,7 @@
+package kscience.kmath.memory
+
+import kotlinx.cinterop.memScoped
+
+public actual typealias DeferScope = kotlinx.cinterop.DeferScope
+
+public actual inline fun <R> withDeferScope(block: DeferScope.() -> R): R = memScoped(block)