Refactor: Home, MyPage, OnBoarding ViewModel에서 유저 정보를 Flow로 구독하도록 변경

Author: wjdrjs00

presentation/src/main/java/com/threegap/bitnagil/presentation/screen/home/HomeViewModel.kt

@@ -11,7 +11,7 @@ import com.threegap.bitnagil.domain.routine.usecase.FetchWeeklyRoutinesUseCase
 import com.threegap.bitnagil.domain.routine.usecase.GetWriteRoutineEventFlowUseCase
 import com.threegap.bitnagil.domain.routine.usecase.RoutineCompletionUseCase
 import com.threegap.bitnagil.domain.routine.usecase.ToggleRoutineUseCase
-import com.threegap.bitnagil.domain.user.usecase.FetchUserProfileUseCase
+import com.threegap.bitnagil.domain.user.usecase.ObserveUserProfileUseCase
 import com.threegap.bitnagil.presentation.screen.home.contract.HomeSideEffect
 import com.threegap.bitnagil.presentation.screen.home.contract.HomeState
 import com.threegap.bitnagil.presentation.screen.home.model.ToggleStrategy
@@ -35,7 +35,7 @@ import javax.inject.Inject
 @HiltViewModel
 class HomeViewModel @Inject constructor(
     private val fetchWeeklyRoutinesUseCase: FetchWeeklyRoutinesUseCase,
-    private val fetchUserProfileUseCase: FetchUserProfileUseCase,
+    private val observeUserProfileUseCase: ObserveUserProfileUseCase,
     private val fetchDailyEmotionUseCase: FetchDailyEmotionUseCase,
     private val routineCompletionUseCase: RoutineCompletionUseCase,
     private val getWriteRoutineEventFlowUseCase: GetWriteRoutineEventFlowUseCase,
@@ -185,7 +185,7 @@ class HomeViewModel @Inject constructor(
     private fun initialize() {
         intent {
             coroutineScope {
-                launch { fetchUserProfile() }
+                launch { observeUserProfile() }
                 launch { fetchDailyEmotion() }
                 launch { fetchWeeklyRoutines(state.currentWeeks) }
                 launch { observeWriteRoutineEvent() }
@@ -246,18 +246,22 @@ class HomeViewModel @Inject constructor(
         }
     }
 
-    private suspend fun fetchUserProfile() {
-        subIntent {
-            reduce { state.copy(loadingCount = state.loadingCount + 1) }
-            fetchUserProfileUseCase().fold(
-                onSuccess = {
-                    reduce { state.copy(userNickname = it.nickname, loadingCount = state.loadingCount - 1) }
-                },
-                onFailure = {
-                    Log.e("HomeViewModel", "유저 정보 가져오기 실패: ${it.message}")
-                    reduce { state.copy(loadingCount = state.loadingCount - 1) }
-                },
-            )
+    private fun observeUserProfile() {
+        intent {
+            repeatOnSubscription {
+                reduce { state.copy(loadingCount = state.loadingCount + 1) }
+                observeUserProfileUseCase().collect { result ->
+                    result.fold(
+                        onSuccess = {
+                            reduce { state.copy(userNickname = it.nickname, loadingCount = state.loadingCount - 1) }
+                        },
+                        onFailure = {
+                            Log.e("HomeViewModel", "유저 정보 가져오기 실패: ${it.message}")
+                            reduce { state.copy(loadingCount = state.loadingCount - 1) }
+                        },
+                    )
+                }
+            }
         }
     }
 

presentation/src/main/java/com/threegap/bitnagil/presentation/screen/mypage/MyPageViewModel.kt

@@ -1,7 +1,7 @@
 package com.threegap.bitnagil.presentation.screen.mypage
 
 import androidx.lifecycle.ViewModel
-import com.threegap.bitnagil.domain.user.usecase.FetchUserProfileUseCase
+import com.threegap.bitnagil.domain.user.usecase.ObserveUserProfileUseCase
 import com.threegap.bitnagil.presentation.screen.mypage.contract.MyPageSideEffect
 import com.threegap.bitnagil.presentation.screen.mypage.contract.MyPageState
 import dagger.hilt.android.lifecycle.HiltViewModel
@@ -12,26 +12,24 @@ import javax.inject.Inject
 
 @HiltViewModel
 class MyPageViewModel @Inject constructor(
-    private val fetchUserProfileUseCase: FetchUserProfileUseCase,
+    private val observeUserProfileUseCase: ObserveUserProfileUseCase,
 ) : ContainerHost<MyPageState, MyPageSideEffect>, ViewModel() {
-    override val container: Container<MyPageState, MyPageSideEffect> = container(initialState = MyPageState.INIT)
 
-    init {
-        loadMyPageInfo()
-    }
-
-    private fun loadMyPageInfo() = intent {
-        fetchUserProfileUseCase().fold(
-            onSuccess = {
-                reduce {
-                    state.copy(
-                        name = it.nickname,
-                        profileUrl = "profileUrl",
-                    )
+    override val container: Container<MyPageState, MyPageSideEffect> =
+        container(
+            initialState = MyPageState.INIT,
+            buildSettings = { repeatOnSubscribedStopTimeout = 5_000L },
+            onCreate = {
+                repeatOnSubscription {
+                    observeUserProfileUseCase().collect { result ->
+                        result.fold(
+                            onSuccess = {
+                                reduce { state.copy(name = it.nickname, profileUrl = "profileUrl") }
+                            },
+                            onFailure = {},
+                        )
+                    }
                 }
             },
-            onFailure = {
-            },
         )
-    }
 }

presentation/src/main/java/com/threegap/bitnagil/presentation/screen/onboarding/OnBoardingViewModel.kt

@@ -8,7 +8,7 @@ import com.threegap.bitnagil.domain.onboarding.usecase.GetOnBoardingsUseCase
 import com.threegap.bitnagil.domain.onboarding.usecase.GetRecommendOnBoardingRoutinesUseCase
 import com.threegap.bitnagil.domain.onboarding.usecase.GetUserOnBoardingUseCase
 import com.threegap.bitnagil.domain.onboarding.usecase.RegisterRecommendOnBoardingRoutinesUseCase
-import com.threegap.bitnagil.domain.user.usecase.FetchUserProfileUseCase
+import com.threegap.bitnagil.domain.user.usecase.ObserveUserProfileUseCase
 import com.threegap.bitnagil.presentation.screen.onboarding.contract.OnBoardingSideEffect
 import com.threegap.bitnagil.presentation.screen.onboarding.contract.OnBoardingState
 import com.threegap.bitnagil.presentation.screen.onboarding.model.OnBoardingItemUiModel
@@ -22,6 +22,7 @@ import dagger.assisted.AssistedInject
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.Job
 import kotlinx.coroutines.async
+import kotlinx.coroutines.flow.first
 import kotlinx.coroutines.isActive
 import org.orbitmvi.orbit.Container
 import org.orbitmvi.orbit.ContainerHost
@@ -34,7 +35,7 @@ class OnBoardingViewModel @AssistedInject constructor(
     private val getRecommendOnBoardingRoutinesUseCase: GetRecommendOnBoardingRoutinesUseCase,
     private val getOnBoardingAbstractUseCase: GetOnBoardingAbstractUseCase,
     private val registerRecommendOnBoardingRoutinesUseCase: RegisterRecommendOnBoardingRoutinesUseCase,
-    private val fetchUserProfileUseCase: FetchUserProfileUseCase,
+    private val observeUserProfileUseCase: ObserveUserProfileUseCase,
     private val getUserOnBoardingUseCase: GetUserOnBoardingUseCase,
     @Assisted private val onBoardingArg: OnBoardingScreenArg,
 ) : ContainerHost<OnBoardingState, OnBoardingSideEffect>, ViewModel() {
@@ -71,7 +72,7 @@ class OnBoardingViewModel @AssistedInject constructor(
     }
 
     private fun loadIntro() = intent {
-        val userName = fetchUserProfileUseCase().getOrNull()?.nickname ?: "-"
+        val userName = observeUserProfileUseCase().first().getOrNull()?.nickname ?: "-"
 
         reduce {
             OnBoardingState.Idle(
@@ -86,7 +87,7 @@ class OnBoardingViewModel @AssistedInject constructor(
     }
 
     private fun loadUserOnBoarding() = intent {
-        val userName = fetchUserProfileUseCase().getOrNull()?.nickname ?: "-"
+        val userName = observeUserProfileUseCase().first().getOrNull()?.nickname ?: "-"
         val userOnBoarding = getUserOnBoardingUseCase().fold(
             onSuccess = { it },
             onFailure = {