From 704265308b1227dbada5a36b2027b588c062ded2 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期一, 12 十月 2026 09:14:37 +0800
Subject: [PATCH] feat(mes): 新增业务码支持并优化追溯查询功能

---
 yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/batch/MesProBatchController.java |   70 ++++++++++++++++++++++++++++-------
 1 files changed, 56 insertions(+), 14 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/batch/MesProBatchController.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/batch/MesProBatchController.java
index 55a450b..bde13d7 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/batch/MesProBatchController.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/batch/MesProBatchController.java
@@ -1,6 +1,7 @@
 package cn.iocoder.yudao.module.mes.controller.admin.pro.batch;
 
 import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.StrUtil;
 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;
@@ -10,6 +11,8 @@
 import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.bagcode.vo.MesProBagCodeRespVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.batch.vo.MesProBatchPageReqVO;
+import cn.iocoder.yudao.module.mes.controller.admin.pro.batch.vo.MesProBatchPreGenerateReqVO;
+import cn.iocoder.yudao.module.mes.controller.admin.pro.batch.vo.MesProBatchPreGenerateRespVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.batch.vo.MesProBatchRespVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.batch.vo.MesProBatchSaveReqVO;
 import cn.iocoder.yudao.module.mes.controller.admin.pro.batch.vo.MesProBatchTraceRespVO;
@@ -75,10 +78,27 @@
     private MesProBagCodeService bagCodeService;
 
     @PostMapping("/create")
-    @Operation(summary = "鍒涘缓鐢熶骇鎵规锛堣嚜鍔ㄧ敓鎴愭壒娆″彿锛�")
+    @Operation(summary = "鍒涘缓鐢熶骇鎵规锛堜竴娆″綍鍏ヤ笟鍔′俊鎭紝鑷姩鐢熸垚鎵规鍙凤級")
     @PreAuthorize("@ss.hasPermission('mes:pro-batch:create')")
     public CommonResult<Long> createBatch(@Valid @RequestBody MesProBatchSaveReqVO createReqVO) {
         return success(batchService.createBatch(createReqVO));
+    }
+
+    @PostMapping("/pre-generate")
+    @Operation(summary = "棰勭敓鎴愮敓浜ф壒娆′簩缁寸爜锛堜粎鐢熸垚涓存椂鎵规鐮侊紝涓嶅綍鍏ヤ笟鍔′俊鎭級")
+    @PreAuthorize("@ss.hasPermission('mes:pro-batch:create')")
+    public CommonResult<List<MesProBatchPreGenerateRespVO>> preGenerateBatch(
+            @Valid @RequestBody MesProBatchPreGenerateReqVO preGenerateReqVO) {
+        List<MesProBatchDO> list = batchService.preGenerateBatch(preGenerateReqVO);
+        return success(BeanUtils.toBean(list, MesProBatchPreGenerateRespVO.class));
+    }
+
+    @PutMapping("/fill-info")
+    @Operation(summary = "琛ュ綍鐢熶骇鎵规涓氬姟淇℃伅锛堢敓鎴愭寮忔壒娆″彿锛屼粎涓存椂鎵规鍙敤锛�")
+    @PreAuthorize("@ss.hasPermission('mes:pro-batch:update')")
+    public CommonResult<Boolean> fillBatchInfo(@Valid @RequestBody MesProBatchSaveReqVO fillReqVO) {
+        batchService.fillBatchInfo(fillReqVO);
+        return success(true);
     }
 
     @PutMapping("/update")
@@ -161,7 +181,7 @@
     @Parameter(name = "code", description = "鎵规鍙凤紙浜岀淮鐮佸唴瀹癸級", required = true, example = "PC20260818L01001")
     @PreAuthorize("@ss.hasPermission('mes:pro-batch:query')")
     public CommonResult<MesProBatchTraceRespVO> getBatchByCode(@RequestParam("code") String code) {
-        MesProBatchDO batch = batchService.getBatchByCode(code);
+        MesProBatchDO batch = batchService.getBatchByScanCode(code);
         if (batch == null) {
             throw exception(PRO_BATCH_NOT_EXISTS);
         }
@@ -183,29 +203,50 @@
         if (pallets.size() + 1 > QrCodeImageUtils.MAX_QR_EXPORT) {
             throw exception(PRO_QR_EXPORT_TOO_MANY);
         }
-        QrCodeImageUtils.writeZip(response, "鎵规浜岀淮鐮�-" + batch.getCode(), buildBatchQrLabels(batch, pallets));
+        String qrCode = StrUtil.blankToDefault(batch.getCode(), batch.getTempCode());
+        QrCodeImageUtils.writeZip(response, "鎵规浜岀淮鐮�-" + qrCode, buildBatchQrLabels(batch, pallets));
     }
 
     // ==================== 鎷兼帴 VO ====================
