feat(mes): 优化缺陷统计计算逻辑，重构模板校验方法，增强代码可读性

YunaiV
1 parent 015eab68
Showing 4 changed files with 76 additions and 52 deletions
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/qc/lqc/MesQcIqcDefectServiceImpl.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/qc/lqc/MesQcIqcServiceImpl.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/qc/template/MesQcTemplateService.java
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/qc/template/MesQcTemplateServiceImpl.java
 package cn.iocoder.yudao.module.mes.service.qc.lqc;
+import cn.hutool.core.util.ObjUtil;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
 import cn.iocoder.yudao.module.mes.controller.admin.qc.iqc.vo.defect.MesQcIqcDefectPageReqVO;
@@ -124,6 +125,7 @@ public class MesQcIqcDefectServiceImpl implements MesQcIqcDefectService {
      *
      * @param iqcId 来料检验单 ID
      */
+    @SuppressWarnings("DuplicatedCode")
     private void recalculateDefectStats(Long iqcId) {
         // 1. 查询所有缺陷记录
         List<MesQcIqcDefectDO> defects = iqcDefectMapper.selectListByIqcId(iqcId);
@@ -131,44 +133,38 @@ public class MesQcIqcDefectServiceImpl implements MesQcIqcDefectService {
         // 2. 更新每行的缺陷数量
         List<MesQcIqcLineDO> lines = iqcLineService.selectListByIqcId(iqcId);
         for (MesQcIqcLineDO line : lines) {
-            int critical = 0;
-            int major = 0;
-            int minor = 0;
+            int critical = 0, major = 0, minor = 0;
             for (MesQcIqcDefectDO defect : defects) {
-                // TODO @AI：notEquals continue；减少括号层级；
-                if (Objects.equals(defect.getLineId(), line.getId())) {
-                    // TODO @AI：defaultIfNull；变量改成 quantity
-                    int qty = defect.getDefectQuantity() != null ? defect.getDefectQuantity() : 1;
-                    if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.CRITICAL.getType())) {
-                        critical += qty;
-                    } else if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.MAJOR.getType())) {
-                        major += qty;
-                    } else if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.MINOR.getType())) {
-                        minor += qty;
-                    }
-                    // TODO @AI：抛出异常；未知的缺陷等级
+                if (ObjUtil.notEqual(defect.getLineId(), line.getId())) {
+                    continue;
+                }
+                int quantity = ObjUtil.defaultIfNull(defect.getDefectQuantity(), 1);
+                if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.CRITICAL.getType())) {
+                    critical += quantity;
+                } else if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.MAJOR.getType())) {
+                    major += quantity;
+                } else if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.MINOR.getType())) {
+                    minor += quantity;
+                } else {
+                    throw exception(QC_IQC_DEFECT_LEVEL_UNKNOWN);
                 }
             }
-            MesQcIqcLineDO updateLine = new MesQcIqcLineDO().setId(line.getId())
-                    .setCriticalQuantity(critical).setMajorQuantity(major).setMinorQuantity(minor);
-            // 直接通过 lineService 内部更新（避免循环依赖，使用 mapper）
-            // TODO @AI：单独搞个 updateLineDefectStats 方法；减少 updateLine 变量的作用域；
-            iqcLineMapper.updateById(updateLine);
+            updateLineDefectStats(line.getId(), critical, major, minor);
         }
         // 3.1 汇总主表的缺陷数量
         int totalCritical = 0, totalMajor = 0, totalMinor = 0;
         for (MesQcIqcDefectDO defect : defects) {
-            // TODO @AI：defaultIfNull；变量改成 quantity
-            int qty = defect.getDefectQuantity() != null ? defect.getDefectQuantity() : 1;
+            int quantity = ObjUtil.defaultIfNull(defect.getDefectQuantity(), 1);
             if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.CRITICAL.getType())) {
-                totalCritical += qty;
+                totalCritical += quantity;
             } else if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.MAJOR.getType())) {
-                totalMajor += qty;
+                totalMajor += quantity;
             } else if (Objects.equals(defect.getDefectLevel(), MesQcDefectLevelEnum.MINOR.getType())) {
-                totalMinor += qty;
+                totalMinor += quantity;
+            } else {
+                throw exception(QC_IQC_DEFECT_LEVEL_UNKNOWN);
             }
-            // TODO @AI：抛出异常；未知的缺陷等级
         }
         // 3.2 计算缺陷率
         MesQcIqcDO iqc = iqcMapper.selectById(iqcId);
