From 5ff4c37a095012996348ee2408b8f3f33614a7ff Mon Sep 17 00:00:00 2001
From: HuanCheng65 <22636177+HuanCheng65@users.noreply.github.com>
Date: Sun, 12 Mar 2023 14:11:11 +0800
Subject: [PATCH] =?UTF-8?q?pref:=20=E4=BA=8B=E4=BB=B6=E6=B5=81=E6=89=A9?=
 =?UTF-8?q?=E5=B1=95=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../tieba/post/arch/Extensions.kt             | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)

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 <T : UiState, A> Flow<T>.collectPartialAsState(
     }
 }
 
+@Composable
+inline fun <reified Event : UiEvent> Flow<UiEvent>.onEvent(
+    noinline listener: suspend (Event) -> Unit
+) {
+    val coroutineScope = rememberCoroutineScope()
+    DisposableEffect(key1 = listener, key2 = this) {
+        with(coroutineScope) {
+            val job = launch {
+                this@onEvent
+                    .filterIsInstance<Event>()
+                    .cancellable()
+                    .collect {
+                        launch {
+                            listener(it)
+                        }
+                    }
+            }
+
+            onDispose { job.cancel() }
+        }
+    }
+}
+
 @Composable
 inline fun <reified Event : UiEvent> BaseViewModel<*, *, *, *>.onEvent(
     noinline listener: suspend (Event) -> Unit