From 4858d6a68446a5153749eca8fae8ae099ac879a2 Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期一, 25 五月 2026 14:11:49 +0800
Subject: [PATCH] fix:1.销售从库存选

---
 src/main/java/com/ruoyi/stock/service/impl/StockInventoryServiceImpl.java |  607 ++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 516 insertions(+), 91 deletions(-)

diff --git a/src/main/java/com/ruoyi/stock/service/impl/StockInventoryServiceImpl.java b/src/main/java/com/ruoyi/stock/service/impl/StockInventoryServiceImpl.java
index b660824..ceb7b20 100644
--- a/src/main/java/com/ruoyi/stock/service/impl/StockInventoryServiceImpl.java
+++ b/src/main/java/com/ruoyi/stock/service/impl/StockInventoryServiceImpl.java
@@ -1,90 +1,274 @@
 package com.ruoyi.stock.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.ruoyi.common.enums.StockQualifiedRecordTypeEnum;
+import com.ruoyi.approve.service.impl.ApproveProcessServiceImpl;
+import com.ruoyi.approve.vo.ApproveProcessVO;
+import com.ruoyi.basic.mapper.ProductMapper;
+import com.ruoyi.basic.mapper.ProductModelMapper;
+import com.ruoyi.basic.pojo.Product;
+import com.ruoyi.basic.pojo.ProductModel;
+import com.ruoyi.common.enums.StockInQualifiedRecordTypeEnum;
+import com.ruoyi.common.enums.StockInUnQualifiedRecordTypeEnum;
+import com.ruoyi.common.utils.SecurityUtils;
+import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
+import com.ruoyi.framework.security.LoginUser;
 import com.ruoyi.framework.web.domain.R;
 import com.ruoyi.sales.mapper.SalesLedgerProductMapper;
 import com.ruoyi.sales.pojo.SalesLedgerProduct;
-import com.ruoyi.stock.dto.StockInRecordDto;
-import com.ruoyi.stock.dto.StockInventoryDto;
-import com.ruoyi.stock.dto.StockOutRecordDto;
+import com.ruoyi.stock.dto.*;
+import com.ruoyi.stock.execl.FinishedProductInventoryExportData;
+import com.ruoyi.stock.execl.NonFinishedProductInventoryExportData;
 import com.ruoyi.stock.execl.StockInventoryExportData;
 import com.ruoyi.stock.mapper.StockInventoryMapper;
+import com.ruoyi.stock.pojo.StockInRecord;
 import com.ruoyi.stock.pojo.StockInventory;
 import com.ruoyi.stock.service.StockInRecordService;
 import com.ruoyi.stock.service.StockInventoryService;
 import com.ruoyi.stock.service.StockOutRecordService;
-import lombok.AllArgsConstructor;
+import com.ruoyi.stock.service.StockUninventoryService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletResponse;
 import java.math.BigDecimal;
-import java.util.ArrayList;
-import java.util.List;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.util.*;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
-/**
- * <p>
- * 搴撳瓨琛� 鏈嶅姟瀹炵幇绫�
- * </p>
- *
- * @author 鑺杞欢锛堟睙鑻忥級鏈夐檺鍏徃
- * @since 2026-01-21 04:16:36
- */
 @Service
