fix crashes

DatL4g · Dec 17, 2023 · bf6fba7 · bf6fba7
1 parent c26e01d
commit bf6fba7
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/model/src/commonMain/kotlin/dev/datlag/burningseries/model/common/ExtendCoroutine.kt b/model/src/commonMain/kotlin/dev/datlag/burningseries/model/common/ExtendCoroutine.kt
@@ -3,10 +3,7 @@ package dev.datlag.burningseries.model.common
 import kotlinx.coroutines.CancellationException
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.coroutineScope
-import kotlinx.coroutines.flow.Flow
-import kotlinx.coroutines.flow.FlowCollector
-import kotlinx.coroutines.flow.StateFlow
-import kotlinx.coroutines.flow.first
+import kotlinx.coroutines.flow.*
 
 fun <T> scopeCatching(block: () -> T): Result<T> = try {
     Result.success(block())
@@ -55,7 +52,9 @@ inline fun <reified T> Any?.safeCast(): T? {
 suspend fun <T> Flow<T>.collectSafe(collector: FlowCollector<T>) {
     suspendCatching {
         this@collectSafe.collect(collector)
-    }.getOrNull() ?: collector.emit(this.first())
+    }.getOrNull() ?: suspendCatching {
+        this@collectSafe.firstOrNull()
+    }.getOrNull()?.let { collector.emit(it) }
 }
 
 suspend fun <T> StateFlow<T>.collectSafe(collector: FlowCollector<T>) {