make switch extension for Flow inside Actors

vivid-money · Feb 12, 2024 · afddb28 · afddb28
1 parent fec3f6f
commit afddb28
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 9 deletions.
diff --git a/elmslie-core/src/main/kotlin/money/vivid/elmslie/core/store/Actor.kt b/elmslie-core/src/main/kotlin/money/vivid/elmslie/core/store/Actor.kt
@@ -2,10 +2,18 @@ package money.vivid.elmslie.core.store
 
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.catch
+import kotlinx.coroutines.flow.flow
 import kotlinx.coroutines.flow.mapNotNull
+import kotlinx.coroutines.sync.Mutex
+import kotlinx.coroutines.sync.withLock
+import money.vivid.elmslie.core.switcher.Switcher
+import kotlin.reflect.KClass
 
 abstract class Actor<Command : Any, Event : Any> {
 
+    protected val switchers = mutableMapOf<KClass<out Any>, Switcher>()
+    private val mutex = Mutex()
+
     /**
      * Executes a command. This method is performed on the [Dispatchers.IO]
      * [kotlinx.coroutines.Dispatchers.IO] which is set by ElmslieConfig.ioDispatchers()
@@ -17,4 +25,17 @@ abstract class Actor<Command : Any, Event : Any> {
         errorMapper: (error: Throwable) -> Event? = { null },
     ) = mapNotNull { eventMapper(it) }
         .catch { errorMapper(it)?.let { event -> emit(event) } ?: throw it }
+
+    protected fun <T : Any, Command : Any> Flow<T>.switchOnEach(command: Command, delayMillis: Long = 0): Flow<T> {
+        return flow {
+            val switcher = mutex.withLock {
+                switchers.getOrPut(command::class) {
+                    Switcher()
+                }
+            }
+            switcher.switch(delayMillis) { this@switchOnEach }.collect {
+                emit(it)
+            }
+        }
+    }
 }
diff --git a/...nes-loader/src/main/kotlin/money/vivid/elmslie/samples/coroutines/timer/elm/TimerActor.kt b/...nes-loader/src/main/kotlin/money/vivid/elmslie/samples/coroutines/timer/elm/TimerActor.kt
@@ -1,6 +1,7 @@
 package money.vivid.elmslie.samples.coroutines.timer.elm
 
 import kotlinx.coroutines.delay
+import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.flow
 import money.vivid.elmslie.core.store.Actor
 import money.vivid.elmslie.core.switcher.Switcher
@@ -11,18 +12,20 @@ internal object TimerActor : Actor<Command, Event>() {
 
     override fun execute(command: Command) =
         when (command) {
-            is Command.Start ->
-                switcher
-                    .switch { secondsFlow() }
-                    .mapEvents(
-                        eventMapper = { Event.OnTimeTick },
-                        errorMapper = { Event.OnTimeError(it) },
-                    )
-            is Command.Stop -> switcher.cancel()
+            is Command.Start -> secondsFlow()
+                .switchOnEach(command)
+                .mapEvents(
+                    eventMapper = { Event.OnTimeTick },
+                    errorMapper = { Event.OnTimeError(it) },
+                )
+
+            is Command.Stop -> switchers.getOrPut(Command.Start::class) {
+                Switcher()
+            }.cancel()
         }
 
     @Suppress("MagicNumber")
-    private fun secondsFlow() = flow {
+    private fun secondsFlow(): Flow<Int> = flow {
         repeat(10) {
             delay(1000)
             emit(it)