@@ -191,4 +187,14 @@ public class MesQcIqcDefectServiceImpl implements MesQcIqcDefectService {
         iqcMapper.updateById(updateIqc);
     }
+    // TODO @AI：在 iqclineservice 里，提供一个方法，直接根据行 ID 列表，批量更新缺陷统计数量。然后这里直接调用即可，避免循环更新数据库。
+    /**
+     * 更新检验行的缺陷统计数量
+     */
+    private void updateLineDefectStats(Long lineId, int critical, int major, int minor) {
+        MesQcIqcLineDO updateLine = new MesQcIqcLineDO().setId(lineId)
+                .setCriticalQuantity(critical).setMajorQuantity(major).setMinorQuantity(minor);
+        iqcLineMapper.updateById(updateLine);
+    }
+
 }
 package cn.iocoder.yudao.module.mes.service.qc.lqc;
+import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.ObjUtil;
+import cn.iocoder.yudao.framework.common.exception.ErrorCode;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
 import cn.iocoder.yudao.module.mes.controller.admin.qc.iqc.vo.MesQcIqcPageReqVO;
@@ -9,17 +11,14 @@ import cn.iocoder.yudao.module.mes.dal.dataobject.qc.lqc.MesQcIqcDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.qc.template.MesQcTemplateDO;
 import cn.iocoder.yudao.module.mes.dal.dataobject.qc.template.MesQcTemplateItemDO;
 import cn.iocoder.yudao.module.mes.dal.mysql.qc.lqc.MesQcIqcMapper;
-import cn.iocoder.yudao.module.mes.dal.mysql.qc.template.MesQcTemplateItemMapper;
-import cn.iocoder.yudao.module.mes.dal.mysql.qc.template.MesQcTemplateMapper;
 import cn.iocoder.yudao.module.mes.enums.qc.MesQcIqcStatusEnum;
 import cn.iocoder.yudao.module.mes.enums.qc.MesQcTypeEnum;
+import cn.iocoder.yudao.module.mes.service.qc.template.MesQcTemplateService;
 import jakarta.annotation.Resource;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.validation.annotation.Validated;