-@AllArgsConstructor
 public class StockInventoryServiceImpl extends ServiceImpl<StockInventoryMapper, StockInventory> implements StockInventoryService {
 
-    private  StockInventoryMapper stockInventoryMapper;
+    @Autowired
+    private StockInventoryMapper stockInventoryMapper;
+    @Autowired
     private StockInRecordService stockInRecordService;
+    @Autowired
     private StockOutRecordService stockOutRecordService;
+    @Autowired
     private SalesLedgerProductMapper salesLedgerProductMapper;
+    @Autowired
+    private ApproveProcessServiceImpl approveProcessService;
+    @Autowired
+    private ProductModelMapper productModelMapper;
+    @Autowired
+    private ProductMapper productMapper;
+    @Autowired
+    private StockUninventoryService stockUninventoryService;
+
     @Override
     public IPage<StockInventoryDto> pagestockInventory(Page page, StockInventoryDto stockInventoryDto) {
         return stockInventoryMapper.pagestockInventory(page, stockInventoryDto);
     }
 
-    //鍏ュ簱璋冪敤
+    @Override
+    public List<FinishedProductTreeDto> finishedProductList(StockInventoryDto stockInventoryDto) {
+        // 鏌ヨ搴撳瓨鏁版嵁
+        List<FinishedProductTreeDto> dataList = stockInventoryMapper.selectFinishedProductList(stockInventoryDto);
+        if (dataList.isEmpty()) {
+            return new ArrayList<>();
+        }
+
+        // 鎸変骇鍝両D鍒嗙粍锛屾瀯寤烘爲褰㈢粨鏋�
+        Map<Long, FinishedProductTreeDto> productMap = new LinkedHashMap<>();
+        for (FinishedProductTreeDto data : dataList) {
+            Long productId = data.getProductId();
+            if (!productMap.containsKey(productId)) {
+                // 鍒涘缓浜у搧澶х被鑺傜偣
+                FinishedProductTreeDto productNode = new FinishedProductTreeDto();
+                productNode.setProductId(productId);
+                productNode.setProductName(data.getProductName());
+                productNode.setLabel(data.getLabel());
+                productNode.setChildren(new ArrayList<>());
+                productMap.put(productId, productNode);
+            }
+            // 娣诲姞搴撳瓨鍙跺瓙鑺傜偣
+            FinishedProductTreeDto leafNode = new FinishedProductTreeDto();
+            leafNode.setStockId(data.getStockId());
+            leafNode.setProductModelId(data.getProductModelId());
+            leafNode.setModel(data.getModel());
+            leafNode.setProcessCategory(data.getProcessCategory());
+            leafNode.setVoltage(data.getVoltage());
+            leafNode.setMaterialCode(data.getMaterialCode());
+            leafNode.setUnit(data.getUnit());
+            leafNode.setQualitity(data.getQualitity());
+            productMap.get(productId).getChildren().add(leafNode);
+        }
+
+        return new ArrayList<>(productMap.values());
+    }
+
+    @Override
+    public IPage<StockInventoryDto> pageListCombinedStockInventory(Page page, StockInventoryDto stockInventoryDto) {
+        return stockInventoryMapper.pageListCombinedStockInventory(page, stockInventoryDto);
+    }
+
+    /**
+     * 鍚堟牸鍏ュ簱锛氬厛鐢熸垚鍏ュ簱璁板綍锛屽啀璧板鎵规祦銆�
+     */
     @Override
     @Transactional(rollbackFor = Exception.class)
     public Boolean addstockInventory(StockInventoryDto stockInventoryDto) {
-        //鏂板鍏ュ簱璁板綍鍐嶆坊鍔犲簱瀛�
+        List<StockInventory> stockInventoryList = stockInventoryMapper.selectList(null);
         StockInRecordDto stockInRecordDto = new StockInRecordDto();
         stockInRecordDto.setRecordId(stockInventoryDto.getRecordId());
         stockInRecordDto.setRecordType(stockInventoryDto.getRecordType());
         stockInRecordDto.setStockInNum(stockInventoryDto.getQualitity());
         stockInRecordDto.setProductModelId(stockInventoryDto.getProductModelId());
+        stockInRecordDto.setRemark(stockInventoryDto.getRemark());
+        stockInRecordDto.setWarnNum(stockInventoryDto.getWarnNum());
+        stockInRecordDto.setLockedQuantity(stockInventoryDto.getLockedQuantity());
+        stockInRecordDto.setProcessCategory(normalizeDimension(stockInventoryDto.getProcessCategory()));
+        stockInRecordDto.setVoltage(normalizeDimension(stockInventoryDto.getVoltage()));
+        stockInRecordDto.setApproveStatus(0);
         stockInRecordDto.setType("0");
-        stockInRecordService.add(stockInRecordDto);
-        //鍐嶈繘琛屾柊澧炲簱瀛樻暟閲忓簱瀛�
-        //鍏堟煡璇㈠簱瀛樿〃涓殑浜у搧鏄惁瀛樺湪锛屼笉瀛樺湪鏂板锛屽瓨鍦ㄦ洿鏂�
-        StockInventory oldStockInventory = stockInventoryMapper.selectOne(new QueryWrapper<StockInventory>().lambda().eq(StockInventory::getProductModelId, stockInventoryDto.getProductModelId()));
-        if (ObjectUtils.isEmpty(oldStockInventory)) {
-            StockInventory newStockInventory = new StockInventory();
-            newStockInventory.setProductModelId(stockInventoryDto.getProductModelId());
-            newStockInventory.setQualitity(stockInventoryDto.getQualitity());
-            newStockInventory.setVersion(1);
-            newStockInventory.setRemark(stockInventoryDto.getRemark());
-            newStockInventory.setLockedQuantity(stockInventoryDto.getLockedQuantity());
-            newStockInventory.setWarnNum(stockInventoryDto.getWarnNum());
-            stockInventoryMapper.insert(newStockInventory);
-        }else {
-             stockInventoryMapper.updateAddStockInventory(stockInventoryDto);
+        if (StringUtils.isBlank(stockInventoryDto.getBatchNo())) {
+            String batchNo;
+            LocalDate now = LocalDate.now();
+            String monthFlag = now.format(DateTimeFormatter.ofPattern("MM"));
+            int maxSeq = getCurrentMonthMaxSeq(stockInventoryDto, monthFlag, stockInventoryList);
+            int newSeq = maxSeq + 1;
+            String seqStr = String.format("%03d", newSeq);
+            ProductModel productModel = productModelMapper.selectById(stockInventoryDto.getProductModelId());
+            batchNo = stockInventoryDto.getMaterialCode() + productModel.getModel() + "P" + monthFlag + seqStr;
+            stockInRecordDto.setBatchNo(batchNo);
+        } else {
+            stockInRecordDto.setBatchNo(stockInventoryDto.getBatchNo());
+        }
+        Long id = stockInRecordService.add(stockInRecordDto);
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        if (id != null) {
+            try {
+                addApproveByPurchase(loginUser, stockInRecordDto, id);
+            } catch (Exception e) {
+                throw new RuntimeException(e.getMessage());
+            }
         }
         return true;
     }
 
-    //鍑哄簱璋冪敤
+    private static int getCurrentMonthMaxSeq(StockInventoryDto dto, String monthFlag, List<StockInventory> existingList) {
+        int maxSeq = 0;
+        String prefix = dto.getMaterialCode() + dto.getProductModelName() + "P" + monthFlag;
+        Pattern pattern = Pattern.compile(Pattern.quote(prefix) + "(\\d{3})");
+        for (StockInventory item : existingList) {
+            String batchNo = item.getBatchNo();
+            if (batchNo == null) {
+                continue;
+            }
+            Matcher matcher = pattern.matcher(batchNo);
+            if (matcher.find()) {
+                int seq = Integer.parseInt(matcher.group(1));
+                if (seq > maxSeq) {
+                    maxSeq = seq;
+                }
+            }
+        }
+        return maxSeq;
+    }
+
+    @Override
+    public void addApproveByPurchase(LoginUser loginUser, StockInRecordDto stockInRecordDto, Long id) throws Exception {
+        ApproveProcessVO approveProcessVO = new ApproveProcessVO();
+        approveProcessVO.setApproveType(9);
+        approveProcessVO.setApproveDeptId(loginUser.getCurrentDeptId());
+        approveProcessVO.setApproveReason(stockInRecordDto.getInboundBatches());
+        approveProcessVO.setApproveUserIds(String.valueOf(1));
+        approveProcessVO.setApproveUser(loginUser.getUserId());
+        approveProcessVO.setApproveTime(LocalDate.now().toString());
+        approveProcessVO.setInventoryReview(false);
+        approveProcessVO.setStorageType("鍚堟牸鍏ュ簱");
+        approveProcessVO.setRecordId(id);
+        approveProcessService.addApprove(approveProcessVO);
+    }
+
+    /**
+     * 鎸夊簱瀛樺敮涓�閿悎骞跺簱瀛樸��
+     * 鎴愬搧浣跨敤 product_model_id + processCategory + voltage锛涘叾浠栧叆搴撳彧鎸� product_model_id銆�
+     */
+    @Override
+    public void updateOrCreateStockInventory(StockInRecord stockInRecord) {
+        String processCategory = normalizeDimension(stockInRecord.getProcessCategory());
+        String voltage = normalizeDimension(stockInRecord.getVoltage());
+        StockInventory oldStockInventory = findInventoryForMerge(
+                stockInRecord.getProductModelId(),
+                processCategory,
+                voltage
+        );
+
+        if (ObjectUtils.isEmpty(oldStockInventory)) {
+            StockInventory newStockInventory = new StockInventory();
+            newStockInventory.setProductModelId(stockInRecord.getProductModelId());
+            newStockInventory.setQualitity(defaultDecimal(stockInRecord.getStockInNum()));
+            newStockInventory.setVersion(1);
+            newStockInventory.setRemark(stockInRecord.getRemark());
+            newStockInventory.setLockedQuantity(defaultDecimal(stockInRecord.getLockedQuantity()));
+            newStockInventory.setWarnNum(stockInRecord.getWarnNum());
+            newStockInventory.setProcessCategory(processCategory);
+            newStockInventory.setVoltage(voltage);
+            newStockInventory.setBatchNo(StringUtils.trimToEmpty(stockInRecord.getBatchNo()));
+            stockInventoryMapper.insert(newStockInventory);
+            return;
+        }
+
+        oldStockInventory.setQualitity(defaultDecimal(oldStockInventory.getQualitity()).add(defaultDecimal(stockInRecord.getStockInNum())));
+        oldStockInventory.setVersion(oldStockInventory.getVersion() == null ? 1 : oldStockInventory.getVersion() + 1);
+        if (StringUtils.isNotBlank(stockInRecord.getRemark())) {
+            oldStockInventory.setRemark(stockInRecord.getRemark());
+        }
+        if (stockInRecord.getWarnNum() != null) {
+            oldStockInventory.setWarnNum(stockInRecord.getWarnNum());
+        }
+        if (stockInRecord.getLockedQuantity() != null) {
+            oldStockInventory.setLockedQuantity(defaultDecimal(oldStockInventory.getLockedQuantity()).add(stockInRecord.getLockedQuantity()));
+        }
+        oldStockInventory.setProcessCategory(processCategory);
+        oldStockInventory.setVoltage(voltage);
+        oldStockInventory.setUpdateTime(LocalDateTime.now());
+        stockInventoryMapper.updateById(oldStockInventory);
+    }
+
+    /**
+     * 涓嶅鏍稿叆搴擄紝鐩存帴鍐欏叆鍏ュ簱璁板綍鍜屽簱瀛樸��
+     */
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public Boolean addstockInventoryNoReview(StockInventoryDto stockInventoryDto) {
+        StockInRecordDto stockInRecordDto = new StockInRecordDto();
+        stockInRecordDto.setRecordId(stockInventoryDto.getRecordId());
+        stockInRecordDto.setRecordType(stockInventoryDto.getRecordType());
+        stockInRecordDto.setStockInNum(stockInventoryDto.getQualitity());
+        stockInRecordDto.setProductModelId(stockInventoryDto.getProductModelId());
+        stockInRecordDto.setRemark(stockInventoryDto.getRemark());
+        stockInRecordDto.setWarnNum(stockInventoryDto.getWarnNum());
+        stockInRecordDto.setLockedQuantity(stockInventoryDto.getLockedQuantity());
+        stockInRecordDto.setProcessCategory(normalizeDimension(stockInventoryDto.getProcessCategory()));
+        stockInRecordDto.setVoltage(normalizeDimension(stockInventoryDto.getVoltage()));
+        stockInRecordDto.setBatchNo(stockInventoryDto.getBatchNo());
+        stockInRecordDto.setType("0");
+        stockInRecordService.add(stockInRecordDto);
+        updateOrCreateStockInventory(toStockInRecord(stockInRecordDto));
+        return true;
+    }
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public Boolean subtractStockInventory(StockInventoryDto stockInventoryDto) {
-        //  鏂板鍑哄簱璁板綍
+        if (stockInventoryDto.getQualitity() != null
+                && stockInventoryDto.getQualitity().compareTo(BigDecimal.ZERO) == 0) {
+            return true;
+        }
+
         StockOutRecordDto stockOutRecordDto = new StockOutRecordDto();
         stockOutRecordDto.setRecordId(stockInventoryDto.getRecordId());
         stockOutRecordDto.setRecordType(stockInventoryDto.getRecordType());
@@ -92,107 +276,304 @@
         stockOutRecordDto.setProductModelId(stockInventoryDto.getProductModelId());
         stockOutRecordDto.setType("0");
         stockOutRecordService.add(stockOutRecordDto);
-        StockInventory oldStockInventory = stockInventoryMapper.selectOne(new QueryWrapper<StockInventory>().lambda().eq(StockInventory::getProductModelId, stockInventoryDto.getProductModelId()));
-        BigDecimal lockedQty = oldStockInventory.getLockedQuantity();
-        if (lockedQty == null) {
-            lockedQty = BigDecimal.ZERO;
-        }
-        if (stockInventoryDto.getQualitity().compareTo(oldStockInventory.getQualitity().subtract(lockedQty)) > 0) {
-            throw new RuntimeException("搴撳瓨涓嶈冻鏃犳硶鍑哄簱");
+
+        //閿�鍞嚭搴撴寜鐓т繚瀛樼殑搴撳瓨id
+        if (stockInventoryDto.getStockId() != null) {
+            StockInventory stockInventory = stockInventoryMapper.selectById(stockInventoryDto.getStockId());
+            if (ObjectUtils.isEmpty(stockInventory)) {
+                throw new RuntimeException("浜у搧搴撳瓨涓嶅瓨鍦�");
+            }
+            BigDecimal lockedQty = defaultDecimal(stockInventory.getLockedQuantity());
+            if (stockInventoryDto.getQualitity().compareTo(defaultDecimal(stockInventory.getQualitity()).subtract(lockedQty)) > 0) {
+                ProductModel productModel = productModelMapper.selectById(stockInventoryDto.getProductModelId());
+                Product product = productMapper.selectById(productModel.getProductId());
+                throw new RuntimeException(product.getProductName() + "/" + productModel.getModel() + "搴撳瓨涓嶈冻锛屾棤娉曞嚭搴�");
+            }
+            stockInventory.setQualitity(defaultDecimal(stockInventory.getQualitity()).subtract(stockInventoryDto.getQualitity()));
+            stockInventory.setVersion(stockInventory.getVersion() == null ? 1 : stockInventory.getVersion() + 1);
+            stockInventory.setUpdateTime(LocalDateTime.now());
+            stockInventoryMapper.updateById(stockInventory);
+            return true;
         }
 
+        if (StringUtils.isBlank(stockInventoryDto.getBatchNo()) && !usesDimensionIdentity(normalizeDimension(stockInventoryDto.getProcessCategory()), normalizeDimension(stockInventoryDto.getVoltage()))) {
+            List<StockInventory> stockInventories = stockInventoryMapper.selectList(new QueryWrapper<StockInventory>().lambda()
+                    .eq(StockInventory::getProductModelId, stockInventoryDto.getProductModelId())
+                    .orderByAsc(StockInventory::getId));
+            if (ObjectUtils.isEmpty(stockInventories)) {
+                throw new RuntimeException("浜у搧搴撳瓨涓嶅瓨鍦�");
+            }
+
+            BigDecimal remainingQty = stockInventoryDto.getQualitity() == null ? BigDecimal.ZERO : stockInventoryDto.getQualitity();
+            for (StockInventory stockInventory : stockInventories) {
+                BigDecimal lockedQty = defaultDecimal(stockInventory.getLockedQuantity());
+                BigDecimal availableQty = defaultDecimal(stockInventory.getQualitity()).subtract(lockedQty);
+                if (availableQty.compareTo(BigDecimal.ZERO) <= 0) {
+                    continue;
+                }
+
+                BigDecimal deductQty = remainingQty.min(availableQty);
+                stockInventory.setQualitity(defaultDecimal(stockInventory.getQualitity()).subtract(deductQty));
+                stockInventory.setVersion(stockInventory.getVersion() == null ? 1 : stockInventory.getVersion() + 1);
+                stockInventory.setUpdateTime(LocalDateTime.now());
+                stockInventoryMapper.updateById(stockInventory);
+
+                remainingQty = remainingQty.subtract(deductQty);
+                if (remainingQty.compareTo(BigDecimal.ZERO) <= 0) {
+                    return true;
+                }
+            }
+
+            ProductModel productModel = productModelMapper.selectById(stockInventoryDto.getProductModelId());
+            Product product = productMapper.selectById(productModel.getProductId());
+            throw new RuntimeException(product.getProductName() + "/" + productModel.getModel() + "搴撳瓨涓嶈冻锛屾棤娉曞嚭搴�");
+        }
+
+        StockInventory oldStockInventory = findInventoryForMerge(
+                stockInventoryDto.getProductModelId(),
+                stockInventoryDto.getProcessCategory(),
+                stockInventoryDto.getVoltage()
+        );
         if (ObjectUtils.isEmpty(oldStockInventory)) {
             throw new RuntimeException("浜у搧搴撳瓨涓嶅瓨鍦�");
-        } else {
-            stockInventoryMapper.updateSubtractStockInventory(stockInventoryDto);
         }
+        BigDecimal lockedQty = defaultDecimal(oldStockInventory.getLockedQuantity());
+        if (stockInventoryDto.getQualitity().compareTo(defaultDecimal(oldStockInventory.getQualitity()).subtract(lockedQty)) > 0) {
+            ProductModel productModel = productModelMapper.selectById(stockInventoryDto.getProductModelId());
+            Product product = productMapper.selectById(productModel.getProductId());
+            throw new RuntimeException(product.getProductName() + "/" + productModel.getModel() + "搴撳瓨涓嶈冻锛屾棤娉曞嚭搴�");
+        }
+
+        oldStockInventory.setQualitity(defaultDecimal(oldStockInventory.getQualitity()).subtract(stockInventoryDto.getQualitity()));
+        oldStockInventory.setVersion(oldStockInventory.getVersion() == null ? 1 : oldStockInventory.getVersion() + 1);
+        oldStockInventory.setUpdateTime(LocalDateTime.now());
+        stockInventoryMapper.updateById(oldStockInventory);
         return true;
     }
 
     @Override
-    public R importStockInventory(MultipartFile file) {
+    public R importStockInventory(MultipartFile file, Integer productType) {
         try {
-            // 鏌ヨ鎵�鏈夌殑浜у搧
             List<SalesLedgerProduct> salesLedgerProducts = salesLedgerProductMapper.selectProduct();
+            Map<String, SalesLedgerProduct> productMap = new HashMap<>();
+            for (SalesLedgerProduct product : salesLedgerProducts) {
+                String key = product.getProductCategory() + "|" + product.getSpecificationModel();
+                productMap.put(key, product);
+            }
 
-            ExcelUtil<StockInventoryExportData> util = new ExcelUtil<StockInventoryExportData>(StockInventoryExportData.class);
-            List<StockInventoryExportData> list = util.importExcel(file.getInputStream());
-
-            // 璁板綍鏈壘鍒板尮閰嶉」鐨勬暟鎹�
             List<String> unmatchedRecords = new ArrayList<>();
+            int successCount = 0;
 
-            list.forEach(dto -> {
-                boolean matched = false;
-                for (SalesLedgerProduct item : salesLedgerProducts) {
-                    if (item.getProductCategory().equals(dto.getProductName()) &&
-                            item.getSpecificationModel().equals(dto.getModel())) {
-                        StockInventoryDto stockInventoryDto = new StockInventoryDto();
-                        stockInventoryDto.setRecordId(0L);
-                        stockInventoryDto.setRecordType(StockQualifiedRecordTypeEnum.CUSTOMIZATION_STOCK_IN.getCode());
-                        stockInventoryDto.setQualitity(dto.getQualitity());
-                        stockInventoryDto.setRemark(dto.getRemark());
-                        stockInventoryDto.setWarnNum(dto.getWarnNum());
-                        if (ObjectUtils.isNotEmpty(dto.getLockedQuantity())&&dto.getLockedQuantity().compareTo(dto.getQualitity())>0) {
-                            throw new RuntimeException("鍐荤粨鏁伴噺涓嶈兘瓒呰繃鏈瀵煎叆鐨勫簱瀛樻暟閲�");
+            // productType: 1=鎴愬搧, 0=闈炴垚鍝�
+            if (productType == 1) {
+                // 鎴愬搧瀵煎叆锛堝寘鍚伐搴忕被鍒拰鐢靛帇锛�
+                ExcelUtil<FinishedProductInventoryExportData> util = new ExcelUtil<>(FinishedProductInventoryExportData.class);
+                List<FinishedProductInventoryExportData> list = util.importExcel(file.getInputStream());
+
+                for (FinishedProductInventoryExportData dto : list) {
+                    String key = dto.getProductName() + "|" + dto.getModel();
+                    SalesLedgerProduct matchedProduct = productMap.get(key);
+
+                    if (matchedProduct != null) {
+                        if (dto.getQualifiedQuantity() != null && dto.getQualifiedQuantity().compareTo(BigDecimal.ZERO) > 0) {
+                            StockInventoryDto stockInventoryDto = new StockInventoryDto();
+                            stockInventoryDto.setRecordId(0L);
+                            stockInventoryDto.setRecordType(StockInQualifiedRecordTypeEnum.CUSTOMIZATION_STOCK_IN.getCode());
+                            stockInventoryDto.setQualitity(dto.getQualifiedQuantity());
+                            stockInventoryDto.setRemark(dto.getRemark());
+                            stockInventoryDto.setWarnNum(dto.getWarnNum());
+                            stockInventoryDto.setBatchNo(dto.getQualifiedBatchNo());
+                            stockInventoryDto.setProcessCategory(dto.getProcessCategory());
+                            stockInventoryDto.setVoltage(dto.getVoltage());
+
+                            if (ObjectUtils.isNotEmpty(dto.getQualifiedLockedQuantity())) {
+                                if (dto.getQualifiedLockedQuantity().compareTo(dto.getQualifiedQuantity()) > 0) {
+                                    throw new RuntimeException("鍚堟牸鍐荤粨鏁伴噺涓嶈兘瓒呰繃鏈瀵煎叆鐨勫悎鏍煎簱瀛樻暟閲�");
+                                }
+                                stockInventoryDto.setLockedQuantity(dto.getQualifiedLockedQuantity());
+                            } else {
+                                stockInventoryDto.setLockedQuantity(BigDecimal.ZERO);
+                            }
+
+                            stockInventoryDto.setProductModelId(matchedProduct.getProductModelId());
+                            this.addstockInventory(stockInventoryDto);
+                            successCount++;
                         }
-                        stockInventoryDto.setLockedQuantity(dto.getLockedQuantity());
-                        stockInventoryDto.setProductModelId(item.getProductModelId());
-                        this.addstockInventory(stockInventoryDto);
-                        matched = true;
-                        break; // 鎵惧埌鍖归厤椤瑰悗璺冲嚭寰幆
+
+                        if (dto.getUnQualifiedQuantity() != null && dto.getUnQualifiedQuantity().compareTo(BigDecimal.ZERO) > 0) {
+                            StockUninventoryDto stockUninventoryDto = new StockUninventoryDto();
+                            stockUninventoryDto.setRecordId(0L);
+                            stockUninventoryDto.setRecordType(StockInUnQualifiedRecordTypeEnum.CUSTOMIZATION_UNSTOCK_IN.getCode());
+                            stockUninventoryDto.setQualitity(dto.getUnQualifiedQuantity());
+                            stockUninventoryDto.setRemark(dto.getRemark());
+                            stockUninventoryDto.setBatchNo(dto.getUnQualifiedBatchNo());
+
+                            if (ObjectUtils.isNotEmpty(dto.getUnQualifiedLockedQuantity())) {
+                                if (dto.getUnQualifiedLockedQuantity().compareTo(dto.getUnQualifiedQuantity()) > 0) {
+                                    throw new RuntimeException("涓嶅悎鏍煎喕缁撴暟閲忎笉鑳借秴杩囨湰娆″鍏ョ殑涓嶅悎鏍煎簱瀛樻暟閲�");
+                                }
+                                stockUninventoryDto.setLockedQuantity(dto.getUnQualifiedLockedQuantity());
+                            } else {
+                                stockUninventoryDto.setLockedQuantity(BigDecimal.ZERO);
+                            }
+
+                            stockUninventoryDto.setProductModelId(matchedProduct.getProductModelId());
+                            stockUninventoryService.addStockUninventory(stockUninventoryDto);
+                            successCount++;
+                        }
+                    } else {
+                        String unmatchedRecord = "浜у搧鍚嶇О锛�" + dto.getProductName() + "锛屽瀷鍙凤細" + dto.getModel() + " 鏈尮閰嶅埌搴撳瓨浜у搧";
+                        unmatchedRecords.add(unmatchedRecord);
                     }
                 }
-                if (!matched) {
-                    // 璁板綍鏈尮閰嶇殑鏁版嵁
-                    String unmatchedInfo = String.format("浜у搧鍚嶇О锛�%s锛岃鏍煎瀷鍙凤細%s",
-                            dto.getProductName(), dto.getModel());
-                    unmatchedRecords.add(unmatchedInfo);
+            } else {
+                // 闈炴垚鍝佸鍏ワ紙涓嶅寘鍚伐搴忕被鍒拰鐢靛帇锛�
+                ExcelUtil<NonFinishedProductInventoryExportData> util = new ExcelUtil<>(NonFinishedProductInventoryExportData.class);
+                List<NonFinishedProductInventoryExportData> list = util.importExcel(file.getInputStream());
+
+                for (NonFinishedProductInventoryExportData dto : list) {
+                    String key = dto.getProductName() + "|" + dto.getModel();
+                    SalesLedgerProduct matchedProduct = productMap.get(key);
+
+                    if (matchedProduct != null) {
+                        if (dto.getQualifiedQuantity() != null && dto.getQualifiedQuantity().compareTo(BigDecimal.ZERO) > 0) {
+                            StockInventoryDto stockInventoryDto = new StockInventoryDto();
+                            stockInventoryDto.setRecordId(0L);
+                            stockInventoryDto.setRecordType(StockInQualifiedRecordTypeEnum.CUSTOMIZATION_STOCK_IN.getCode());
+                            stockInventoryDto.setQualitity(dto.getQualifiedQuantity());
+                            stockInventoryDto.setRemark(dto.getRemark());
+                            stockInventoryDto.setWarnNum(dto.getWarnNum());
+                            stockInventoryDto.setBatchNo(dto.getQualifiedBatchNo());
+
+                            if (ObjectUtils.isNotEmpty(dto.getQualifiedLockedQuantity())) {
+                                if (dto.getQualifiedLockedQuantity().compareTo(dto.getQualifiedQuantity()) > 0) {
+                                    throw new RuntimeException("鍚堟牸鍐荤粨鏁伴噺涓嶈兘瓒呰繃鏈瀵煎叆鐨勫悎鏍煎簱瀛樻暟閲�");
+                                }
+                                stockInventoryDto.setLockedQuantity(dto.getQualifiedLockedQuantity());
+                            } else {
+                                stockInventoryDto.setLockedQuantity(BigDecimal.ZERO);
+                            }
+
+                            stockInventoryDto.setProductModelId(matchedProduct.getProductModelId());
+                            this.addstockInventory(stockInventoryDto);
+                            successCount++;
+                        }
+
+                        if (dto.getUnQualifiedQuantity() != null && dto.getUnQualifiedQuantity().compareTo(BigDecimal.ZERO) > 0) {
+                            StockUninventoryDto stockUninventoryDto = new StockUninventoryDto();
+                            stockUninventoryDto.setRecordId(0L);
+                            stockUninventoryDto.setRecordType(StockInUnQualifiedRecordTypeEnum.CUSTOMIZATION_UNSTOCK_IN.getCode());
+                            stockUninventoryDto.setQualitity(dto.getUnQualifiedQuantity());
+                            stockUninventoryDto.setRemark(dto.getRemark());
+                            stockUninventoryDto.setBatchNo(dto.getUnQualifiedBatchNo());
+
+                            if (ObjectUtils.isNotEmpty(dto.getUnQualifiedLockedQuantity())) {
+                                if (dto.getUnQualifiedLockedQuantity().compareTo(dto.getUnQualifiedQuantity()) > 0) {
+                                    throw new RuntimeException("涓嶅悎鏍煎喕缁撴暟閲忎笉鑳借秴杩囨湰娆″鍏ョ殑涓嶅悎鏍煎簱瀛樻暟閲�");
+                                }
+                                stockUninventoryDto.setLockedQuantity(dto.getUnQualifiedLockedQuantity());
+                            } else {
+                                stockUninventoryDto.setLockedQuantity(BigDecimal.ZERO);
+                            }
+
+                            stockUninventoryDto.setProductModelId(matchedProduct.getProductModelId());
+                            stockUninventoryService.addStockUninventory(stockUninventoryDto);
+                            successCount++;
+                        }
+                    } else {
+                        String unmatchedRecord = "浜у搧鍚嶇О锛�" + dto.getProductName() + "锛屽瀷鍙凤細" + dto.getModel() + " 鏈尮閰嶅埌搴撳瓨浜у搧";
+                        unmatchedRecords.add(unmatchedRecord);
+                    }
                 }
-            });
-            // 鏋勫缓杩斿洖淇℃伅
+            }
+
             StringBuilder message = new StringBuilder();
             if (!unmatchedRecords.isEmpty()) {
-                message.append("浠ヤ笅浜у搧鏈壘鍒板尮閰嶉」锛歕n");
+                message.append("瀵煎叆鎴愬姛 ").append(successCount).append(" 鏉¤褰曪紝浠ヤ笅浜у搧鏈尮閰嶏細\n");
                 for (String record : unmatchedRecords) {
                     message.append(record).append("\n");
                 }
-                throw new RuntimeException(message.toString());
+                return R.ok(message.toString());
             }
+
+            return R.ok("瀵煎叆鎴愬姛锛屽叡澶勭悊 " + successCount + " 鏉¤褰�");
         } catch (Exception e) {
-            e.printStackTrace();
-            return R.fail("瀵煎叆澶辫触锛�" + e.getMessage());
+            log.error("搴撳瓨瀵煎叆澶辫触", e);
+            return R.fail("搴撳瓨瀵煎叆澶辫触锛�" + e.getMessage());
         }
-        return R.ok("瀵煎叆鎴愬姛");
     }
 
-
     @Override
-    public void exportStockInventory(HttpServletResponse response, StockInventoryDto stockInventoryDto) {
-
+    public void exportStockInventory(HttpServletResponse response, StockInventoryDto stockInventoryDto, Integer productType) {
         List<StockInventoryExportData> list = stockInventoryMapper.listStockInventoryExportData(stockInventoryDto);
-        ExcelUtil<StockInventoryExportData> util = new ExcelUtil<>(StockInventoryExportData.class);
-        util.exportExcel(response,list, "搴撳瓨淇℃伅");
+
+        // productType: 1=鎴愬搧, 0=闈炴垚鍝�
+        if (productType == 1) {
+            // 鎴愬搧瀵煎嚭锛堝寘鍚伐搴忕被鍒拰鐢靛帇锛�
+            List<FinishedProductInventoryExportData> finishedList = new ArrayList<>();
+            for (StockInventoryExportData data : list) {
+                FinishedProductInventoryExportData finishedData = new FinishedProductInventoryExportData();
+                finishedData.setProductName(data.getProductName());
+                finishedData.setModel(data.getModel());
+                finishedData.setUnit(data.getUnit());
+                finishedData.setMaterialCode(data.getMaterialCode());
+                finishedData.setProcessCategory(data.getProcessCategory());
+                finishedData.setVoltage(data.getVoltage());
+                finishedData.setQualifiedBatchNo(data.getQualifiedBatchNo());
+                finishedData.setUnQualifiedBatchNo(data.getUnQualifiedBatchNo());
+                finishedData.setQualifiedQuantity(data.getQualifiedQuantity());
+                finishedData.setUnQualifiedQuantity(data.getUnQualifiedQuantity());
+                finishedData.setWarnNum(data.getWarnNum());
+                finishedData.setQualifiedLockedQuantity(data.getQualifiedLockedQuantity());
+                finishedData.setUnQualifiedLockedQuantity(data.getUnQualifiedLockedQuantity());
+                finishedData.setRemark(data.getRemark());
+                finishedList.add(finishedData);
+            }
+            ExcelUtil<FinishedProductInventoryExportData> util = new ExcelUtil<>(FinishedProductInventoryExportData.class);
+            util.exportExcel(response, finishedList, "鎴愬搧搴撳瓨淇℃伅");
+        } else {
+            // 闈炴垚鍝佸鍑猴紙涓嶅寘鍚伐搴忕被鍒拰鐢靛帇锛�
+            List<NonFinishedProductInventoryExportData> nonFinishedList = new ArrayList<>();
+            for (StockInventoryExportData data : list) {
+                NonFinishedProductInventoryExportData nonFinishedData = new NonFinishedProductInventoryExportData();
+                nonFinishedData.setProductName(data.getProductName());
+                nonFinishedData.setModel(data.getModel());
+                nonFinishedData.setUnit(data.getUnit());
+                nonFinishedData.setMaterialCode(data.getMaterialCode());
+                nonFinishedData.setQualifiedBatchNo(data.getQualifiedBatchNo());
+                nonFinishedData.setUnQualifiedBatchNo(data.getUnQualifiedBatchNo());
+                nonFinishedData.setQualifiedQuantity(data.getQualifiedQuantity());
+                nonFinishedData.setUnQualifiedQuantity(data.getUnQualifiedQuantity());
+                nonFinishedData.setWarnNum(data.getWarnNum());
+                nonFinishedData.setQualifiedLockedQuantity(data.getQualifiedLockedQuantity());
+                nonFinishedData.setUnQualifiedLockedQuantity(data.getUnQualifiedLockedQuantity());
+                nonFinishedData.setRemark(data.getRemark());
+                nonFinishedList.add(nonFinishedData);
+            }
+            ExcelUtil<NonFinishedProductInventoryExportData> util = new ExcelUtil<>(NonFinishedProductInventoryExportData.class);
+            util.exportExcel(response, nonFinishedList, "闈炴垚鍝佸簱瀛樹俊鎭�");
+        }
     }
 
     @Override
     public IPage<StockInRecordDto> stockInventoryPage(StockInventoryDto stockInventoryDto, Page page) {
-        return stockInventoryMapper.stockInventoryPage(stockInventoryDto,page);
+        return stockInventoryMapper.stockInventoryPage(stockInventoryDto, page);
     }
 
     @Override
     public IPage<StockInventoryDto> stockInAndOutRecord(StockInventoryDto stockInventoryDto, Page page) {
-        return stockInventoryMapper.stockInAndOutRecord(stockInventoryDto,page);
+        return stockInventoryMapper.stockInAndOutRecord(stockInventoryDto, page);
     }
 
     @Override
     public Boolean frozenStock(StockInventoryDto stockInventoryDto) {
         StockInventory stockInventory = stockInventoryMapper.selectById(stockInventoryDto.getId());
-        if (stockInventory.getQualitity().compareTo(stockInventoryDto.getLockedQuantity())<0) {
+        if (stockInventory.getQualitity().compareTo(stockInventoryDto.getLockedQuantity()) < 0) {
             throw new RuntimeException("鍐荤粨鏁伴噺涓嶈兘瓒呰繃搴撳瓨鏁伴噺");
         }
         if (ObjectUtils.isEmpty(stockInventory.getLockedQuantity())) {
             stockInventory.setLockedQuantity(stockInventoryDto.getLockedQuantity());
-        }else {
+        } else {
             stockInventory.setLockedQuantity(stockInventory.getLockedQuantity().add(stockInventoryDto.getLockedQuantity()));
         }
         return this.updateById(stockInventory);
@@ -201,10 +582,54 @@
     @Override
     public Boolean thawStock(StockInventoryDto stockInventoryDto) {
         StockInventory stockInventory = stockInventoryMapper.selectById(stockInventoryDto.getId());
-        if (stockInventory.getLockedQuantity().compareTo(stockInventoryDto.getLockedQuantity())<0) {
+        if (stockInventory.getLockedQuantity().compareTo(stockInventoryDto.getLockedQuantity()) < 0) {
             throw new RuntimeException("瑙e喕鏁伴噺涓嶈兘瓒呰繃鍐荤粨鏁伴噺");
         }
         stockInventory.setLockedQuantity(stockInventory.getLockedQuantity().subtract(stockInventoryDto.getLockedQuantity()));
         return this.updateById(stockInventory);
     }
+
+    // 鍏ュ簱鍚堝苟鍞竴閿細鎴愬搧鎸夌被鍒拰鐢靛帇锛屽叾浣欏彧鎸変骇鍝佽鏍笺��
+    private StockInventory findInventoryForMerge(Long productModelId, String processCategory, String voltage) {
+        LambdaQueryWrapper<StockInventory> queryWrapper = new LambdaQueryWrapper<StockInventory>()
+                .eq(StockInventory::getProductModelId, productModelId)
+                .orderByAsc(StockInventory::getId);
+        if (usesDimensionIdentity(processCategory, voltage)) {
+            queryWrapper.eq(StockInventory::getProcessCategory, normalizeDimension(processCategory));
+            queryWrapper.eq(StockInventory::getVoltage, normalizeDimension(voltage));
+        }
+        List<StockInventory> inventories = stockInventoryMapper.selectList(queryWrapper);
+        return inventories.isEmpty() ? null : inventories.get(0);
+    }
+
+    // 灏嗗叆搴撹褰� DTO 澶嶅埗涓烘寔涔呭寲瀹炰綋銆�
+    private StockInRecord toStockInRecord(StockInRecordDto stockInRecordDto) {
+        StockInRecord stockInRecord = new StockInRecord();
+        stockInRecord.setRecordId(stockInRecordDto.getRecordId());
+        stockInRecord.setRecordType(stockInRecordDto.getRecordType());
+        stockInRecord.setStockInNum(stockInRecordDto.getStockInNum());
+        stockInRecord.setProductModelId(stockInRecordDto.getProductModelId());
+        stockInRecord.setRemark(stockInRecordDto.getRemark());
+        stockInRecord.setType(stockInRecordDto.getType());
+        stockInRecord.setLockedQuantity(stockInRecordDto.getLockedQuantity());
+        stockInRecord.setWarnNum(stockInRecordDto.getWarnNum());
+        stockInRecord.setApproveStatus(stockInRecordDto.getApproveStatus());
+        stockInRecord.setBatchNo(stockInRecordDto.getBatchNo());
+        stockInRecord.setProcessCategory(stockInRecordDto.getProcessCategory());
+        stockInRecord.setVoltage(stockInRecordDto.getVoltage());
+        return stockInRecord;
+    }
+
+    // 鍙甯︽湁鎴愬搧缁村害锛屽氨鎸夌淮搴︿綔涓哄敮涓�閿��
+    private boolean usesDimensionIdentity(String processCategory, String voltage) {
+        return StringUtils.isNotBlank(processCategory) || StringUtils.isNotBlank(voltage);
+    }
+
+    private String normalizeDimension(String value) {
+        return StringUtils.trimToEmpty(value);
+    }
+
+    private BigDecimal defaultDecimal(BigDecimal value) {
+        return value == null ? BigDecimal.ZERO : value;
+    }
 }

--
Gitblit v1.9.3