动火作业

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 { } else {
list = workerInfoMapper.selectWorkerInfoList(page, map); list = workerInfoMapper.selectWorkerInfoList(page, map);
} }
List<Long> workerIds = list.stream().map(l -> l.getId()).collect(Collectors.toList()); if (list.size() > 0) {
List<WorkerCertificate> workerCertificate = workerCertificateService.list(Wrappers.<WorkerCertificate>lambdaQuery() List<Long> workerIds = list.stream().map(l -> l.getId()).collect(Collectors.toList());
.in(WorkerCertificate::getWorkerId, workerIds) List<WorkerCertificate> workerCertificate = workerCertificateService.list(Wrappers.<WorkerCertificate>lambdaQuery()
.eq(WorkerCertificate::getCertificateType, 45)); .in(WorkerCertificate::getWorkerId, workerIds)
for (WorkerInfo workerInfo : list) { .eq(WorkerCertificate::getCertificateType, 118));
List<WorkerCertificate> collect = workerCertificate.stream().filter(w -> w.getWorkerId().toString().equals(workerInfo.getId().toString())).collect(Collectors.toList()); for (WorkerInfo workerInfo : list) {
workerInfo.setFireCertificate(collect.size() > 0 ? collect.get(0).getCertificateNumber() : ""); 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); return page.setRecords(list);
} }