feat(mes): 优化工作站模块，新增获取车间列表和车间 Map 方法，重构表单弹窗逻辑

YunaiV
1 parent 35b28b3c
Showing 10 changed files with 58 additions and 37 deletions
yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/collection/CollectionUtils.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/md/workstation/MesMdWorkstationController.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/dataobject/md/workstation/MesMdWorkshopDO.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/dataobject/md/workstation/MesMdWorkstationDO.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/dataobject/md/workstation/MesMdWorkstationMachineDO.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/dataobject/md/workstation/MesMdWorkstationToolDO.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/dal/dataobject/md/workstation/MesMdWorkstationWorkerDO.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/md/item/MesMdProductBomServiceImpl.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/md/workstation/MesMdWorkshopService.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/md/workstation/MesMdWorkshopServiceImpl.java
@@ -349,4 +349,27 @@ public class CollectionUtils {
         return (LinkedHashSet<T>) toCollection(LinkedHashSet.class, elementType, value);
     }
+    public static boolean dfs(Long node, Map<Long, Set<Long>> graph) {
+        return dfs(node, graph, new HashSet<>(), new HashSet<>());
+    }
+
+    private static boolean dfs(Long node, Map<Long, Set<Long>> graph, Set<Long> visited, Set<Long> inStack) {
+        if (inStack.contains(node)) {
+            return true;
+        }
+        if (visited.contains(node)) {
+            return false;
+        }
+        visited.add(node);
+        inStack.add(node);
+        Set<Long> neighbors = graph.getOrDefault(node, Collections.emptySet());
+        for (Long neighbor : neighbors) {
+            if (dfs(neighbor, graph, visited, inStack)) {
+                return true;
+            }
+        }
+        inStack.remove(node);
+        return false;
+    }
+
 }
 \ No newline at end of file
@@ -30,6 +30,7 @@ import java.io.IOException;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
 import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
