动火作业

This commit is contained in:
pengjie 2024-05-17 09:04:10 +08:00
parent e9d1fd8faa
commit 185e944070

View File

@ -196,13 +196,15 @@ public class WorkerInfoServiceImpl extends ServiceImpl<WorkerInfoMapper, WorkerI
} else {
list = workerInfoMapper.selectWorkerInfoList(page, map);
}
List<Long> workerIds = list.stream().map(l -> l.getId()).collect(Collectors.toList());
List<WorkerCertificate> workerCertificate = workerCertificateService.list(Wrappers.<WorkerCertificate>lambdaQuery()
.in(WorkerCertificate::getWorkerId, workerIds)
.eq(WorkerCertificate::getCertificateType, 45));
for (WorkerInfo workerInfo : list) {
List<WorkerCertificate> collect = workerCertificate.stream().filter(w -> w.getWorkerId().toString().equals(workerInfo.getId().toString())).collect(Collectors.toList());
workerInfo.setFireCertificate(collect.size() > 0 ? collect.get(0).getCertificateNumber() : "");
if (list.size() > 0) {
List<Long> workerIds = list.stream().map(l -> l.getId()).collect(Collectors.toList());
List<WorkerCertificate> workerCertificate = workerCertificateService.list(Wrappers.<WorkerCertificate>lambdaQuery()
.in(WorkerCertificate::getWorkerId, workerIds)
.eq(WorkerCertificate::getCertificateType, 118));
for (WorkerInfo workerInfo : list) {
List<WorkerCertificate> collect = workerCertificate.stream().filter(w -> w.getWorkerId().toString().equals(workerInfo.getId().toString())).collect(Collectors.toList());
workerInfo.setFireCertificate(collect.size() > 0 ? collect.get(0).getCertificateNumber() : "");
}
}
return page.setRecords(list);
}