From 6e65afa7e21e7d325dbe515af2dcb45e273c5fd0 Mon Sep 17 00:00:00 2001
From: tarsin <yuanqingxiang233@163.com>
Date: Mon, 3 Jul 2023 20:24:22 +0800
Subject: [PATCH] Share DownloadManager callback flow

---
 .../java/com/hippo/ehviewer/download/DownloadManager.kt   | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/com/hippo/ehviewer/download/DownloadManager.kt b/app/src/main/java/com/hippo/ehviewer/download/DownloadManager.kt
index f7f8efba7..49caf9464 100644
--- a/app/src/main/java/com/hippo/ehviewer/download/DownloadManager.kt
+++ b/app/src/main/java/com/hippo/ehviewer/download/DownloadManager.kt
@@ -36,10 +36,14 @@ import com.hippo.ehviewer.yorozuya.SimpleHandler
 import com.hippo.ehviewer.yorozuya.collect.LongList
 import com.hippo.unifile.UniFile
 import eu.kanade.tachiyomi.util.lang.withUIContext
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.channels.awaitClose
 import kotlinx.coroutines.coroutineScope
+import kotlinx.coroutines.flow.SharingStarted
 import kotlinx.coroutines.flow.callbackFlow
 import kotlinx.coroutines.flow.filter
+import kotlinx.coroutines.flow.shareIn
 import kotlinx.coroutines.launch
 import splitties.init.appCtx
 import splitties.preferences.edit
@@ -257,6 +261,8 @@ object DownloadManager : OnSpiderListener {
         }
     }
 
+    private val callbackFlowScope = CoroutineScope(Dispatchers.IO)
+
     private val _stateFlow = callbackFlow {
         val listener = object : DownloadInfoListener {
             override fun onAdd(info: DownloadInfo, list: List<DownloadInfo>, position: Int) {
@@ -296,7 +302,7 @@ object DownloadManager : OnSpiderListener {
         awaitClose {
             removeDownloadInfoListener(listener)
         }
-    }
+    }.shareIn(callbackFlowScope, SharingStarted.Eagerly)
 
     fun stateFlow(gid: Long) = _stateFlow.filter { it.gid == gid }