From d1439e537a47804c7c78e3e78ccd1aa24e2fdd23 Mon Sep 17 00:00:00 2001
From: gongchunyi <deslre0381@gmail.com>
Date: 星期三, 11 三月 2026 13:18:43 +0800
Subject: [PATCH] feat: 生产计划数据的导入、导出、模板下载

---
 src/main/java/com/ruoyi/productionPlan/service/impl/ProductionPlanServiceImpl.java |  143 +++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 143 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/ruoyi/productionPlan/service/impl/ProductionPlanServiceImpl.java b/src/main/java/com/ruoyi/productionPlan/service/impl/ProductionPlanServiceImpl.java
index f6d2414..a93d311 100644
--- a/src/main/java/com/ruoyi/productionPlan/service/impl/ProductionPlanServiceImpl.java
+++ b/src/main/java/com/ruoyi/productionPlan/service/impl/ProductionPlanServiceImpl.java
@@ -7,10 +7,16 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.common.exception.ServiceException;
 import com.ruoyi.common.utils.StringUtils;
+import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.common.utils.http.HttpUtils;
+import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.framework.config.AliDingConfig;
+import com.ruoyi.production.pojo.ProductOrder;
+import com.ruoyi.production.service.ProductOrderService;
 import com.ruoyi.productionPlan.dto.ProductionPlanDto;
+import com.ruoyi.productionPlan.dto.ProductionPlanImportDto;
 import com.ruoyi.productionPlan.dto.ProductionPlanSummaryDto;
 import com.ruoyi.productionPlan.mapper.ProductionPlanMapper;
 import com.ruoyi.productionPlan.pojo.ProductionPlan;
@@ -19,7 +25,10 @@
 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.nio.charset.StandardCharsets;
 import java.time.Instant;
 import java.time.LocalDateTime;
@@ -30,6 +39,8 @@
 import java.util.*;
 import java.util.concurrent.locks.ReentrantLock;
 import java.util.stream.Collectors;
