From db46314b9c56256a31c9114bd45d9bd63368cbd8 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期三, 29 七月 2026 18:06:22 +0800
Subject: [PATCH] feat(crm): 实现CRM客户管理查看全部权限功能

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/dv/machinery/MesDvMachineryServiceImpl.java |   32 ++++++++++++++++++++++++++++----
 1 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/dv/machinery/MesDvMachineryServiceImpl.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/dv/machinery/MesDvMachineryServiceImpl.java
index 9e74cf6..2325b04 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/dv/machinery/MesDvMachineryServiceImpl.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/dv/machinery/MesDvMachineryServiceImpl.java
@@ -54,6 +54,9 @@
     @Lazy
     private MesMdWorkshopService workshopService;
     @Resource
+    @Lazy
+    private cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService workstationService;
+    @Resource
     private MesWmBarcodeService barcodeService;
     @Resource
     @Lazy
@@ -201,7 +204,7 @@
             throw exception(DV_MACHINERY_IMPORT_LIST_IS_EMPTY);
         }
 
-        // 2. 鎵归噺鍔犺浇璁惧绫诲瀷鍜岃溅闂达紝鏋勫缓缂栫爜鍒板疄浣撶殑鏄犲皠
+        // 2. 鎵归噺鍔犺浇璁惧绫诲瀷銆佽溅闂村拰宸ヤ綔绔欙紝鏋勫缓缂栫爜鍒板疄浣撶殑鏄犲皠
         List<MesDvMachineryTypeDO> allTypes = machineryTypeService.getMachineryTypeList(
                 new cn.iocoder.yudao.module.mes.controller.admin.dv.machinery.vo.type.MesDvMachineryTypeListReqVO());
         Map<String, MesDvMachineryTypeDO> typeCodeMap = allTypes.stream()
@@ -210,6 +213,13 @@
                 cn.iocoder.yudao.framework.common.enums.CommonStatusEnum.ENABLE.getStatus());
         Map<String, MesMdWorkshopDO> workshopCodeMap = allWorkshops.stream()
                 .collect(Collectors.toMap(MesMdWorkshopDO::getCode, w -> w, (a, b) -> a));
+        List<cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO> allWorkstations =
+                workstationService.getWorkstationListByStatus(
+                        cn.iocoder.yudao.framework.common.enums.CommonStatusEnum.ENABLE.getStatus());
+        Map<String, cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO> workstationCodeMap =
+                allWorkstations.stream().collect(Collectors.toMap(
+                        cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO::getCode,
+                        w -> w, (a, b) -> a));
 
         // 3. 閬嶅巻锛岄�愪釜鍒涘缓 or 鏇存柊
         MesDvMachineryImportRespVO respVO = MesDvMachineryImportRespVO.builder()
@@ -249,24 +259,38 @@
                 return;
             }
 
-            // 3.4 鍒ゆ柇锛氬垱寤� or 鏇存柊
+            // 3.4 鏍¢獙宸ヤ綔绔欑紪鐮侊紙鍙�夛級
+            Long workstationId = null;
+            if (StrUtil.isNotBlank(importItem.getWorkstationCode())) {
+                cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO workstation =
+                        workstationCodeMap.get(importItem.getWorkstationCode());
+                if (workstation == null) {
+                    respVO.getFailureCodes().put(key, "宸ヤ綔绔欑紪鐮乕" + importItem.getWorkstationCode() + "]涓嶅瓨鍦�");
+                    return;
+                }
+                workstationId = workstation.getId();
+            }
+
+            // 3.5 鍒ゆ柇锛氬垱寤� or 鏇存柊
             MesDvMachineryDO existMachinery = machineryMapper.selectByCode(importItem.getCode());
             if (existMachinery == null) {
-                // 3.4.1 鍒涘缓
+                // 3.5.1 鍒涘缓
                 MesDvMachineryDO machinery = BeanUtils.toBean(importItem, MesDvMachineryDO.class);
                 machinery.setMachineryTypeId(machineryType.getId());
                 machinery.setWorkshopId(workshop.getId());
+                machinery.setWorkstationId(workstationId);
                 machineryMapper.insert(machinery);
                 // 鑷姩鐢熸垚鏉$爜
                 barcodeService.autoGenerateBarcode(BarcodeBizTypeEnum.MACHINERY.getValue(),
                         machinery.getId(), machinery.getCode(), machinery.getName());
                 respVO.getCreateCodes().add(importItem.getCode());
             } else if (updateSupport) {
-                // 3.4.2 鏇存柊
+                // 3.5.2 鏇存柊
                 MesDvMachineryDO updateObj = BeanUtils.toBean(importItem, MesDvMachineryDO.class);
                 updateObj.setId(existMachinery.getId());
                 updateObj.setMachineryTypeId(machineryType.getId());
                 updateObj.setWorkshopId(workshop.getId());
+                updateObj.setWorkstationId(workstationId);
                 machineryMapper.updateById(updateObj);
                 respVO.getUpdateCodes().add(importItem.getCode());
             } else {

--
Gitblit v1.9.3