refactor(#286): spaceMap을 이용한 코드를 nowSpace를 활용하도록 리팩토링

boostcampwm2023 · Dec 13, 2023 · 928c5cd · 928c5cd
1 parent a5d8226
commit 928c5cd
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/AOS/app/src/main/java/boostcamp/and07/mindsync/ui/main/MainViewModel.kt b/AOS/app/src/main/java/boostcamp/and07/mindsync/ui/main/MainViewModel.kt
@@ -50,14 +50,11 @@ class MainViewModel
         fun getSpaces() {
             viewModelScope.launch(coroutineExceptionHandler) {
                 profileSpaceRepository.getSpaces().collectLatest { responseSpaces ->
-                    val spaceMap: Map<String, Space> =
-                        _uiState.value.spaces.associateBy { space -> space.id }
-                    val newSpaces = mutableListOf<Space>()
-                    responseSpaces.forEach { space ->
-                        if (spaceMap.containsKey(space.id)) {
-                            newSpaces.add(space.copy(isSelected = spaceMap[space.id]!!.isSelected))
+                    val newSpaces = responseSpaces.map { space ->
+                        if (space.id == _uiState.value.nowSpace?.id) {
+                            space.copy(isSelected = true)
                         } else {
-                            newSpaces.add(space)
+                            space
                         }
                     }
                     _uiState.update { uiState ->