+
+    @PostMapping("/generate-business-code")
+    @Operation(summary = "琛ョ敓鎴愬凡鏈夌敓浜ф壒娆′笟鍔$爜锛屽凡鏈夌爜鍊间笉鍙�")
+    @PreAuthorize("@ss.hasPermission('mes:pro-batch:create')")
+    public CommonResult<Integer> generateBusinessCodes(@Valid @RequestBody
+            cn.iocoder.yudao.module.mes.controller.admin.pro.vo.MesProBusinessCodeGenerateReqVO reqVO) {
+        return success(batchService.generateBusinessCodes(reqVO.getIds()));
+    }
+
+    @GetMapping("/export-business-qr")
+    @Operation(summary = "瀵煎嚭鎵规鍙婃墍灞炴墭鐩樹笟鍔′簩缁寸爜锛屽唴瀹逛负绾笟鍔$爜鍊�")
+    @PreAuthorize("@ss.hasPermission('mes:pro-batch:export')")
+    @ApiAccessLog(operateType = EXPORT)
+    public void exportBusinessQr(@RequestParam("batchId") Long batchId,
+                                  HttpServletResponse response) throws IOException {
+        MesProBatchDO batch = batchService.getBatch(batchId);
+        if (batch == null) throw exception(PRO_BATCH_NOT_EXISTS);
+        if (cn.iocoder.yudao.module.mes.enums.pro.MesProBatchStatusEnum.CANCELED.getStatus().equals(batch.getStatus()))
+            throw new cn.iocoder.yudao.framework.common.exception.ServiceException(400, "宸蹭綔搴熸壒娆′笉鍙鍑轰笟鍔′簩缁寸爜");
+        List<MesProPalletDO> pallets = palletService.getPalletListByBatchId(batchId).stream()
+                .filter(record -> !cn.iocoder.yudao.module.mes.enums.pro.MesProPalletStatusEnum.CANCELED.getStatus().equals(record.getStatus())).toList();
+        if (pallets.size() + 1 > QrCodeImageUtils.MAX_QR_EXPORT) throw exception(PRO_QR_EXPORT_TOO_MANY);
+        List<QrLabelInfo> labels = new java.util.ArrayList<>();
+        labels.add(MesProQrLabelBuilder.buildBatchBusinessCodeLabel(batch));
+        pallets.forEach(pallet -> labels.add(MesProQrLabelBuilder.buildPalletBusinessCodeLabel(pallet)));
+        QrCodeImageUtils.writeZip(response, "鎵规涓氬姟浜岀淮鐮�-" + batch.getBusinessCode(), labels);
+    }
 
     private List<MesProBatchRespVO> buildBatchRespVOList(List<MesProBatchDO> list) {
         if (CollUtil.isEmpty(list)) {
             return Collections.emptyList();
         }
-        // 1. 鑾峰緱鍏宠仈鏁版嵁锛堣矗浠讳汉銆佸搧绉嶇墿鏂欙級
+        // 1. 鑾峰緱鍏宠仈鏁版嵁锛堣矗浠讳汉锛夛紱鍝佺宸蹭笅娌夊埌琚嬬爜鏁版嵁鍖咃紝鎵规鍒楄〃涓嶅啀灞曠ず
         Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(
                 convertSet(list, MesProBatchDO::getOwnerUserId));
-        Map<Long, MesMdmItemDO> itemMap = mdmItemService.getItemMap(
-                convertSet(list, MesProBatchDO::getItemId));
-        // 2. 鏋勫缓缁撴灉
+        Map<Long, MesMdmItemDO> itemMap = mdmItemService.getItemMap(convertSet(list, MesProBatchDO::getItemId));
         return BeanUtils.toBean(list, MesProBatchRespVO.class, vo -> {
-            MapUtils.findAndThen(userMap, vo.getOwnerUserId(), user ->
-                    vo.setOwnerUserName(user.getNickname()));
-            MapUtils.findAndThen(itemMap, vo.getItemId(), item -> {
-                vo.setItemCode(item.getCode());
-                vo.setItemName(item.getName());
-                vo.setItemSpecification(item.getSpecification());
-            });
+            MapUtils.findAndThen(userMap, vo.getOwnerUserId(), user -> vo.setOwnerUserName(user.getNickname()));
+            MapUtils.findAndThen(itemMap, vo.getItemId(), item -> { vo.setItemCode(item.getCode()); vo.setItemName(item.getName()); });
         });
     }
 
@@ -243,6 +284,7 @@
         respVO.setPallets(BeanUtils.toBean(palletService.getPalletListByBatchId(batch.getId()), MesProPalletRespVO.class));
         // 琚嬬爜鍒楄〃锛堝皝椤� 2000锛宐agTotal 璁板綍鎬绘暟锛�
         List<MesProBagCodeDO> bags = bagCodeService.getBagCodeListByBatchId(batch.getId());
+        respVO.setPendingBagQuantity((int) bags.stream().filter(b -> !Integer.valueOf(30).equals(b.getStatus()) && b.getInboundEventId() == null).count());
         respVO.setBagTotal(bags.size());
         if (bags.size() > QrCodeImageUtils.MAX_QR_EXPORT) {
             bags = bags.subList(0, QrCodeImageUtils.MAX_QR_EXPORT);

--
Gitblit v1.9.3