diff --git a/app/src/main/java/com/huanchengfly/tieba/post/arch/Extensions.kt b/app/src/main/java/com/huanchengfly/tieba/post/arch/Extensions.kt index f19fdef1..8a1ab214 100644 --- a/app/src/main/java/com/huanchengfly/tieba/post/arch/Extensions.kt +++ b/app/src/main/java/com/huanchengfly/tieba/post/arch/Extensions.kt @@ -56,6 +56,29 @@ fun Flow.collectPartialAsState( } } +@Composable +inline fun Flow.onEvent( + noinline listener: suspend (Event) -> Unit +) { + val coroutineScope = rememberCoroutineScope() + DisposableEffect(key1 = listener, key2 = this) { + with(coroutineScope) { + val job = launch { + this@onEvent + .filterIsInstance() + .cancellable() + .collect { + launch { + listener(it) + } + } + } + + onDispose { job.cancel() } + } + } +} + @Composable inline fun BaseViewModel<*, *, *, *>.onEvent( noinline listener: suspend (Event) -> Unit