feat 功能修正

xiaoyi

3 天以前 abf1c0a35d85d38239ff5d2ed746a4e4b797c9d1

 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/wm/stocktaking/task/MesWmStockTakingTaskLineServiceImpl.java

@@ -7,6 +7,8 @@
import cn.iocoder.yudao.module.mes.controller.admin.wm.materialstock.vo.MesWmMaterialStockListReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.line.MesWmStockTakingTaskLinePageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.line.MesWmStockTakingTaskLineSaveReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.line.MesWmStockTakingTaskLineImportExcelVO;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.line.MesWmStockTakingTaskLineImportRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.wm.stocktaking.task.vo.result.MesWmStockTakingTaskResultSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.materialstock.MesWmMaterialStockDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.stocktaking.plan.MesWmStockTakingPlanParamDO;
@@ -35,7 +37,7 @@
/**
 * MES 盘点任务行 Service 实现类
 *
 * @author 芋道源码
 * @author 超级管理员
 */
@Service
@Validated
@@ -50,6 +52,8 @@
    private MesWmMaterialStockService materialStockService;
    @Resource
    private MesWmStockTakingTaskService stockTakingTaskService;
    @Resource
    private cn.iocoder.yudao.module.mdm.api.item.MdmItemApi mdmItemApi;

    @Override
    @Transactional(rollbackFor = Exception.class)
@@ -230,4 +234,58 @@
        return param != null ? param.getValueId() : null;
    }

    @Override
    public MesWmStockTakingTaskLineImportRespVO importTakingQuantity(Long taskId,
                                                                    List<MesWmStockTakingTaskLineImportExcelVO> list) {
        MesWmStockTakingTaskLineImportRespVO respVO = new MesWmStockTakingTaskLineImportRespVO();
        respVO.setSuccessCount(0);
        respVO.setFailureCount(0);
        respVO.setFailureMessages(new java.util.ArrayList<>());
        if (CollUtil.isEmpty(list)) {
            return respVO;
        }
        // 1. 获取盘点任务行
        List<MesWmStockTakingTaskLineDO> lines = stockTakingTaskLineMapper.selectListByTaskId(taskId);
        if (CollUtil.isEmpty(lines)) {
            throw exception(cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.WM_STOCK_TAKING_TASK_LINE_NOT_EXISTS);
        }
        // 2. 构建 物料编码 -> 物料ID 映射
        java.util.Set<Long> itemIds = lines.stream().map(MesWmStockTakingTaskLineDO::getItemId)
                .filter(Objects::nonNull).collect(java.util.stream.Collectors.toSet());
        java.util.Map<Long, cn.iocoder.yudao.module.mdm.api.item.dto.MdmItemRespDTO> itemMap =
                mdmItemApi.getItemMap(itemIds);
        java.util.Map<String, Long> codeToItemIdMap = new java.util.HashMap<>();
        itemMap.values().forEach(item -> {
            if (item.getCode() != null) {
                codeToItemIdMap.put(item.getCode(), item.getId());
            }
        });
        // 3. 逐行匹配并写入实盘数量
        for (MesWmStockTakingTaskLineImportExcelVO row : list) {
            Long itemId = codeToItemIdMap.get(row.getItemCode());
            if (itemId == null) {
                respVO.setFailureCount(respVO.getFailureCount() + 1);
                respVO.getFailureMessages().add("物料编码不存在: " + row.getItemCode());
                continue;
            }
            MesWmStockTakingTaskLineDO matched = lines.stream()
                    .filter(line -> Objects.equals(line.getItemId(), itemId)
                            && Objects.equals(line.getBatchCode(), row.getBatchCode()))
                    .findFirst().orElse(null);
            if (matched == null) {
                respVO.setFailureCount(respVO.getFailureCount() + 1);
                respVO.getFailureMessages().add("未找到对应盘点行: 物料=" + row.getItemCode() + " 批次=" + row.getBatchCode());
                continue;
            }
            if (row.getTakingQuantity() == null) {
                respVO.setFailureCount(respVO.getFailureCount() + 1);
                respVO.getFailureMessages().add("实盘数量为空: 物料=" + row.getItemCode());
                continue;
            }
            updateStockTakingTaskLineTakingQuantity(matched.getId(), row.getTakingQuantity());
            respVO.setSuccessCount(respVO.getSuccessCount() + 1);
        }
        return respVO;
    }

}