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/service/pro/bagcode/MesProBagCodeServiceImpl.java |   58 +++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 55 insertions(+), 3 deletions(-)

diff --git a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/bagcode/MesProBagCodeServiceImpl.java b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/bagcode/MesProBagCodeServiceImpl.java
index d7f9b24..2310eff 100644
--- a/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/bagcode/MesProBagCodeServiceImpl.java
+++ b/yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/service/pro/bagcode/MesProBagCodeServiceImpl.java
@@ -89,6 +89,8 @@
 
     @Resource
     private MesCodeRedisDAO codeRedisDao;
+    @Resource
+    private cn.iocoder.yudao.module.mes.service.trace.integrity.MesTraceIntegrityService integrityService;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -122,8 +124,10 @@
             for (int i = 0; i < palletCount; i++) {
                 pallets.add(new MesProPalletDO()
                         .setTempCode(codeRedisDao.generate())
+                        .setBusinessCode(codeRedisDao.generate())
                         .setBagCount(perPallet)
                         .setAutoGenerated(true)
+                        .setItemId(item.getId())
                         .setStatus(MesProPalletStatusEnum.IN_USE.getStatus()));
             }
             palletMapper.insertBatch(pallets);
@@ -137,6 +141,7 @@
             bagCodes.add(new MesProBagCodeDO()
                     .setUuid(uuid)
                     .setCode(codeRedisDao.generate())
+                    .setBusinessCode(codeRedisDao.generate())
                     .setBagNo(bagNo)
                     .setItemId(item.getId())
                     .setItemCode(item.getCode())
@@ -169,6 +174,7 @@
             bagCodes.add(new MesProBagCodeDO()
                     .setUuid(uuid)
                     .setCode(codeRedisDao.generate())
+                    .setBusinessCode(codeRedisDao.generate())
                     .setBagNo(bagNo)
                     .setItemId(item.getId())
                     .setItemCode(item.getCode())
@@ -238,20 +244,31 @@
         }
         // 褰曞叆鐨勮鐮佸繀椤诲睘浜庡悓涓�鍝佺锛屼笖涓庢壒娆″凡鏈夎鐮佸搧绉嶄竴鑷达紙鍘嗗彶绌虹櫧琚嬬爜 item 涓虹┖涓嶅弬涓庢牎楠岋級
         Set<Long> itemIds = new LinkedHashSet<>();
+        if (batch.getItemId() != null) itemIds.add(batch.getItemId());
         bagCodes.stream().map(MesProBagCodeDO::getItemId).filter(Objects::nonNull).forEach(itemIds::add);
         itemIds.addAll(bagCodeMapper.selectDistinctItemIdsByBatchId(batch.getId()));
+        for (Long palletId : bagCodes.stream().map(MesProBagCodeDO::getPalletId).filter(Objects::nonNull).distinct().sorted().toList()) {
+            MesProPalletDO pallet = palletMapper.selectByIdForUpdate(palletId);
+            if (pallet == null) throw exception(PRO_PALLET_NOT_EXISTS);
+            if (pallet.getBatchId() != null && !pallet.getBatchId().equals(batch.getId()))
+                throw exception(PRO_BAG_CODE_BIND_PALLET_MISMATCH);
+            Long palletItem = cn.iocoder.yudao.module.mes.service.pro.pallet.MesProPalletItemResolver.resolve(pallet, bagCodeMapper.selectListByPalletId(palletId));
+            if (palletItem != null) itemIds.add(palletItem);
+        }
         if (itemIds.size() > 1) {
             throw exception(PRO_BAG_CODE_BIND_BATCH_ITEM_MISMATCH);
         }
+        if (!itemIds.isEmpty()) batch.setItemId(itemIds.iterator().next());
         if (bagCodeMapper.bindBatchByIds(ids, batch.getId(), batch.getCode(),
                 List.of(MesProBagCodeStatusEnum.GENERATED.getStatus(),
                         MesProBagCodeStatusEnum.PRINTED.getStatus())) != ids.size()) {
             throw exception(PRO_BAG_CODE_STATUS_NOT_GENERATED);
         }
-        batchMapper.updateById(new MesProBatchDO().setId(batch.getId())
+        batchMapper.updateById(new MesProBatchDO().setId(batch.getId()).setItemId(batch.getItemId())
                 .setBagQuantity(bagCodeMapper.selectEffectiveCountByBatchId(batch.getId())));
         // 绾ц仈锛氭湰缁勮鐮佺粦瀹氱殑鏈叆鎵规鎵樼洏锛屽垎閰嶆寮忔墭鐩樼爜骞剁粦瀹氭壒娆�
         bindPalletsToBatch(bagCodes, batch);
+        integrityService.recordBatchIntegrity(batch.getId());
     }
 
     /**
@@ -273,6 +290,7 @@
             }
             String palletCode = codeRedisDao.generate();
             palletMapper.updateById(new MesProPalletDO().setId(palletId)
+                    .setItemId(batch.getItemId())
                     .setCode(palletCode)
                     .setBatchId(batch.getId())
                     .setBatchCode(batch.getCode())
@@ -345,6 +363,7 @@
             MesProBagCodeDO newBagCode = new MesProBagCodeDO()
                     .setUuid(replaceUuid)
                     .setCode(codeRedisDao.generate())
+                    .setBusinessCode(codeRedisDao.generate())
                     .setBagNo(newBagNo)
                     .setBatchId(batch.getId())
                     .setBatchCode(batch.getCode())
@@ -447,9 +466,16 @@
         }
         // 2. 閿佸畾骞舵牎楠岃鐮�
         List<MesProBagCodeDO> bagCodes = validateBagCodesForBind(bindReqVO.getIds(), pallet, true);
+        Long itemId = cn.iocoder.yudao.module.mes.service.pro.pallet.MesProPalletItemResolver.resolve(pallet,
+                bagCodeMapper.selectListByPalletId(pallet.getId()));
+        for (MesProBagCodeDO bag : bagCodes) {
+            if (bag.getItemId() == null) throw new cn.iocoder.yudao.framework.common.exception.ServiceException(400, "琚嬬爜缂哄皯鍝佺");
+            itemId = cn.iocoder.yudao.module.mes.service.pro.pallet.MesProPalletItemResolver.merge(itemId, bag.getItemId());
+        }
+        palletMapper.updateById(new MesProPalletDO().setId(pallet.getId()).setItemId(itemId));
         // 3. 缁戝畾琚嬬爜骞剁疮鍔犳墭鐩樺凡瑁呰鏁帮紙鎵樼洏鏃犲閲忎笂闄愶級
         int currentCount = pallet.getBagCount() == null ? 0 : pallet.getBagCount();
-        bagCodeMapper.bindPalletByIds(bindReqVO.getIds(), pallet.getId(), pallet.getCode());
+        bagCodeMapper.bindPalletByIds(bindReqVO.getIds(), pallet.getId(), StrUtil.isNotBlank(pallet.getCode()) ? pallet.getCode() : pallet.getTempCode());
         palletService.updatePalletBagCount(pallet.getId(), currentCount + bagCodes.size());
     }
 
@@ -501,6 +527,32 @@
     @Override
     public MesProBagCodeDO getBagCodeByCode(String code) {
         return bagCodeMapper.selectByCode(code);
+    }
+
+    @Override
+    public MesProBagCodeDO getBagCodeByScanCode(String code) {
+        return bagCodeMapper.selectByScanCode(code);
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public int generateBusinessCodes(List<Long> ids) {
+        if (CollUtil.isEmpty(ids) || ids.size() > 200 || ids.stream().anyMatch(Objects::isNull)) {
+            throw new cn.iocoder.yudao.framework.common.exception.ServiceException(400, "璇烽�夋嫨 1 鑷� 200 鏉¤鐮�");
+        }
+        List<Long> uniqueIds = ids.stream().distinct().sorted().toList();
+        List<MesProBagCodeDO> bags = bagCodeMapper.selectListByIdsForUpdate(uniqueIds);
+        if (bags.size() != uniqueIds.size()) throw exception(PRO_BAG_CODE_NOT_EXISTS);
+        if (bags.stream().anyMatch(bag -> MesProBagCodeStatusEnum.CANCELED.getStatus().equals(bag.getStatus()))) {
+            throw new cn.iocoder.yudao.framework.common.exception.ServiceException(400, "宸蹭綔搴熻鐮佷笉鍙ˉ鐢熸垚涓氬姟鐮�");
+        }
+        int count = 0;
+        for (MesProBagCodeDO bag : bags) {
+            if (StrUtil.isNotBlank(bag.getBusinessCode())) continue;
+            bagCodeMapper.updateById(new MesProBagCodeDO().setId(bag.getId()).setBusinessCode(codeRedisDao.generate()));
+            count++;
+        }
+        return count;
     }
 
     @Override
@@ -564,7 +616,7 @@
             throw exception(PRO_BAG_CODE_NOT_EXISTS);
         }
         for (MesProBagCodeDO bagCode : bagCodes) {
-            if (!pallet.getBatchId().equals(bagCode.getBatchId())) {
+            if (!Objects.equals(pallet.getBatchId(), bagCode.getBatchId())) {
                 throw exception(PRO_BAG_CODE_BIND_PALLET_MISMATCH);
             }
             if (bagCode.getPalletId() != null) {

--
Gitblit v1.9.3