@@ -110,14 +111,14 @@ public class MesMdWorkstationController {
     // ==================== 拼接 VO ====================
+    @SuppressWarnings("CodeBlock2Expr")
     private List<MesMdWorkstationRespVO> buildWorkstationRespVOList(List<MesMdWorkstationDO> list) {
         if (CollUtil.isEmpty(list)) {
             return Collections.emptyList();
         }
         // 1. 批量获取车间信息
-        Map<Long, MesMdWorkshopDO> workshopMap = convertMap(
-                workshopService.getWorkshopListByStatus(null),
-                MesMdWorkshopDO::getId);
+        Map<Long, MesMdWorkshopDO> workshopMap = workshopService.getWorkshopMap(
+                convertSet(list, MesMdWorkstationDO::getWorkshopId));
         // 2. 拼接 VO
         return BeanUtils.toBean(list, MesMdWorkstationRespVO.class, vo -> {
             MapUtils.findAndThen(workshopMap, vo.getWorkshopId(),
@@ -40,14 +40,12 @@ public class MesMdWorkshopDO extends BaseDO {
      * 面积（平方米）
      */
     private BigDecimal area;
-    // DONE @AI：通过 @ 去关联
     /**
      * 负责人用户编号
      *
      * 关联 AdminUserDO 的 id
      */
     private Long chargeUserId;
-    // DONE @AI：通过 @ 去关联
     /**
      * 状态
      *
@@ -38,28 +38,24 @@ public class MesMdWorkstationDO extends BaseDO {
      * 工作站地点
      */
     private String address;
-    // DONE @AI：通过 @ 去关联，按照别的模块
     /**
      * 所在车间编号
      *
      * 关联 {@link MesMdWorkshopDO#getId()}
      */
     private Long workshopId;
-    // DONE @AI：通过 @ 去关联，按照别的模块
     /**
      * 工序编号
      *
      * TODO @芋艿：等 pro 模块，关联工序表
      */
     private Long processId;
-    // DONE @AI：通过 @ 去关联，按照别的模块
     /**
      * 线边库编号
      *
      * TODO @芋艿：等 wm 模块，关联仓库表
      */
     private Long warehouseId;
-    // DONE @AI：通过 @ 去关联，按照别的模块
     /**
      * 库区编号
      *
@@ -72,7 +68,6 @@ public class MesMdWorkstationDO extends BaseDO {
      * TODO @芋艿：等 wm 模块，关联库位表
      */
     private Long areaId;
-    // DONE @AI：通过 @ 去关联，按照别的模块
     /**
      * 状态
      *
@@ -6,7 +6,6 @@ import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
 import lombok.*;
-// DONE @AI：设备资源表；已修正所有中文注释
 /**
  * MES 设备资源 DO
  *
@@ -6,7 +6,6 @@ import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
 import lombok.*;
-// DONE @AI：工装夹具资源表；已修正所有中文注释
 /**
  * MES 工装夹具资源 DO
  *
@@ -6,7 +6,6 @@ import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
 import lombok.*;
-// DONE @AI：人力资源表；已修正所有中文注释
 /**
  * MES 人力资源 DO
  *
 package cn.iocoder.yudao.module.mes.service.md.item;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils;
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
 import cn.iocoder.yudao.module.mes.controller.admin.md.item.vo.bom.MesMdProductBomPageReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.md.item.vo.bom.MesMdProductBomSaveReqVO;
@@ -92,6 +93,7 @@ public class MesMdProductBomServiceImpl implements MesMdProductBomService {
         }
     }
+    // TODO @AI：hasXXXCycle 这样？
     /**
      * 检测新增边 (itemId -> bomItemId) 后，BOM 图是否存在闭环
      *
@@ -108,36 +110,14 @@ public class MesMdProductBomServiceImpl implements MesMdProductBomService {
         // 1.3  添加待新增的边
         graph.computeIfAbsent(itemId, k -> new HashSet<>()).add(bomItemId);
         // 2. DFS 检测环
-        Set<Long> visited = new HashSet<>();
-        Set<Long> inStack = new HashSet<>();
         for (Long node : graph.keySet()) {
-            if (dfs(node, graph, visited, inStack)) {
+            if (CollectionUtils.dfs(node, graph)) {
                 return true;
             }
         }
         return false;
     }
-    // DONE @AI：dfs 为 BOM 环检测的私有方法，属于业务逻辑，保留在 ServiceImpl 中
-    private boolean dfs(Long node, Map<Long, Set<Long>> graph, Set<Long> visited, Set<Long> inStack) {
-        if (inStack.contains(node)) {
-            return true;
-        }
-        if (visited.contains(node)) {
-            return false;
-        }
-        visited.add(node);
-        inStack.add(node);
-        Set<Long> neighbors = graph.getOrDefault(node, Collections.emptySet());
-        for (Long neighbor : neighbors) {
-            if (dfs(neighbor, graph, visited, inStack)) {
-                return true;
-            }
-        }
-        inStack.remove(node);
-        return false;
-    }
-
     @Override
     public MesMdProductBomDO getProductBom(Long id) {
         return productBomMapper.selectById(id);
@@ -6,9 +6,12 @@ import cn.iocoder.yudao.module.mes.controller.admin.md.workstation.vo.workshop.M
 import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkshopDO;
 import jakarta.validation.Valid;
+import java.util.Collection;
 import java.util.List;
+import java.util.Map;
+
+import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMap;
-// DONE @AI：方法注释
 /**
  * MES 车间 Service 接口
  *
@@ -55,6 +58,24 @@ public interface MesMdWorkshopService {
     PageResult<MesMdWorkshopDO> getWorkshopPage(MesMdWorkshopPageReqVO pageReqVO);
     /**
+     * 获得车间列表
+     *
+     * @param ids 编号列表
+     * @return 车间列表
+     */
+    List<MesMdWorkshopDO> getWorkshopList(Collection<Long> ids);
+
+    /**
+     * 获得车间 Map
+     *
+     * @param ids 编号列表
+     * @return 车间 Map
+     */
+    default Map<Long, MesMdWorkshopDO> getWorkshopMap(Collection<Long> ids) {
+        return convertMap(getWorkshopList(ids), MesMdWorkshopDO::getId);
+    }
+
+    /**
      * 获得指定状态的车间列表
      *
      * @param status 状态
@@ -14,6 +14,7 @@ import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
 import org.springframework.validation.annotation.Validated;
+import java.util.Collection;
 import java.util.List;
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
@@ -113,6 +114,11 @@ public class MesMdWorkshopServiceImpl implements MesMdWorkshopService {
     }
     @Override
+    public List<MesMdWorkshopDO> getWorkshopList(Collection<Long> ids) {
+        return workshopMapper.selectByIds(ids);
+    }
+
+    @Override
     public List<MesMdWorkshopDO> getWorkshopListByStatus(Integer status) {
         return workshopMapper.selectListByStatus(status);
     }