From c35bd84886fb0c4e4ee0ffeeb3819a8a8df532b5 Mon Sep 17 00:00:00 2001
From: tuqin <tuqin@tuqin.cn>
Date: 星期一, 02 十一月 2026 13:13:36 +0800
Subject: [PATCH] feat(safety): 安全生产六期 - 风险可视化看板

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/dv/machinery/MesDvMachineryController.java |    8 +++++++-
 1 files changed, 7 insertions(+), 1 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/dv/machinery/MesDvMachineryController.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/dv/machinery/MesDvMachineryController.java
index 932db26..864bf15 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/dv/machinery/MesDvMachineryController.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/dv/machinery/MesDvMachineryController.java
@@ -52,6 +52,8 @@
     private MesDvMachineryTypeService machineryTypeService;
     @Resource
     private MesMdWorkshopService workshopService;
+    @Resource
+    private cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService workstationService;
 
     @PostMapping("/create")
     @Operation(summary = "鍒涘缓璁惧")
@@ -151,17 +153,21 @@
         if (CollUtil.isEmpty(list)) {
             return Collections.emptyList();
         }
-        // 1. 鎵归噺鑾峰彇璁惧绫诲瀷鍜岃溅闂翠俊鎭�
+        // 1. 鎵归噺鑾峰彇璁惧绫诲瀷銆佽溅闂村拰宸ヤ綔绔欎俊鎭�
         Map<Long, MesDvMachineryTypeDO> machineryTypeMap = machineryTypeService.getMachineryTypeMap(
                 convertSet(list, MesDvMachineryDO::getMachineryTypeId));
         Map<Long, MesMdWorkshopDO> workshopMap = workshopService.getWorkshopMap(
                 convertSet(list, MesDvMachineryDO::getWorkshopId));
+        Map<Long, cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO> workstationMap =
+                workstationService.getWorkstationMap(convertSet(list, MesDvMachineryDO::getWorkstationId));
         // 2. 鎷兼帴 VO
         return BeanUtils.toBean(list, MesDvMachineryRespVO.class, vo -> {
             MapUtils.findAndThen(machineryTypeMap, vo.getMachineryTypeId(),
                     machineryType -> vo.setMachineryTypeName(machineryType.getName()));
             MapUtils.findAndThen(workshopMap, vo.getWorkshopId(),
                     workshop -> vo.setWorkshopName(workshop.getName()));
+            MapUtils.findAndThen(workstationMap, vo.getWorkstationId(),
+                    workstation -> vo.setWorkstationName(workstation.getName()));
         });
     }
 

--
Gitblit v1.9.3