From 9b3595e45ff94e93b375729dec18e1b8f4eae8e7 Mon Sep 17 00:00:00 2001
From: GG Bond <whatever@tagkong.com>
Date: Tue, 28 Nov 2023 14:25:53 +0800
Subject: [PATCH] Reduce allocation

---
 .../kotlin/com/hippo/ehviewer/download/DownloadManager.kt  | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/app/src/main/kotlin/com/hippo/ehviewer/download/DownloadManager.kt b/app/src/main/kotlin/com/hippo/ehviewer/download/DownloadManager.kt
index 3fe32cef7..fbdcd2ddf 100644
--- a/app/src/main/kotlin/com/hippo/ehviewer/download/DownloadManager.kt
+++ b/app/src/main/kotlin/com/hippo/ehviewer/download/DownloadManager.kt
@@ -248,10 +248,9 @@ object DownloadManager : OnSpiderListener {
 
     @Stable
     @Composable
-    fun collectDownloadState(gid: Long): State<Int> {
-        val flow = remember { stateFlow(gid).map { getDownloadState(it) } }
-        return flow.collectAsState(getDownloadState(gid))
-    }
+    fun collectDownloadState(gid: Long): State<Int> = remember {
+        notifyFlow.transform { if (it.gid == gid) emit(getDownloadState(gid)) }
+    }.collectAsState(getDownloadState(gid))
 
     @Stable
     @Composable