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 |   96 ++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 93 insertions(+), 3 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 f2e238c..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,13 +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.basic.pojo.Product;
+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;
@@ -22,7 +25,9 @@
 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;
@@ -34,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>
@@ -88,6 +95,7 @@
      * 鍚堝苟鐢熶骇璁″垝
      */
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public boolean combine(ProductionPlanDto productionPlanDto) {
         if (productionPlanDto.getIds() == null || productionPlanDto.getIds().isEmpty()) {
             return false;
@@ -116,17 +124,47 @@
                 .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();
-        //  鍚堝苟鐢熶骇璁″垝ids
         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;
     }
 
@@ -392,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