From 7847585181770cfccb6de88021018b3b548d4efa Mon Sep 17 00:00:00 2001
From: pengjie <17373303529@163.com>
Date: Sat, 18 May 2024 14:22:02 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E4=BD=9C=E4=B8=9A=E8=B4=9F?=
 =?UTF-8?q?=E8=B4=A3=E4=BA=BA=E7=9A=84=E8=AF=81=E4=B9=A6=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../worker/controller/WorkerCertificateController.java   | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/main/java/com/zhgd/xmgl/modules/worker/controller/WorkerCertificateController.java b/src/main/java/com/zhgd/xmgl/modules/worker/controller/WorkerCertificateController.java
index f306c8b46..a3d3daba5 100644
--- a/src/main/java/com/zhgd/xmgl/modules/worker/controller/WorkerCertificateController.java
+++ b/src/main/java/com/zhgd/xmgl/modules/worker/controller/WorkerCertificateController.java
@@ -6,6 +6,8 @@ import com.zhgd.annotation.OperLog;
 import com.zhgd.jeecg.common.api.vo.Result;
 import com.zhgd.jeecg.common.execption.OpenAlertException;
 import com.zhgd.jeecg.common.mybatis.EntityMap;
+import com.zhgd.xmgl.modules.basicdata.entity.SystemUser;
+import com.zhgd.xmgl.modules.basicdata.service.ISystemUserService;
 import com.zhgd.xmgl.modules.worker.entity.WorkerCertificate;
 import com.zhgd.xmgl.modules.worker.service.IWorkerCertificateService;
 import com.zhgd.xmgl.util.DateUtils;
@@ -44,6 +46,9 @@ public class WorkerCertificateController {
     @Autowired
     private IWorkerCertificateService workerCertificateService;
 
+    @Autowired
+    private ISystemUserService systemUserService;
+
     /**
      * 分页列表查询
      *
@@ -163,6 +168,10 @@ public class WorkerCertificateController {
     public Result<List<WorkerCertificate>> queryByType(@RequestBody Map<String, Object> map) {
         Result<List<WorkerCertificate>> result = new Result<List<WorkerCertificate>>();
         String workerId = MapUtils.getString(map, "workerId");
+        if (StringUtils.isBlank(workerId)) {
+            String userId = MapUtils.getString(map, "userId");
+            workerId = systemUserService.getById(userId).getWorkerId().toString();
+        }
         Integer certificateType = MapUtils.getInteger(map, "certificateType");
         List<WorkerCertificate> workerCertificate = workerCertificateService.list(Wrappers.<WorkerCertificate>lambdaQuery()
                 .eq(WorkerCertificate::getWorkerId, workerId)