/
MobiusEffectRouter.kt
36 lines (31 loc) · 1.13 KB
/
MobiusEffectRouter.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
package kt.mobius.flow
import kotlinx.coroutines.flow.*
import kotlin.reflect.KClass
internal class MobiusEffectRouter<F : Any, E>(
private val effectClasses: Set<KClass<*>>,
private val effectPerformers: List<FlowTransformer<F, E>>,
private val ignoredEffects: List<KClass<*>>
) : FlowTransformer<F, E> {
private val unhandledEffectHandler =
flowTransformer<F, E> { effects: Flow<F> ->
effects
.filter { effect ->
ignoredEffects.none { effectClass ->
effectClass.isInstance(effect)
} && effectClasses.none { effectClass ->
effectClass.isInstance(effect)
}
}
.map { effect -> throw UnknownEffectException(effect) }
}
override fun invoke(input: Flow<F>): Flow<E> {
return (effectPerformers + unhandledEffectHandler)
.map { transform -> transform(input) }
.merge()
}
}
public data class UnknownEffectException(
val effect: Any
) : RuntimeException() {
override val message: String = effect.toString()
}