[subscription-cancellation-fix] Cancel subscriptions on main thread

dreipol · Oct 3, 2023 · 7f4309f · 7f4309f
1 parent 72a1fe4
commit 7f4309f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/commonMain/kotlin/ch/dreipol/dreimultiplatform/reduxkotlin/MainThreadStore.kt b/src/commonMain/kotlin/ch/dreipol/dreimultiplatform/reduxkotlin/MainThreadStore.kt
@@ -36,7 +36,9 @@ public class MainThreadStore<State, Action>(private val delegate: TypedStore<Sta
     override val store: Store<State> = delegate.store
 
     override val subscribe: (StoreSubscriber) -> StoreSubscription = { subscriber ->
-        runOnMainThread { delegate.subscribe(subscriber) }
+        val subscription = runOnMainThread { delegate.subscribe(subscriber) }
+        val onMainThread = { runOnMainThread { subscription() } }
+        onMainThread
     }
 
     private fun <T> runOnMainThread(block: () -> T): T =