From 40e4ce58f634598fafd205dfe54cb90e8b63b6f2 Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期五, 20 六月 2025 17:30:15 +0800
Subject: [PATCH] 生产加工

---
 main-business/src/main/java/com/ruoyi/business/service/impl/ProductionMasterServiceImpl.java |  249 ++++++++++++++++++++++++++-----------------------
 1 files changed, 133 insertions(+), 116 deletions(-)

diff --git a/main-business/src/main/java/com/ruoyi/business/service/impl/ProductionMasterServiceImpl.java b/main-business/src/main/java/com/ruoyi/business/service/impl/ProductionMasterServiceImpl.java
index ff10b10..4d3b647 100644
--- a/main-business/src/main/java/com/ruoyi/business/service/impl/ProductionMasterServiceImpl.java
+++ b/main-business/src/main/java/com/ruoyi/business/service/impl/ProductionMasterServiceImpl.java
@@ -19,7 +19,6 @@
 
 import java.math.BigDecimal;
 import java.util.*;
-import java.util.function.Function;
 import java.util.stream.Collectors;
 
 /**
@@ -116,140 +115,158 @@
 
     @Override
     @Transactional
-    public int addOrEditPM(ProductionMasterDto productionMasterDto) {
-        // 1. 鐩存帴璁$畻鑱氬悎鍊硷紝閬垮厤鍒涘缓澶氫綑鍒楄〃
-        BigDecimal totalPurchasePrice = BigDecimal.ZERO;
-        BigDecimal totalLaborCost = BigDecimal.ZERO;
-        BigDecimal totalEnergyConsumptionCost = BigDecimal.ZERO;
-        BigDecimal totalTotalCost = BigDecimal.ZERO;
-        BigDecimal totalEquipmentDepreciation = BigDecimal.ZERO;
-        BigDecimal totalProductionQuantity = BigDecimal.ZERO;
+    public int addOrEditPM(ProductionMasterDto dto) {
+        Long masterId = dto.getId();
 
-        for (Production production : productionMasterDto.getProductionList()) {
-            totalPurchasePrice = totalPurchasePrice.add(production.getPurchasePrice());
-            totalLaborCost = totalLaborCost.add(production.getLaborCost());
-            totalEnergyConsumptionCost = totalEnergyConsumptionCost.add(production.getEnergyConsumptionCost());
-            totalTotalCost = totalTotalCost.add(production.getTotalCost());
-            totalEquipmentDepreciation = totalEquipmentDepreciation.add(production.getEquipmentDepreciation());
-            totalProductionQuantity = production.getProductionQuantity().add(totalProductionQuantity);
+        // 缂栬緫鍦烘櫙锛氬洖婊氭棫搴撳瓨骞跺垹闄ゆ棫璁板綍
+        if (masterId != null) {
+            rollbackOldInventory(masterId);
+            deleteChildRecords(masterId);
         }
-        //鐓ょ瀛楁
-        List<Long> coalIds = productionMasterDto.getProductionList().stream()
-                .map(Production::getCoalId)
-                .collect(Collectors.toList());
 
-        List<CoalInfo> coalInfos = coalInfoMapper.selectList(new LambdaQueryWrapper<CoalInfo>().in(CoalInfo::getId, coalIds));
+        // 鏍¢獙浣跨敤閲忓苟鍑忓皯搴撳瓨
+        validateAndReduceInventory(dto.getProductionInventoryList());
 
-        // 2. 鍒涘缓涓昏〃瀵硅薄
-        ProductionMaster productionMaster = new ProductionMaster();
-        productionMaster.setProductionQuantity(totalProductionQuantity);
-        productionMaster.setTotalCost(totalTotalCost);
-        productionMaster.setEquipmentDepreciation(totalEquipmentDepreciation);
-        productionMaster.setEnergyConsumptionCost(totalEnergyConsumptionCost);
-        productionMaster.setLaborCost(totalLaborCost);
-        productionMaster.setCoal(coalInfos.stream().map(CoalInfo::getCoal).collect(Collectors.joining(",")));
-        productionMaster.setCoalId(coalIds.stream().map(String::valueOf).collect(Collectors.joining(",")));
+        // 鏋勯�犱富琛ㄥ疄浣撳璞�
+        ProductionMaster master = buildProductionMaster(dto);
 
-        Long masterId = productionMasterDto.getId();
-        productionMaster.setId(masterId);
-
-        // 3. 缁熶竴瀛愯〃澶勭悊閫昏緫
+        // 鎻掑叆鎴栨洿鏂颁富琛�
         if (masterId == null) {
-            productionMasterMapper.insert(productionMaster);
-            masterId = productionMaster.getId(); // 鑾峰彇鏂扮敓鎴愮殑ID
+            productionMasterMapper.insert(master);
+            masterId = master.getId();
         } else {
-            // 鍒犻櫎鍏宠仈瀛愯〃鏁版嵁
-            productionMapper.delete(new LambdaQueryWrapper<Production>()
-                    .eq(Production::getProductionMasterId, masterId));
-
-            productionInventoryMapper.delete(new LambdaQueryWrapper<ProductionInventory>()
-                    .eq(ProductionInventory::getProductionMasterId, masterId));
-
-            productionMasterMapper.updateById(productionMaster);
+            master.setId(masterId);
+            productionMasterMapper.updateById(master);
         }
-        //搴撳瓨鏇存柊
-        for (ProductionInventory productionInventory : productionMasterDto.getProductionInventoryList()) {
-            OfficialInventory officialInventory = officialInventoryMapper.selectById(productionInventory.getOfficialId());
-            BigDecimal subtract = officialInventory.getInventoryQuantity().subtract(new BigDecimal(productionInventory.getUsedQuantity()));
-            if (subtract.compareTo(BigDecimal.ZERO) < 0) {
-                throw new BaseException("搴撳瓨涓嶈冻");
-            }
-            officialInventory.setInventoryQuantity(subtract);
-            officialInventoryMapper.updateById(officialInventory);
-        }
-        
-        // 4. 鎵归噺鎻掑叆瀛愯〃鏁版嵁
-        batchInsertProductions(masterId, productionMasterDto.getProductionList());
-        batchInsertInventories(masterId, productionMasterDto.getProductionInventoryList());
 
-        //5. 鎻掑叆鍒板緟鍏ュ簱
-        for (Production production : productionMasterDto.getProductionList()) {
-            PendingInventory pendingInventory = new PendingInventory();
-            pendingInventory.setCoalId(production.getCoalId());
-            pendingInventory.setInventoryQuantity(production.getProductionQuantity());
-            pendingInventory.setSupplierName("鐢熶骇鍔犲伐鍏ュ簱");
-            pendingInventory.setTotalPriceIncludingTax(production.getTotalCost());
-            pendingInventory.setPriceIncludingTax(production.getPurchasePrice());
-        }
+        // 鎵归噺鎻掑叆鐢熶骇璁板綍涓庡簱瀛樿褰�
+        batchInsertProductions(masterId, dto.getProductionList());
+        batchInsertInventories(masterId, dto.getProductionInventoryList());
+
+        // 鎻掑叆寰呭叆搴撴暟鎹�
+        insertPendingInventory(dto.getProductionList());
+
         return 1;
     }
 
-    // 鎵归噺鎻掑叆鐢熶骇鏁版嵁
-    private void batchInsertProductions(Long masterId, List<Production> productions) {
-        if (productions.isEmpty()) {
-            return;
-        }
-        // 1. 鏀堕泦鎵�鏈夐渶瑕佹煡璇㈢殑coalId
-        List<Long> coalIds = productions.stream()
-                .map(Production::getCoalId)
-                .filter(Objects::nonNull)
-                .distinct()
-                .collect(Collectors.toList());
+    /**
+     * 鍥炴粴鏃х殑搴撳瓨鏁版嵁锛堝皢搴撳瓨鏁伴噺杩樺師锛�
+     */
+    private void rollbackOldInventory(Long masterId) {
+        List<ProductionInventory> oldInventories = productionInventoryMapper.selectList(
+                new LambdaQueryWrapper<ProductionInventory>().eq(ProductionInventory::getProductionMasterId, masterId));
 
-        // 2. 鎵归噺鏌ヨcoalInfo鏁版嵁
-        Map<Long, CoalInfo> coalInfoMap = coalIds.isEmpty() ?
-                Collections.emptyMap() :
-                coalInfoMapper.selectList(new LambdaQueryWrapper<CoalInfo>().in(CoalInfo::getId, coalIds))
-                        .stream()
-                        .collect(Collectors.toMap(CoalInfo::getId, Function.identity()));
-        if (coalInfoMap.isEmpty()){
-            throw new BaseException("鐓ょ淇℃伅涓嶅瓨鍦�");
-        }
-
-        // 3. 鍑嗗鎵归噺鎻掑叆鏁版嵁
-        List<Production> batchInsertList = productions.stream()
-                .map(production -> {
-                    Production p = new Production(); // 鍒涘缓鏂板璞¢伩鍏嶅壇浣滅敤
-                    BeanUtils.copyProperties(production, p);
-                    // 澶嶅埗蹇呰瀛楁
-                    p.setProductionMasterId(masterId);
-                    p.setCoalId(production.getCoalId());
-                    return p;
-                })
-                .collect(Collectors.toList());
-        if (!batchInsertList.isEmpty()) {
-            for (Production production : batchInsertList) {
-                production.setId(null);
-                productionMapper.insert(production);
+        for (ProductionInventory oldInv : oldInventories) {
+            OfficialInventory inv = officialInventoryMapper.selectById(oldInv.getOfficialId());
+            if (inv != null) {
+                inv.setInventoryQuantity(inv.getInventoryQuantity().add(new BigDecimal(oldInv.getUsedQuantity())));
+                officialInventoryMapper.updateById(inv);
             }
         }
     }
 
-    // 鎵归噺鎻掑叆搴撳瓨鏁版嵁
-    private void batchInsertInventories(Long masterId, List<ProductionInventory> inventories) {
-        List<ProductionInventory> insertList = inventories.stream()
-                .peek(inv -> {
-                    inv.setId(null);
-                    inv.setProductionMasterId(masterId);
-                })
-                .collect(Collectors.toList());
+    /**
+     * 鍒犻櫎鏃х殑瀛愯〃鏁版嵁鍙婂緟鍏ュ簱鏁版嵁
+     */
+    private void deleteChildRecords(Long masterId) {
+        productionMapper.delete(new LambdaQueryWrapper<Production>().eq(Production::getProductionMasterId, masterId));
+        productionInventoryMapper.delete(new LambdaQueryWrapper<ProductionInventory>().eq(ProductionInventory::getProductionMasterId, masterId));
+        pendingInventoryMapper.delete(new LambdaQueryWrapper<PendingInventory>().eq(PendingInventory::getMasterId, masterId));
+    }
 
-        if (!insertList.isEmpty()) {
-            for (ProductionInventory inventory : inventories) {
-                inventory.setId(null);
-                inventory.setProductionMasterId(masterId);
-                productionInventoryMapper.insert(inventory);
+    /**
+     * 鏍¢獙姣忔潯浣跨敤閲忔槸鍚﹁冻澶燂紝骞跺噺灏戞寮忓簱瀛樻暟閲�
+     */
+    private void validateAndReduceInventory(List<ProductionInventory> inventoryList) {
+        for (ProductionInventory inv : inventoryList) {
+            OfficialInventory official = officialInventoryMapper.selectById(inv.getOfficialId());
+            BigDecimal used = new BigDecimal(inv.getUsedQuantity());
+            if (official.getInventoryQuantity().compareTo(used) < 0) {
+                throw new BaseException("搴撳瓨涓嶈冻");
             }
+            official.setInventoryQuantity(official.getInventoryQuantity().subtract(used));
+            officialInventoryMapper.updateById(official);
+        }
+    }
+
+    /**
+     * 鏋勯�犱富琛ㄥ璞″苟鑱氬悎瀛楁鍊硷紙濡傛�绘垚鏈�佺叅绉嶇瓑锛�
+     */
+    private ProductionMaster buildProductionMaster(ProductionMasterDto dto) {
+        BigDecimal totalPurchase = BigDecimal.ZERO;
+        BigDecimal totalLabor = BigDecimal.ZERO;
+        BigDecimal totalEnergy = BigDecimal.ZERO;
+        BigDecimal totalCost = BigDecimal.ZERO;
+        BigDecimal totalDepreciation = BigDecimal.ZERO;
+        BigDecimal totalQuantity = BigDecimal.ZERO;
+
+        List<Long> coalIds = new ArrayList<>();
+
+        for (Production p : dto.getProductionList()) {
+            totalPurchase = totalPurchase.add(p.getPurchasePrice());
+            totalLabor = totalLabor.add(p.getLaborCost());
+            totalEnergy = totalEnergy.add(p.getEnergyConsumptionCost());
+            totalCost = totalCost.add(p.getTotalCost());
+            totalDepreciation = totalDepreciation.add(p.getEquipmentDepreciation());
+            totalQuantity = totalQuantity.add(p.getProductionQuantity());
+            coalIds.add(p.getCoalId());
+        }
+
+        List<CoalInfo> coalInfos = coalInfoMapper.selectList(new LambdaQueryWrapper<CoalInfo>().in(CoalInfo::getId, coalIds));
+
+        ProductionMaster master = new ProductionMaster();
+        master.setProductionQuantity(totalQuantity);
+        master.setTotalCost(totalCost);
+        master.setLaborCost(totalLabor);
+        master.setEnergyConsumptionCost(totalEnergy);
+        master.setEquipmentDepreciation(totalDepreciation);
+        master.setCoal(coalInfos.stream().map(CoalInfo::getCoal).collect(Collectors.joining(",")));
+        master.setCoalId(coalIds.stream().map(String::valueOf).collect(Collectors.joining(",")));
+
+        return master;
+    }
+
+    /**
+     * 鎵归噺鎻掑叆鐢熶骇瀛愯〃鏁版嵁
+     */
+    private void batchInsertProductions(Long masterId, List<Production> list) {
+        if (list.isEmpty()) return;
+
+        for (Production p : list) {
+            Production copy = new Production();
+            BeanUtils.copyProperties(p, copy);
+            copy.setId(null);
+            copy.setProductionMasterId(masterId);
+            productionMapper.insert(copy);
+        }
+    }
+
+    /**
+     * 鎵归噺鎻掑叆搴撳瓨浣跨敤瀛愯〃鏁版嵁
+     */
+    private void batchInsertInventories(Long masterId, List<ProductionInventory> list) {
+        if (list.isEmpty()) return;
+
+        for (ProductionInventory p : list) {
+            p.setId(null);
+            p.setProductionMasterId(masterId);
+            productionInventoryMapper.insert(p);
+        }
+    }
+
+    /**
+     * 灏嗗姞宸ヤ骇鐢熺殑浜у搧璁板綍鍒板緟鍏ュ簱琛�
+     */
+    private void insertPendingInventory(List<Production> list) {
+        for (Production p : list) {
+            PendingInventory pending = new PendingInventory();
+            pending.setCoalId(p.getCoalId());
+            pending.setInventoryQuantity(p.getProductionQuantity());
+            pending.setSupplierName("鐢熶骇鍔犲伐鍏ュ簱");
+            pending.setTotalPriceIncludingTax(p.getTotalCost());
+            pending.setPriceIncludingTax(p.getPurchasePrice());
+            pending.setPriceIncludingTax(p.getPurchasePrice());
+            pendingInventoryMapper.insert(pending);
         }
     }
 

--
Gitblit v1.9.3