+
+import static com.ruoyi.productionPlan.enums.DataSourceTypeEnum.PRODUCTION_FORECAST;
 
 /**
  * <br>
@@ -50,6 +61,9 @@
 
     @Autowired
     private ProductionPlanMapper productionPlanMapper;
+
+    @Autowired
+    private ProductOrderService productOrderService;
 
     /**
      * 鍚屾閿侊紝纭繚鎵嬪姩鍜屽畾鏃朵换鍔′笉鍚屾椂鎵ц
@@ -75,6 +89,83 @@
     @Override
     public void syncProdDataJob() {
         syncProdData(2);
+    }
+
+    /**
+     * 鍚堝苟鐢熶骇璁″垝
+     */
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public boolean combine(ProductionPlanDto productionPlanDto) {
+        if (productionPlanDto.getIds() == null || productionPlanDto.getIds().isEmpty()) {
+            return false;
+        }
+
+        //  鏌ヨ涓荤敓浜ц鍒�
+        List<ProductionPlan> plans = productionPlanMapper.selectBatchIds(productionPlanDto.getIds());
+
+        //  鏍¢獙鏄惁瀛樺湪涓嶅悓鐨勪骇鍝佸悕绉�
+        String firstProductName = plans.get(0).getProductName();
+        if (plans.stream().anyMatch(p -> !p.getProductName().equals(firstProductName))) {
+            log.warn("鍚堝苟澶辫触锛屽瓨鍦ㄤ笉鍚岀殑浜у搧鍚嶇О");
+            return false;
+        }
+
+        // 鏍¢獙鏄惁瀛樺湪涓嶅悓鐨勪骇鍝佽鏍�
+        String firstProductSpec = plans.get(0).getProductSpec();
+        if (plans.stream().anyMatch(p -> !p.getProductSpec().equals(firstProductSpec))) {
+            log.warn("鍚堝苟澶辫触锛屽瓨鍦ㄤ笉鍚岀殑浜у搧瑙勬牸");
+            return false;
+        }
+
+
+        // 鍙犲姞鏂规暟
+        BigDecimal totalVolume = plans.stream()
+                .map(ProductionPlan::getVolume)
+                .filter(v -> v != null)
+                .reduce(BigDecimal.ZERO, BigDecimal::add);
+        // 鍒ゆ柇涓嬪彂鏁伴噺鏄惁澶т簬绛変簬鏂规暟
+        if (productionPlanDto.getTotalAssignedQuantity().compareTo(totalVolume) > 0) {
+            log.warn("鍚堝苟澶辫触锛屼笅鍙戞暟閲忎笉鑳藉ぇ浜庢柟鏁�");
+            return false;
+        }
+
+        // 鏍规嵁涓嬪彂鏁伴噺锛屼粠绗竴涓敓浜ц鍒掑紑濮嬪垎閰嶆柟鏁�
+        BigDecimal assignedVolume = BigDecimal.ZERO;
+        for (ProductionPlan plan : plans) {
+            BigDecimal volume = plan.getVolume();
+            if (volume == null) {
+                continue;
+            }
+
+            if (assignedVolume.add(volume).compareTo(productionPlanDto.getTotalAssignedQuantity()) >= 0) {
+                // 鏈�鍚庝竴涓鍒掞紝鍒嗛厤鍓╀綑鏂规暟
+                plan.setAssignedQuantity(productionPlanDto.getTotalAssignedQuantity().subtract(assignedVolume));
+                break;
+            }
+
+            // 鍒嗛厤褰撳墠璁″垝鏂规暟
+            plan.setAssignedQuantity(volume);
+            productionPlanMapper.updateById(plan);
+            assignedVolume = assignedVolume.add(volume);
+        }
+
+        // 鍒涘缓鐢熶骇璁㈠崟
+        ProductOrder productOrder = new ProductOrder();
+        String combineIds = StringUtils.join(productionPlanDto.getIds(), ",");
+        productOrder.setCombineProductionPlanIds(combineIds);
+        productOrder.setQuantity(productionPlanDto.getTotalAssignedQuantity());
+        productOrder.setPlanCompleteTime(productionPlanDto.getPlanCompleteTime());
+        productOrderService.addProductOrder(productOrder);
+        return true;
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public boolean add(ProductionPlanDto productionPlanDto) {
+        productionPlanDto.setDataSourceType(PRODUCTION_FORECAST.getCode());
+        productionPlanMapper.insert(productionPlanDto);
+        return true;
     }
 
     /**
@@ -339,4 +430,56 @@
     public List<ProductionPlanSummaryDto> summaryByProductType(ProductionPlanSummaryDto query) {
         return baseMapper.selectSummaryByProductType(query);
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void importProdData(MultipartFile file) {
+        if (file == null || file.isEmpty()) {
+            throw new ServiceException("瀵煎叆鏁版嵁涓嶈兘涓虹┖");
+        }
+        ExcelUtil<ProductionPlanImportDto> excelUtil = new ExcelUtil<>(ProductionPlanImportDto.class);
+        List<ProductionPlanImportDto> list;
+        try {
+            list = excelUtil.importExcel(file.getInputStream());
+        } catch (Exception e) {
+            log.error("鐢熶骇闇�姹侲xcel瀵煎叆澶辫触", e);
+            throw new ServiceException("Excel瑙f瀽澶辫触");
+        }
+
+        if (list == null || list.isEmpty()) {
+            throw new ServiceException("Excel娌℃湁鏁版嵁");
+        }
+        List<ProductionPlan> entityList = new ArrayList<>(list.size());
+        ProductionPlan entity;
+        for (ProductionPlanImportDto dto : list) {
+            entity = new ProductionPlan();
+            BeanUtils.copyProperties(dto, entity);
+            entity.setCreateTime(LocalDateTime.now());
+            entity.setUpdateTime(LocalDateTime.now());
+            entity.setDataSourceType(2);
+            entity.setDataSyncType(1);
+
+            entityList.add(entity);
+        }
+        this.saveBatch(entityList);
+    }
+
+    @Override
+    public void exportProdData(HttpServletResponse response, List<Long> ids) {
+        List<ProductionPlan> list;
+        if (ids != null && !ids.isEmpty()) {
+            list = baseMapper.selectBatchIds(ids);
+        } else {
+            list = baseMapper.selectList(null);
+        }
+
+        List<ProductionPlanImportDto> exportList = new ArrayList<>();
+        for (ProductionPlan entity : list) {
+            ProductionPlanImportDto dto = new ProductionPlanImportDto();
+            BeanUtils.copyProperties(entity, dto);
+            exportList.add(dto);
+        }
+        ExcelUtil<ProductionPlanImportDto> util = new ExcelUtil<>(ProductionPlanImportDto.class);
+        util.exportExcel(response, exportList, "閿�鍞敓浜ч渶姹傛暟鎹�");
+    }
 }

--
Gitblit v1.9.3