cancel flow method; rename extension

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

Hide file tree

Showing 2 changed files with 10 additions and 10 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,6 +2,7 @@ package money.vivid.elmslie.core.store
 
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.catch
+import kotlinx.coroutines.flow.emptyFlow
 import kotlinx.coroutines.flow.flow
 import kotlinx.coroutines.flow.mapNotNull
 import kotlinx.coroutines.sync.Mutex
@@ -11,7 +12,7 @@ import kotlin.reflect.KClass
 
 abstract class Actor<Command : Any, Event : Any> {
 
-    protected val switchers = mutableMapOf<KClass<out Any>, Switcher>()
+    private val switchers = mutableMapOf<KClass<out Any>, Switcher>()
     private val mutex = Mutex()
 
     /**
@@ -26,16 +27,20 @@ abstract class Actor<Command : Any, Event : Any> {
     ) = 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> {
+    protected fun <T : Any, Command : Any> Flow<T>.asSwitchFlow(command: Command, delayMillis: Long = 0): Flow<T> {
         return flow {
             val switcher = mutex.withLock {
                 switchers.getOrPut(command::class) {
                     Switcher()
                 }
             }
-            switcher.switch(delayMillis) { this@switchOnEach }.collect {
+            switcher.switch(delayMillis) { this@asSwitchFlow }.collect {
                 emit(it)
             }
         }
     }
+
+    protected fun <T : Any> cancelSwitchFlow(command: KClass<out Any>): Flow<T> {
+        return switchers[command]?.cancel() ?: emptyFlow()
+    }
 }
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
@@ -4,24 +4,19 @@ 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
 
 internal object TimerActor : Actor<Command, Event>() {
 
-    private val switcher = Switcher()
-
     override fun execute(command: Command) =
         when (command) {
             is Command.Start -> secondsFlow()
-                .switchOnEach(command)
+                .asSwitchFlow(command)
                 .mapEvents(
                     eventMapper = { Event.OnTimeTick },
                     errorMapper = { Event.OnTimeError(it) },
                 )
 
-            is Command.Stop -> switchers.getOrPut(Command.Start::class) {
-                Switcher()
-            }.cancel()
+            is Command.Stop -> cancelSwitchFlow(Command.Start::class)
         }
 
     @Suppress("MagicNumber")