|
@@ -65,7 +65,7 @@ public class ProjectRequirementController {
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
Integer companyId = userMapper.selectById(request.getHeader("TOKEN")).getCompanyId();
|
|
Integer companyId = userMapper.selectById(request.getHeader("TOKEN")).getCompanyId();
|
|
int startIndex = (pageIndex-1)*pageSize;
|
|
int startIndex = (pageIndex-1)*pageSize;
|
|
- long total = projectMapper.selectCount(new QueryWrapper<Project>().eq("company_id", companyId));
|
|
|
|
|
|
+
|
|
LocalDate now = LocalDate.now();
|
|
LocalDate now = LocalDate.now();
|
|
now = now.minusDays(7);
|
|
now = now.minusDays(7);
|
|
String startDate = DateTimeFormatter.ofPattern("yyyy-MM-dd").format(now);
|
|
String startDate = DateTimeFormatter.ofPattern("yyyy-MM-dd").format(now);
|
|
@@ -74,8 +74,16 @@ public class ProjectRequirementController {
|
|
List<Project> projectList = projectMapper.selectList(new QueryWrapper<Project>().eq("company_id", companyId));
|
|
List<Project> projectList = projectMapper.selectList(new QueryWrapper<Project>().eq("company_id", companyId));
|
|
List<Integer> collect = projectList.stream().map(Project::getId).collect(Collectors.toList());
|
|
List<Integer> collect = projectList.stream().map(Project::getId).collect(Collectors.toList());
|
|
List<TaskGroup> taskGroups = taskGroupMapper.selectList(new QueryWrapper<TaskGroup>().eq("name", groupName).in("project_id", collect));
|
|
List<TaskGroup> taskGroups = taskGroupMapper.selectList(new QueryWrapper<TaskGroup>().eq("name", groupName).in("project_id", collect));
|
|
- projectIds = taskGroups.stream().map(TaskGroup::getProjectId).collect(Collectors.toList());
|
|
|
|
|
|
+ projectIds = taskGroups.stream().map(TaskGroup::getProjectId).distinct().collect(Collectors.toList());
|
|
|
|
+ }
|
|
|
|
+ long total = 0;
|
|
|
|
+ if (projectIds.size() > 0) {
|
|
|
|
+ total = projectIds.size();
|
|
|
|
+ } else {
|
|
|
|
+ QueryWrapper<Project> queryWrapper = new QueryWrapper<Project>().eq("company_id", companyId);
|
|
|
|
+ total = projectMapper.selectCount(queryWrapper);
|
|
}
|
|
}
|
|
|
|
+
|
|
List<ProjectRequirement> projectRequirements = projectRequirementMapper.customSelect(companyId, startDate, projectId, projectIds.size()==0?null:projectIds, startIndex, pageSize);
|
|
List<ProjectRequirement> projectRequirements = projectRequirementMapper.customSelect(companyId, startDate, projectId, projectIds.size()==0?null:projectIds, startIndex, pageSize);
|
|
HashMap map = new HashMap();
|
|
HashMap map = new HashMap();
|
|
map.put("total", total);
|
|
map.put("total", total);
|