liu
2026-09-02 1fb5fc73e8df33b25337ab34f669ab03e0d9ce06
yudao-module-mdm/src/main/java/cn/iocoder/yudao/module/mdm/controller/admin/unit/MdmUnitMeasureController.java
@@ -1,9 +1,12 @@
package cn.iocoder.yudao.module.mdm.controller.admin.unit;
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureExcelVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureImportExcelVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasureImportRespVO;
import cn.iocoder.yudao.module.mdm.controller.admin.unit.vo.MdmUnitMeasurePageReqVO;
@@ -23,9 +26,13 @@
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
@Tag(name = "管理后台 - MDM 计量单位")
@@ -101,8 +108,37 @@
        return success(unitMeasureService.getUnitMeasureList(ids));
    }
    @GetMapping("/export-excel")
    @Operation(summary = "导出计量单位 Excel")
    @PreAuthorize("@ss.hasPermission('mdm:unit-measure:export')")
    @ApiAccessLog(operateType = EXPORT)
    public void exportUnitMeasureExcel(@Valid MdmUnitMeasurePageReqVO pageReqVO,
                                       HttpServletResponse response) throws IOException {
        pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
        List<MdmUnitMeasureDO> list = unitMeasureService.getUnitMeasurePage(pageReqVO).getList();
        // 主单位编码映射:一次查询全部计量单位,用于解析 primaryId -> 编码
        MdmUnitMeasurePageReqVO allReqVO = new MdmUnitMeasurePageReqVO();
        allReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
        Map<Long, String> unitCodeMap = new HashMap<>();
        for (MdmUnitMeasureDO unit : unitMeasureService.getUnitMeasurePage(allReqVO).getList()) {
            unitCodeMap.put(unit.getId(), unit.getCode());
        }
        // 组装导出 VO(主单位编码从 DO.primaryId 解析)
        List<MdmUnitMeasureExcelVO> excelList = new ArrayList<>();
        for (MdmUnitMeasureDO unit : list) {
            MdmUnitMeasureExcelVO excelVO = BeanUtils.toBean(unit, MdmUnitMeasureExcelVO.class);
            if (unit.getPrimaryId() != null) {
                excelVO.setPrimaryCode(unitCodeMap.get(unit.getPrimaryId()));
            }
            excelList.add(excelVO);
        }
        // 导出 Excel
        ExcelUtils.write(response, "计量单位数据.xls", "计量单位", MdmUnitMeasureExcelVO.class, excelList);
    }
    @GetMapping("/get-import-template")
    @Operation(summary = "获得导入计量单位模板")
    @PreAuthorize("@ss.hasPermission('mdm:unit-measure:import')")
    public void importTemplate(HttpServletResponse response) throws IOException {
        List<MdmUnitMeasureImportExcelVO> list = Arrays.asList(
                MdmUnitMeasureImportExcelVO.builder().code("PC").name("片").primaryFlag(true).changeRate(new java.math.BigDecimal("1")).status(0).remark("示例:主单位").build(),