[#159] Wrap flow with flowTransform and use the .first() to get only …

…first emit from datastore
nimblehq · Jan 12, 2023 · 41c98c3 · 41c98c3
1 parent 2b5b0f6
commit 41c98c3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/sample-compose/app/src/main/java/co/nimblehq/sample/compose/ui/screens/home/HomeViewModel.kt b/sample-compose/app/src/main/java/co/nimblehq/sample/compose/ui/screens/home/HomeViewModel.kt
@@ -41,7 +41,6 @@ class HomeViewModel @Inject constructor(
 
     init {
         execute {
-            showLoading()
             val getModelsFlow = getModelsUseCase()
             val getFirstTimeLaunchPreferencesFlow = getFirstTimeLaunchPreferencesUseCase()
 
@@ -52,12 +51,13 @@ class HomeViewModel @Inject constructor(
                 if (firstTimeLaunch) {
                     updateFirstTimeLaunchPreferencesUseCase(false)
                 }
+            }.onStart {
+                showLoading()
+            }.onCompletion {
+                hideLoading()
             }.catch {
                 _error.emit(it)
-                hideLoading()
-            }.collect {
-                hideLoading()
-            }
+            }.collect()
         }
     }
 

diff --git a/.../src/main/java/co/nimblehq/sample/compose/data/repository/AppPreferencesRepositoryImpl.kt b/.../src/main/java/co/nimblehq/sample/compose/data/repository/AppPreferencesRepositoryImpl.kt
@@ -3,6 +3,7 @@ package co.nimblehq.sample.compose.data.repository
 import android.util.Log
 import androidx.datastore.core.DataStore
 import androidx.datastore.preferences.core.*
+import co.nimblehq.sample.compose.data.extensions.flowTransform
 import co.nimblehq.sample.compose.domain.repository.AppPreferencesRepository
 import kotlinx.coroutines.flow.*
 import java.io.IOException
@@ -16,8 +17,9 @@ class AppPreferencesRepositoryImpl @Inject constructor(
         val FIRST_TIME_LAUNCH = booleanPreferencesKey("FIRST_TIME_LAUNCH")
     }
 
-    override fun getFirstTimeLaunchPreferences(): Flow<Boolean> = appPreferencesDataStore.data
-        .catch { exception ->
+    override fun getFirstTimeLaunchPreferences(): Flow<Boolean> = flowTransform {
+        appPreferencesDataStore.data
+            .catch { exception ->
             if (exception is IOException) {
                 Log.e(
                     AppPreferencesRepositoryImpl::class.simpleName,
@@ -30,7 +32,8 @@ class AppPreferencesRepositoryImpl @Inject constructor(
             }
         }.map { preferences ->
             preferences[PreferencesKeys.FIRST_TIME_LAUNCH] ?: true
-        }
+        }.first()
+    }
 
     override suspend fun updateFirstTimeLaunchPreferences(isFirstTimeLaunch: Boolean) {
         appPreferencesDataStore.edit { preferences ->