diff --git a/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerSafeEducationWorkerServiceImpl.java b/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerSafeEducationWorkerServiceImpl.java index 677a04de7..5e9b0ebc3 100644 --- a/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerSafeEducationWorkerServiceImpl.java +++ b/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerSafeEducationWorkerServiceImpl.java @@ -139,11 +139,14 @@ public class WorkerSafeEducationWorkerServiceImpl extends ServiceImpl eduIds = workerSafeEducationService.list(new LambdaQueryWrapper() .eq(WorkerSafeEducation::getVideoType, 1)).stream().map(WorkerSafeEducation::getId).collect(Collectors.toList()); + List workIds = workerInfoService.list(new LambdaQueryWrapper() + .eq(WorkerInfo::getProjectSn, projectSn)).stream().map(WorkerInfo::getId).collect(Collectors.toList()); List list = new ArrayList<>(); if (CollUtil.isNotEmpty(eduIds)) { list = this.list(Wrappers.lambdaQuery() .eq(WorkerSafeEducationWorker::getProjectSn, projectSn) .in(WorkerSafeEducationWorker::getEduId, eduIds) + .in(CollUtil.isNotEmpty(workIds), WorkerSafeEducationWorker::getWorkerId, workIds) .orderByDesc(WorkerSafeEducationWorker::getId)); } resultMap.put("educationNum", list.size()); @@ -160,9 +163,12 @@ public class WorkerSafeEducationWorkerServiceImpl extends ServiceImpl educationWorkers = new ArrayList<>(); + List workIds = workerInfoService.list(new LambdaQueryWrapper() + .eq(WorkerInfo::getProjectSn, projectSn)).stream().map(WorkerInfo::getId).collect(Collectors.toList()); if (list.size() > 0) { educationWorkers = this.list(Wrappers.lambdaQuery() .in(WorkerSafeEducationWorker::getEduId, list) + .in(CollUtil.isNotEmpty(workIds), WorkerSafeEducationWorker::getWorkerId, workIds) .ge(WorkerSafeEducationWorker::getStudyTime, DateUtil.offsetDay(new Date(), -7))); } List> resultList = new ArrayList<>(); @@ -185,8 +191,12 @@ public class WorkerSafeEducationWorkerServiceImpl extends ServiceImpl workIds = workerInfoService.list(new LambdaQueryWrapper() + .eq(WorkerInfo::getProjectSn, projectSn)).stream().map(WorkerInfo::getId).collect(Collectors.toList()); List workerList = this.list(Wrappers.lambdaQuery() - .eq(WorkerSafeEducationWorker::getProjectSn, projectSn)); + .eq(WorkerSafeEducationWorker::getProjectSn, projectSn) + .in(CollUtil.isNotEmpty(workIds), WorkerSafeEducationWorker::getWorkerId, workIds) + ); List> resultList = new ArrayList<>(); for (int i = 0; i < type.length; i++) { int j = i + 1; @@ -210,15 +220,17 @@ public class WorkerSafeEducationWorkerServiceImpl extends ServiceImpl teamInfoList = teamInfoService.list(Wrappers.lambdaQuery().eq(TeamInfo::getProjectSn, projectSn)); int count = workerInfoService.count(Wrappers.lambdaQuery() .eq(WorkerInfo::getProjectSn, projectSn) - .eq(WorkerInfo::getPersonType, 1) .eq(WorkerInfo::getInserviceType, 1)); List eduIds = workerSafeEducationService.list(Wrappers.lambdaQuery() .eq(WorkerSafeEducation::getVideoType, 1) .eq(WorkerSafeEducation::getProjectSn, projectSn)).stream().map(WorkerSafeEducation::getId).collect(Collectors.toList()); + List workIds = workerInfoService.list(new LambdaQueryWrapper() + .eq(WorkerInfo::getProjectSn, projectSn)).stream().map(WorkerInfo::getId).collect(Collectors.toList()); List workerEduList = new ArrayList<>(); if (CollUtil.isNotEmpty(eduIds)) { workerEduList = this.list(Wrappers.lambdaQuery() .in(WorkerSafeEducationWorker::getEduId, eduIds) + .in(CollUtil.isNotEmpty(workIds), WorkerSafeEducationWorker::getWorkerId, workIds) .eq(WorkerSafeEducationWorker::getProjectSn, projectSn) .ge(WorkerSafeEducationWorker::getStudyTime, DateUtil.offsetMonth(new Date(), -1))); } @@ -230,7 +242,7 @@ public class WorkerSafeEducationWorkerServiceImpl extends ServiceImpl teamMap = new HashMap<>(); teamMap.put("name", teamInfo.getTeamName()); - teamMap.put("num", workerFilter.stream().filter(w -> w.getTeamId().toString().equals(teamInfo.getId().toString())).count()); + teamMap.put("num", workerFilter.stream().filter(w -> Objects.equals(w.getTeamId(), teamInfo.getId())).count()); resultList.add(teamMap); } resultMap.put("teamStat", resultList);