From 09a6947f5fcef8dfabee5f5aca829becfbc21b47 Mon Sep 17 00:00:00 2001
From: gongchunyi <deslre0381@gmail.com>
Date: 星期二, 10 二月 2026 15:40:30 +0800
Subject: [PATCH] fix: 产品规格型号导入与下载模版

---
 src/main/java/com/ruoyi/basic/controller/ProductController.java         |   22 +++++++++--
 src/main/java/com/ruoyi/basic/dto/ProductModelExportDto.java            |   24 ++++++++++++
 src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java |   57 +++++++++++++++++++---------
 src/main/java/com/ruoyi/basic/service/IProductModelService.java         |    3 +
 4 files changed, 82 insertions(+), 24 deletions(-)

diff --git a/src/main/java/com/ruoyi/basic/controller/ProductController.java b/src/main/java/com/ruoyi/basic/controller/ProductController.java
index a97bd7b..252ab4d 100644
--- a/src/main/java/com/ruoyi/basic/controller/ProductController.java
+++ b/src/main/java/com/ruoyi/basic/controller/ProductController.java
@@ -5,10 +5,12 @@
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.ruoyi.basic.dto.ProductDto;
 import com.ruoyi.basic.dto.ProductModelDto;
+import com.ruoyi.basic.dto.ProductModelExportDto;
 import com.ruoyi.basic.dto.ProductTreeDto;
 import com.ruoyi.basic.pojo.ProductModel;
 import com.ruoyi.basic.service.IProductModelService;
 import com.ruoyi.basic.service.IProductService;
+import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.framework.aspectj.lang.annotation.Log;
 import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
 import com.ruoyi.framework.web.controller.BaseController;
@@ -23,6 +25,7 @@
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
+import javax.servlet.http.HttpServletResponse;
 import java.util.List;
 
 @RestController
@@ -124,9 +127,20 @@
     /**
      * 瀵煎叆浜у搧
      */
-    @Log(title = "瀵煎叆浜у搧",businessType = BusinessType.IMPORT)
-    @PostMapping("import")
-    public AjaxResult importProduct(MultipartFile file) {
-        return AjaxResult.success(productModelService.importProduct(file));
+    @PostMapping("/import")
+    @Log(title = "瀵煎叆浜у搧", businessType = BusinessType.IMPORT)
+    public AjaxResult importProductModel(@RequestParam("file") MultipartFile file, Integer productId) {
+        return productModelService.importProductModel(file, productId);
+    }
+
+    /**
+     * 浜у搧瀵煎叆妯℃澘
+     */
+    @GetMapping("/export")
+    @ApiOperation("浜у搧瀵煎叆妯℃澘")
+    @Log(title = "浜у搧瀵煎叆妯℃澘", businessType = BusinessType.EXPORT)
+    public void importProduct(HttpServletResponse response) {
+        ExcelUtil<ProductModelExportDto> excelUtil = new ExcelUtil<>(ProductModelExportDto.class);
+        excelUtil.importTemplateExcel(response, "浜у搧瑙勬牸瀵煎叆妯℃澘");
     }
 }
diff --git a/src/main/java/com/ruoyi/basic/dto/ProductModelExportDto.java b/src/main/java/com/ruoyi/basic/dto/ProductModelExportDto.java
new file mode 100644
index 0000000..c8c838a
--- /dev/null
+++ b/src/main/java/com/ruoyi/basic/dto/ProductModelExportDto.java
@@ -0,0 +1,24 @@
+package com.ruoyi.basic.dto;
+
+import com.ruoyi.framework.aspectj.lang.annotation.Excel;
+import lombok.Data;
+
+/**
+ * <br>
+ * 浜у搧瀵煎嚭妯℃澘
+ * </br>
+ *
+ * @author deslrey
+ * @version 1.0
+ * @since 2026/02/10 14:39
+ */
+@Data
+public class ProductModelExportDto {
+
+    @Excel(name = "瑙勬牸鍨嬪彿")
+    private String model;
+
+    @Excel(name = "鍗曚綅")
+    private String unit;
+
+}
diff --git a/src/main/java/com/ruoyi/basic/service/IProductModelService.java b/src/main/java/com/ruoyi/basic/service/IProductModelService.java
index f254612..dd268d6 100644
--- a/src/main/java/com/ruoyi/basic/service/IProductModelService.java
+++ b/src/main/java/com/ruoyi/basic/service/IProductModelService.java
@@ -6,6 +6,7 @@
 import com.ruoyi.basic.dto.ProductDto;
 import com.ruoyi.basic.dto.ProductModelDto;
 import com.ruoyi.basic.pojo.ProductModel;
