CLAP-449 팀 작업 현황 카테고리 필터링시 작업 개수 오류 수정

CLAP-449 팀 작업 현황 카테고리 필터링시 작업 개수 오류 수정

Author: hyoseong-Choi

src/main/java/clap/server/application/mapper/response/TeamTaskResponseMapper.java

@@ -3,13 +3,17 @@
 import clap.server.adapter.inbound.web.dto.task.response.TeamStatusResponse;
 import clap.server.adapter.inbound.web.dto.task.response.TeamTaskItemResponse;
 import clap.server.adapter.inbound.web.dto.task.response.TeamTaskResponse;
+import clap.server.adapter.outbound.persistense.entity.task.constant.TaskStatus;
 import clap.server.domain.model.task.Task;
 
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
 
+import static clap.server.adapter.outbound.persistense.entity.task.constant.TaskStatus.IN_PROGRESS;
+import static clap.server.adapter.outbound.persistense.entity.task.constant.TaskStatus.IN_REVIEWING;
+
 public class TeamTaskResponseMapper {
 
     public static List<TeamTaskResponse> toTeamTaskResponses(List<Task> tasks) {
@@ -26,16 +30,16 @@ private static TeamTaskResponse toTeamTaskResponse(Map.Entry<Long, List<Task>> e
                 .collect(Collectors.toList());
 
         Task firstTask = entry.getValue().get(0);
-        int inProgressTaskCount = firstTask.getProcessor().getInProgressTaskCount();
-        int inReviewingTaskCount = firstTask.getProcessor().getInReviewingTaskCount();
+        LinkedHashMap<TaskStatus, List<TeamTaskItemResponse>> collect = taskResponses.stream()
+                .collect(Collectors.groupingBy(TeamTaskItemResponse::taskStatus, LinkedHashMap::new, Collectors.toList()));
 
         return new TeamTaskResponse(
                 entry.getKey(),
                 firstTask.getProcessor().getNickname(),
                 firstTask.getProcessor().getImageUrl(),
                 firstTask.getProcessor().getDepartment().getName(),
-                inProgressTaskCount,
-                inReviewingTaskCount,
+                collect.get(IN_PROGRESS) != null ? collect.get(IN_PROGRESS).size() : 0,
+                collect.get(IN_REVIEWING) != null ? collect.get(IN_REVIEWING).size() : 0,
                 entry.getValue().size(),
                 taskResponses
         );