huminmin
2026-04-22 b7b7c0a641c7bf53d3d31449dc8c31efba58ea97
src/main/java/com/ruoyi/sales/service/impl/SalesLedgerServiceImpl.java
@@ -1,5 +1,8 @@
package com.ruoyi.sales.service.impl;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -8,6 +11,9 @@
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.ruoyi.account.pojo.AccountIncome;
import com.ruoyi.account.service.AccountIncomeService;
import com.ruoyi.basic.mapper.CustomerMapper;
import com.ruoyi.basic.mapper.ProductMapper;
@@ -17,9 +23,11 @@
import com.ruoyi.common.enums.FileNameType;
import com.ruoyi.common.exception.base.BaseException;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.HackLoopTableRenderPolicy;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.device.execl.DeviceMaintenanceExeclDto;
import com.ruoyi.framework.security.LoginUser;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.other.mapper.TempFileMapper;
@@ -34,6 +42,7 @@
import com.ruoyi.sales.dto.*;
import com.ruoyi.sales.mapper.*;
import com.ruoyi.sales.pojo.*;
import com.ruoyi.sales.service.ISalesLedgerProductService;
import com.ruoyi.sales.service.ISalesLedgerService;
import com.ruoyi.stock.mapper.StockInventoryMapper;
import lombok.RequiredArgsConstructor;
@@ -48,11 +57,13 @@
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.io.InputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
@@ -60,11 +71,14 @@
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.YearMonth;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
 * 销售台账Service业务层处理
@@ -129,6 +143,8 @@
    private final QualityInspectMapper qualityInspectMapper;
    private final StockInventoryMapper stockInventoryMapper;
    private ISalesLedgerProductService salesLedgerProductService;
    @Autowired
@@ -209,6 +225,99 @@
    }
    @Override
    public void exportShippingNote(HttpServletResponse response, List<Long> ids) {
        log.info("开始导出送货单,台账ID列表: {}", ids);
        // 参数校验
        if (ids == null || ids.isEmpty()) {
            throw new BaseException("请选择要导出的销售台账");
        }
        List<SalesLedger> salesLedgerList = salesLedgerMapper.selectList(
                Wrappers.<SalesLedger>lambdaQuery()
                        .in(SalesLedger::getId, ids)
        );
        if (salesLedgerList.isEmpty()) {
            throw new BaseException("未找到对应销售台账数据");
        }
        response.setContentType("application/zip");
        response.setHeader("Content-Disposition", "attachment;filename=送货单.zip");
        response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
        // 直接内存打包,抛弃临时文件
        try (ZipOutputStream zos = new ZipOutputStream(response.getOutputStream())) {
            for (SalesLedger ledger : salesLedgerList) {
                log.info("处理合同号: {}", ledger.getSalesContractNo());
                // 查询明细
                SalesLedgerProduct cond = new SalesLedgerProduct();
                cond.setSalesLedgerId(ledger.getId());
                List<SalesLedgerProduct> productList = salesLedgerProductServiceImpl.selectSalesLedgerProductList(cond);
                // 自动序号
                for (int i = 0; i < productList.size(); i++) {
                    productList.get(i).setSerialNumber(i + 1);
                }
                // 关联查询客户信息
                Customer customer = customerMapper.selectById(ledger.getCustomerId());
                // 组装数据
                Map<String, Object> dataMap = new HashMap<>();
                dataMap.put("customerName", ledger.getCustomerName());
                dataMap.put("companyAddress", customer.getCompanyAddress());
                dataMap.put("contactPerson", customer.getContactPerson());
                dataMap.put("contactPhone", customer.getContactPhone());
                dataMap.put("companyPhone", customer.getCompanyPhone());
                dataMap.put("salesContractNo", ledger.getSalesContractNo());
                dataMap.put("deliveryDate", formatDate(LocalDate.now()));
                // 放入商品集合
                dataMap.put("products", productList);
                // 每次循环全新读取模板,彻底解决流问题
                ByteArrayOutputStream bos = new ByteArrayOutputStream();
                try (InputStream templateIs = this.getClass().getResourceAsStream("/static/shipping-note.xlsx")) {
                    if (templateIs == null) throw new BaseException("模板文件不存在");
                    // EasyExcel填充
                    ExcelWriter writer = EasyExcel.write(bos).withTemplate(templateIs).build();
                    WriteSheet sheet = EasyExcel.writerSheet().build();
                    writer.fill(dataMap, sheet);
                    writer.finish();
                }
                // 打包进ZIP
                String excelName = String.format("送货单_%s.xlsx",
                        Optional.ofNullable(ledger.getSalesContractNo()).orElse(String.valueOf(ledger.getId())));
                ZipEntry entry = new ZipEntry(excelName);
                zos.putNextEntry(entry);
                zos.write(bos.toByteArray());
                zos.closeEntry();
                bos.close();
            }
            zos.finish();
            log.info("批量导出完成,共{}份送货单", salesLedgerList.size());
        } catch (BaseException e) {
            log.error("导出失败:{}", e.getMessage(), e);
            throw e;
        } catch (Exception e) {
            log.error("导出异常", e);
            throw new BaseException("导出失败:" + e.getMessage());
        }
    }
    public static String formatDate(LocalDate date) {
        if (date == null) {
            return null;
        }
        return date.format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
    }
    @Override
    public List<Map<String, Object>> getSalesNo() {
        LambdaQueryWrapper<SalesLedger> queryWrapper = Wrappers.lambdaQuery();
        queryWrapper.select(SalesLedger::getId, SalesLedger::getSalesContractNo, SalesLedger::getProjectName);