1010import com .finsight .finsight .domain .storage .exception .code .StorageErrorCode ;
1111import com .finsight .finsight .domain .storage .persistence .entity .FolderEntity ;
1212import com .finsight .finsight .domain .storage .persistence .entity .FolderType ;
13+ import com .finsight .finsight .domain .storage .persistence .repository .FolderItemRepository ;
1314import com .finsight .finsight .domain .storage .persistence .repository .FolderRepository ;
1415import com .finsight .finsight .domain .user .persistence .entity .UserEntity ;
1516import com .finsight .finsight .domain .user .persistence .repository .UserRepository ;
@@ -27,13 +28,17 @@ public class FolderService {
2728 private static final int MAX_FOLDER_COUNT = 10 ;
2829
2930 private final FolderRepository folderRepository ;
31+ private final FolderItemRepository folderItemRepository ;
3032 private final UserRepository userRepository ;
3133
3234 // ν΄λ λͺ©λ‘ μ‘°ν
3335 public List <FolderResponse > getFolders (Long userId , FolderType folderType ) {
3436 List <FolderEntity > folders = folderRepository .findByUserUserIdAndFolderTypeOrderBySortOrderAsc (userId , folderType );
3537 return folders .stream ()
36- .map (FolderResponse ::from )
38+ .map (folder -> {
39+ long itemCount = folderItemRepository .countByFolderFolderId (folder .getFolderId ());
40+ return FolderResponse .from (folder , itemCount );
41+ })
3742 .toList ();
3843 }
3944
@@ -88,7 +93,8 @@ public FolderResponse updateFolder(Long userId, Long folderId, UpdateFolderReque
8893 }
8994
9095 folder .updateFolderName (request .folderName ());
91- return FolderResponse .from (folder );
96+ long itemCount = folderItemRepository .countByFolderFolderId (folder .getFolderId ());
97+ return FolderResponse .from (folder , itemCount );
9298 }
9399
94100 // ν΄λ μμ
0 commit comments