gaoluyang
2026-10-16 7081c41bc3beb0ab49a22974ea4b010ae6854afe
yudao-module-mdm/src/main/java/cn/iocoder/yudao/module/mdm/service/unit/MdmUnitMeasureServiceImpl.java
@@ -2,8 +2,14 @@
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureImportExcelVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureImportReqVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureImportRespVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasurePageReqVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureSaveReqVO;
import cn.iocoder.yudao.module.mdm.dal.dataobject.unit.MdmUnitMeasureDO;
@@ -17,6 +23,7 @@
import java.util.*;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.filterList;
import static cn.iocoder.yudao.module.mdm.enums.ErrorCodeConstants.*;
/**
@@ -35,6 +42,53 @@
    private MdmItemMapper itemMapper;
    @Override
    @Transactional(rollbackFor = Exception.class)
    public MdmUnitMeasureImportRespVO importUnitMeasureList(List<MdmUnitMeasureImportExcelVO> importUnits,
                                                            MdmUnitMeasureImportReqVO importReqVO) {
        // 1. 校验非空(以单位编码作为导入主键)
        importUnits = filterList(importUnits, item -> StrUtil.isNotBlank(item.getCode()));
        if (CollUtil.isEmpty(importUnits)) {
            throw exception(MDM_UNIT_MEASURE_IMPORT_LIST_IS_EMPTY);
        }
        // 2. 逐条处理
        MdmUnitMeasureImportRespVO respVO = MdmUnitMeasureImportRespVO.builder()
                .createUnitCodes(new ArrayList<>())
                .updateUnitCodes(new ArrayList<>())
                .failureUnitCodes(new LinkedHashMap<>())
                .build();
        importUnits.forEach(importUnit -> {
            try {
                // 2.1 判断是否存在(按编码唯一)
                MdmUnitMeasureDO existUnit = unitMeasureMapper.selectByCode(importUnit.getCode());
                if (existUnit == null) {
                    // 2.1.1 新增
                    MdmUnitMeasureDO unit = BeanUtils.toBean(importUnit, MdmUnitMeasureDO.class);
                    if (unit.getStatus() == null) {
                        unit.setStatus(CommonStatusEnum.ENABLE.getStatus());
                    }
                    unitMeasureMapper.insert(unit);
                    respVO.getCreateUnitCodes().add(importUnit.getCode());
                    return;
                }
                // 2.1.2 已存在,判断是否允许更新
                if (!importReqVO.getUpdateSupport()) {
                    respVO.getFailureUnitCodes().put(importUnit.getCode(),
                            StrUtil.format(MDM_UNIT_MEASURE_CODE_EXISTS.getMsg(), importUnit.getCode()));
                    return;
                }
                MdmUnitMeasureDO updateUnit = BeanUtils.toBean(importUnit, MdmUnitMeasureDO.class)
                        .setId(existUnit.getId());
                unitMeasureMapper.updateById(updateUnit);
                respVO.getUpdateUnitCodes().add(importUnit.getCode());
            } catch (ServiceException ex) {
                respVO.getFailureUnitCodes().put(importUnit.getCode(), ex.getMessage());
            }
        });
        return respVO;
    }
    @Override
    public Long createUnitMeasure(MdmUnitMeasureSaveReqVO createReqVO) {
        // 校验数据
        validateUnitMeasureSaveData(createReqVO);