+import com.ruoyi.framework.web.domain.AjaxResult;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
@@ -33,5 +34,5 @@
      */
     IPage<ProductModel> modelListPage(Page page , ProductDto productDto);
 
-    Boolean importProduct(MultipartFile file);
+    AjaxResult importProductModel(MultipartFile file, Integer productId);
 }
diff --git a/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java b/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
index cf9ef84..16a45b1 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
@@ -12,17 +12,17 @@
 import com.ruoyi.basic.pojo.Product;
 import com.ruoyi.basic.pojo.ProductModel;
 import com.ruoyi.basic.service.IProductModelService;
+import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
-import com.ruoyi.common.utils.uuid.IdUtils;
+import com.ruoyi.framework.web.domain.AjaxResult;
 import com.ruoyi.sales.mapper.SalesLedgerProductMapper;
 import com.ruoyi.sales.pojo.SalesLedgerProduct;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.time.LocalDate;
-import java.time.format.DateTimeFormatter;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
@@ -47,13 +47,12 @@
 
         if (productModelDto.getId() == null) {
             ProductModel productModel = new ProductModel();
-            BeanUtils.copyProperties(productModelDto,productModel);
+            BeanUtils.copyProperties(productModelDto, productModel);
             return productModelMapper.insert(productModel);
         } else {
             return productModelMapper.updateById(productModelDto);
         }
     }
-
 
 
     @Override
@@ -76,6 +75,7 @@
 
     /**
      * 鏍规嵁id鏌ヨ浜у搧瑙勬牸鍒嗛〉鏌ヨ
+     *
      * @param page
      * @param productDto
      * @return
@@ -88,24 +88,43 @@
     }
 
     @Override
-    public Boolean importProduct(MultipartFile file) {
+    @Transactional(rollbackFor = Exception.class)
+    public AjaxResult importProductModel(MultipartFile file, Integer productId) {
+        if (productId == null) {
+            return AjaxResult.error("璇峰厛閫夋嫨浜у搧鍐嶅鍏ヨ鏍煎瀷鍙�");
+        }
+
+        Product product = productMapper.selectById(productId);
+        if (product == null) {
+            return AjaxResult.error("閫夋嫨鐨勪骇鍝佷笉瀛樺湪");
+        }
+
         try {
             ExcelUtil<ProductModel> productModelExcelUtil = new ExcelUtil<>(ProductModel.class);
             List<ProductModel> productModelList = productModelExcelUtil.importExcel(file.getInputStream());
-            Map<String, List<ProductModel>> collect = productModelList.stream().collect(Collectors.groupingBy(ProductModel::getProductName));
-            collect.forEach((k,v)->{
-                Product product = productMapper.selectOne(new LambdaQueryWrapper<Product>().eq(Product::getProductName, k).last("LIMIT 1"));
-                if (product != null) {
-                    v.forEach(productModel -> {
-                        productModel.setProductId(product.getId());
-                    });
-                    this.saveOrUpdateBatch(v);
+
+            if (productModelList == null || productModelList.isEmpty()) {
+                return AjaxResult.error("瀵煎叆鏁版嵁涓嶈兘涓虹┖");
+            }
+
+            for (int i = 0; i < productModelList.size(); i++) {
+                ProductModel item = productModelList.get(i);
+                int rowNum = i + 2;
+
+                if (StringUtils.isEmpty(item.getModel())) {
+                    return AjaxResult.error("绗� " + rowNum + " 琛屽鍏ュけ璐�: [瑙勬牸鍨嬪彿] 涓嶈兘涓虹┖");
                 }
-            });
-            return true;
-        }catch (Exception e) {
-            e.printStackTrace();
+                if (StringUtils.isEmpty(item.getUnit())) {
+                    return AjaxResult.error("绗� " + rowNum + " 琛屽鍏ュけ璐�: [鍗曚綅] 涓嶈兘涓虹┖");
+                }
+                item.setProductId(product.getId());
+            }
+
+            saveOrUpdateBatch(productModelList);
+            return AjaxResult.success("鎴愬姛瀵煎叆 " + productModelList.size() + " 鏉℃暟鎹�");
+        } catch (Exception e) {
+            log.error("瀵煎叆浜у搧瑙勬牸寮傚父", e);
+            return AjaxResult.error("瀵煎叆澶辫触");
         }
-        return false;
     }
 }

--
Gitblit v1.9.3