Use onStart in usecase 1

Author: LukasLechnerDev

app/src/main/java/com/lukaslechner/coroutineusecasesonandroid/usecases/flow/usecase1/FlowUseCase1ViewModel.kt

@@ -4,7 +4,9 @@ import androidx.lifecycle.MutableLiveData
 import androidx.lifecycle.viewModelScope
 import com.lukaslechner.coroutineusecasesonandroid.base.BaseViewModel
 import kotlinx.coroutines.flow.launchIn
+import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.flow.onEach
+import kotlinx.coroutines.flow.onStart
 
 class FlowUseCase1ViewModel(
     stockPriceDataSource: StockPriceDataSource
@@ -15,8 +17,14 @@ class FlowUseCase1ViewModel(
     init {
         stockPriceDataSource
             .latestStockList
-            .onEach { stockList ->
-                currentStockPriceAsLiveData.value = UiState.Success(stockList)
+            .map { stockList ->
+                UiState.Success(stockList) as UiState
+            }
+            .onStart {
+                emit(UiState.Loading)
+            }
+            .onEach { uiState ->
+                currentStockPriceAsLiveData.value = uiState
             }
             .launchIn(viewModelScope)
     }