-import java.util.Objects;
-
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;
@@ -35,9 +34,7 @@ public class MesQcIqcServiceImpl implements MesQcIqcService {
     @Resource
     private MesQcIqcMapper iqcMapper;
     @Resource
-    private MesQcTemplateMapper templateMapper;
-    @Resource
-    private MesQcTemplateItemMapper templateItemMapper;
+    private MesQcTemplateService templateService;
     @Resource
     private MesQcIqcLineService iqcLineService;
@@ -51,16 +48,13 @@ public class MesQcIqcServiceImpl implements MesQcIqcService {
         validateIqcCodeUnique(null, createReqVO.getCode());
         // 1.2 查找物料关联的 IQC 检测模板
         Long templateId = createReqVO.getTemplateId();
-        // TODO @AI：直接使用 templateService 的 validateTemplateExists() 方法，后续如果有更多模板相关的校验，也可以放在该方法中
-        MesQcTemplateDO template = templateMapper.selectById(templateId);
-        // TODO @AI：CollUtil.contains 这样更合适；
-        if (template == null || template.getTypes() == null || !template.getTypes().contains(MesQcTypeEnum.IQC.getType())) {
+        MesQcTemplateDO template = templateService.validateTemplateExists(templateId);
+        if (!CollUtil.contains(template.getTypes(), MesQcTypeEnum.IQC.getType())) {
             throw exception(QC_IQC_NO_TEMPLATE);
         }
         // 3.1 从模板的产品关联中获取检测参数（min_check_quantity、max_unqualified_quantity）
-        // TODO @AI：使用 templateItemService 方法，不要直接使用对方的 mapper；
-        MesQcTemplateItemDO templateItem = templateItemMapper.selectByTemplateIdAndItemId(
+        MesQcTemplateItemDO templateItem = templateService.getTemplateItemByTemplateIdAndItemId(
                 templateId, createReqVO.getItemId());
         if (templateItem != null) {
             if (createReqVO.getMinCheckQuantity() == null) {
@@ -85,11 +79,7 @@ public class MesQcIqcServiceImpl implements MesQcIqcService {
         // 1.1 校验存在
         MesQcIqcDO iqc = validateIqcExists(updateReqVO.getId());
         // 1.2 校验状态为草稿
-        // TODO @AI：notEquals；
-        // TODO @AI：抽成一个方法，这样 completeIqc 和 deleteIqc 也可以复用，避免重复代码；
-        if (!Objects.equals(iqc.getStatus(), MesQcIqcStatusEnum.PREPARE.getType())) {
-            throw exception(QC_IQC_ONLY_PREPARE_CAN_COMPLETE);
-        }
+        validatePrepareStatus(iqc, QC_IQC_ONLY_PREPARE_CAN_COMPLETE);
         // 1.3 校验编号唯一
         validateIqcCodeUnique(updateReqVO.getId(), updateReqVO.getCode());
@@ -103,9 +93,7 @@ public class MesQcIqcServiceImpl implements MesQcIqcService {
         // 1.1 校验存在
         MesQcIqcDO iqc = validateIqcExists(id);
         // 1.2 校验状态为草稿
-        if (!Objects.equals(iqc.getStatus(), MesQcIqcStatusEnum.PREPARE.getType())) {
-            throw exception(QC_IQC_ONLY_PREPARE_CAN_COMPLETE);
-        }
+        validatePrepareStatus(iqc, QC_IQC_ONLY_PREPARE_CAN_COMPLETE);
         // 1.3 校验合格品 + 不合格品 = 检测数量
         if (iqc.getCheckQuantity() != null && iqc.getCheckQuantity() > 0) {
             int total = (iqc.getQualifiedQuantity() != null ? iqc.getQualifiedQuantity() : 0)
@@ -129,9 +117,7 @@ public class MesQcIqcServiceImpl implements MesQcIqcService {
         // 1.1 校验存在
         MesQcIqcDO iqc = validateIqcExists(id);
         // 1.2 仅草稿可删
-        if (!Objects.equals(iqc.getStatus(), MesQcIqcStatusEnum.PREPARE.getType())) {
-            throw exception(QC_IQC_ONLY_PREPARE_CAN_DELETE);
-        }
+        validatePrepareStatus(iqc, QC_IQC_ONLY_PREPARE_CAN_DELETE);
         // 2.1 删除主表
         iqcMapper.deleteById(id);
@@ -149,6 +135,13 @@ public class MesQcIqcServiceImpl implements MesQcIqcService {
         return iqc;
     }
+    // TODO @AI：就搞个方法，传递 id 进去；抛出不为 草稿状态的异常；然后在 update/delete/complete 方法中调用这个方法（不用专门 errorCode）
+    private void validatePrepareStatus(MesQcIqcDO iqc, ErrorCode errorCode) {
+        if (ObjUtil.notEqual(iqc.getStatus(), MesQcIqcStatusEnum.PREPARE.getType())) {
+            throw exception(errorCode);
+        }
+    }
+
     private void validateIqcCodeUnique(Long id, String code) {
         MesQcIqcDO iqc = iqcMapper.selectByCode(code);
         if (iqc == null) {
@@ -90,6 +90,23 @@ public interface MesQcTemplateService {
         return convertMap(getTemplateList(ids), MesQcTemplateDO::getId);
     }
+    /**
+     * 校验质检方案存在
+     *
+     * @param id 编号
+     * @return 质检方案
+     */
+    MesQcTemplateDO validateTemplateExists(Long id);
+
+    /**
+     * 根据方案 ID + 产品 ID 获取产品关联
+     *
+     * @param templateId 方案 ID
+     * @param itemId 产品 ID
+     * @return 产品关联（可能为 null）
+     */
+    MesQcTemplateItemDO getTemplateItemByTemplateIdAndItemId(Long templateId, Long itemId);
+
     // ========== 质检方案-检测指标项 ==========
     /**
@@ -84,10 +84,13 @@ public class MesQcTemplateServiceImpl implements MesQcTemplateService {
         templateItemMapper.deleteByTemplateId(id);
     }
-    private void validateTemplateExists(Long id) {
-        if (templateMapper.selectById(id) == null) {
+    @Override
+    public MesQcTemplateDO validateTemplateExists(Long id) {
+        MesQcTemplateDO template = templateMapper.selectById(id);
+        if (template == null) {
             throw exception(QC_TEMPLATE_NOT_EXISTS);
         }
+        return template;
     }
     private void validateTemplateCodeUnique(Long id, String code) {
@@ -101,6 +104,11 @@ public class MesQcTemplateServiceImpl implements MesQcTemplateService {
     }
     @Override
+    public MesQcTemplateItemDO getTemplateItemByTemplateIdAndItemId(Long templateId, Long itemId) {
+        return templateItemMapper.selectByTemplateIdAndItemId(templateId, itemId);
+    }
+
+    @Override
     public MesQcTemplateDO getTemplate(Long id) {
         return templateMapper.selectById(id);
     }