From 78e8144e8713af9e6b3aa14ff1f8b4091d698616 Mon Sep 17 00:00:00 2001
From: gongchunyi <deslre0381@gmail.com>
Date: 星期五, 24 四月 2026 14:25:13 +0800
Subject: [PATCH] feat: 产品维护导出功能
---
src/main/java/com/ruoyi/basic/controller/ProductController.java | 13 +++++-
src/main/java/com/ruoyi/basic/dto/ProductModelDataExportDto.java | 20 ++++++++++
src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java | 62 +++++++++++++++++++++++++++++++
3 files changed, 93 insertions(+), 2 deletions(-)
diff --git a/src/main/java/com/ruoyi/basic/controller/ProductController.java b/src/main/java/com/ruoyi/basic/controller/ProductController.java
index 252ab4d..c14f6b7 100644
--- a/src/main/java/com/ruoyi/basic/controller/ProductController.java
+++ b/src/main/java/com/ruoyi/basic/controller/ProductController.java
@@ -17,8 +17,6 @@
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.sales.pojo.SalesLedgerProduct;
import com.ruoyi.sales.service.ISalesLedgerProductService;
-import com.ruoyi.sales.service.ISalesLedgerService;
-import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
@@ -38,6 +36,7 @@
private IProductModelService productModelService;
@Autowired
private ISalesLedgerProductService salesLedgerProductService;
+
/**
* 鏌ヨ浜у搧
*/
@@ -143,4 +142,14 @@
ExcelUtil<ProductModelExportDto> excelUtil = new ExcelUtil<>(ProductModelExportDto.class);
excelUtil.importTemplateExcel(response, "浜у搧瑙勬牸瀵煎叆妯℃澘");
}
+
+ /**
+ * 浜у搧瀵煎嚭
+ */
+ @GetMapping("exportData")
+ @ApiOperation("浜у搧瀵煎嚭")
+ public void exportData(HttpServletResponse response, Long productId) {
+ productModelService.exportData(response, productId);
+ }
+
}
diff --git a/src/main/java/com/ruoyi/basic/dto/ProductModelDataExportDto.java b/src/main/java/com/ruoyi/basic/dto/ProductModelDataExportDto.java
new file mode 100644
index 0000000..ae53083
--- /dev/null
+++ b/src/main/java/com/ruoyi/basic/dto/ProductModelDataExportDto.java
@@ -0,0 +1,20 @@
+package com.ruoyi.basic.dto;
+
+import com.ruoyi.framework.aspectj.lang.annotation.Excel;
+import lombok.Data;
+
+@Data
+public class ProductModelDataExportDto {
+
+ @Excel(name = "浜у搧澶х被")
+ private String productCategoryName;
+
+ @Excel(name = "浜у搧鍚嶇О")
+ private String productName;
+
+ @Excel(name = "瑙勬牸鍨嬪彿")
+ private String model;
+
+ @Excel(name = "鍗曚綅")
+ private String unit;
+}
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 9382134..41daac9 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/ProductModelServiceImpl.java
@@ -7,6 +7,7 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.basic.dto.ProductDto;
+import com.ruoyi.basic.dto.ProductModelDataExportDto;
import com.ruoyi.basic.dto.ProductModelDto;
import com.ruoyi.basic.mapper.ProductMapper;
import com.ruoyi.basic.mapper.ProductModelMapper;
@@ -25,6 +26,7 @@
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
+import javax.servlet.http.HttpServletResponse;
import java.util.*;
import java.util.stream.Collectors;
@@ -151,4 +153,64 @@
throw new ServiceException("瀵煎叆澶辫触");
}
}
+
+ @Override
+ public void exportData(HttpServletResponse response, Long productId) {
+ List<Product> allProducts = productMapper.selectList(null);
+ Map<Long, Product> productMap = allProducts.stream()
+ .filter(product -> product.getId() != null)
+ .collect(Collectors.toMap(Product::getId, product -> product));
+ Map<Long, List<Long>> parentChildrenMap = new HashMap<>();
+ for (Product product : allProducts) {
+ if (product.getParentId() == null || product.getId() == null) {
+ continue;
+ }
+ parentChildrenMap.computeIfAbsent(product.getParentId(), key -> new ArrayList<>()).add(product.getId());
+ }
+
+ LambdaQueryWrapper<ProductModel> queryWrapper = new LambdaQueryWrapper<>();
+ if (productId != null) {
+ Set<Long> targetProductIds = new HashSet<>();
+ ArrayDeque<Long> queue = new ArrayDeque<>();
+ queue.offer(productId);
+ while (!queue.isEmpty()) {
+ Long currentId = queue.poll();
+ if (currentId == null || !targetProductIds.add(currentId)) {
+ continue;
+ }
+ List<Long> children = parentChildrenMap.get(currentId);
+ if (children != null && !children.isEmpty()) {
+ queue.addAll(children);
+ }
+ }
+ queryWrapper.in(ProductModel::getProductId, targetProductIds);
+ }
+ List<ProductModel> productModelList = list(queryWrapper);
+ List<ProductModelDataExportDto> exportList = productModelList.stream().map(item -> {
+ ProductModelDataExportDto dto = new ProductModelDataExportDto();
+ Product currentProduct = productMap.get(item.getProductId());
+ dto.setProductName(currentProduct == null ? "" : currentProduct.getProductName());
+ dto.setProductCategoryName(resolveRootCategoryName(currentProduct, productMap));
+ dto.setModel(item.getModel());
+ dto.setUnit(item.getUnit());
+ return dto;
+ }).collect(Collectors.toList());
+ ExcelUtil<ProductModelDataExportDto> excelUtil = new ExcelUtil<>(ProductModelDataExportDto.class);
+ excelUtil.exportExcel(response, exportList, "浜у搧瑙勬牸鏁版嵁");
+ }
+
+ private String resolveRootCategoryName(Product currentProduct, Map<Long, Product> productMap) {
+ if (currentProduct == null) {
+ return "";
+ }
+ Product node = currentProduct;
+ while (node.getParentId() != null) {
+ Product parent = productMap.get(node.getParentId());
+ if (parent == null) {
+ break;
+ }
+ node = parent;
+ }
+ return node.getProductName();
+ }
}
--
Gitblit v1.9.3