From 185e9440708acd79a155dfc16b4ced584bd0cfb6 Mon Sep 17 00:00:00 2001
From: pengjie <17373303529@163.com>
Date: Fri, 17 May 2024 09:04:10 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8A=A8=E7=81=AB=E4=BD=9C=E4=B8=9A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/WorkerInfoServiceImpl.java      | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerInfoServiceImpl.java b/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerInfoServiceImpl.java
index ff1151411..5b0026d98 100644
--- a/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerInfoServiceImpl.java
+++ b/src/main/java/com/zhgd/xmgl/modules/worker/service/impl/WorkerInfoServiceImpl.java
@@ -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);
     }