feat: 产品维护导出功能

gongchunyi

4 小时以前 78e8144e8713af9e6b3aa14ff1f8b4091d698616

feat: 产品维护导出功能

 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);
    }

}

 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;
}

 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();
    }
}