From beeba1259f983a5f92e6b8137d2fb05389f184b4 Mon Sep 17 00:00:00 2001
From: chenrui <1187576398@qq.com>
Date: 星期二, 27 五月 2025 11:39:43 +0800
Subject: [PATCH] 开票台账功能修改

---
 src/main/java/com/ruoyi/sales/service/impl/InvoiceLedgerServiceImpl.java |   20 +++++++++++++++-----
 1 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/ruoyi/sales/service/impl/InvoiceLedgerServiceImpl.java b/src/main/java/com/ruoyi/sales/service/impl/InvoiceLedgerServiceImpl.java
index 696f72e..9ce10fb 100644
--- a/src/main/java/com/ruoyi/sales/service/impl/InvoiceLedgerServiceImpl.java
+++ b/src/main/java/com/ruoyi/sales/service/impl/InvoiceLedgerServiceImpl.java
@@ -12,6 +12,7 @@
 import com.ruoyi.sales.excel.InvoiceLedgerExcelDto;
 import com.ruoyi.sales.mapper.InvoiceLedgerFileMapper;
 import com.ruoyi.sales.mapper.InvoiceLedgerMapper;
+import com.ruoyi.sales.mapper.InvoiceRegistrationMapper;
 import com.ruoyi.sales.mapper.InvoiceRegistrationProductMapper;
 import com.ruoyi.sales.pojo.InvoiceLedger;
 import com.ruoyi.sales.pojo.InvoiceLedgerFile;
@@ -49,6 +50,9 @@
 
     @Autowired
     private InvoiceRegistrationProductMapper invoiceRegistrationProductMapper;
+
+    @Autowired
+    private InvoiceRegistrationMapper invoiceRegistrationMapper;
 
     /**
      * 寮�绁ㄥ彴璐︽柊澧�
@@ -158,13 +162,13 @@
     /**
      * 闄勪欢涓嬭浇
      * @param response
-     * @param invoiceLedgerDto
+     * @param invoiceRegistrationProductDto
      * @return
      */
     @Override
-    public void invoiceLedgerExport(HttpServletResponse response, InvoiceLedgerDto invoiceLedgerDto) {
-        List<InvoiceLedgerDto> invoiceLedgerDtoList = invoiceLedgerMapper.invoiceLedgerList(invoiceLedgerDto);
-        List<InvoiceLedgerExcelDto> invoiceLedgerExcelDtoList = invoiceLedgerDtoList.stream().map(item -> {
+    public void invoiceLedgerExport(HttpServletResponse response, InvoiceRegistrationProductDto invoiceRegistrationProductDto) {
+        List<InvoiceRegistrationProductDto> invoiceRegistrationProductList = invoiceRegistrationProductMapper.invoiceRegistrationProductList(invoiceRegistrationProductDto);
+        List<InvoiceLedgerExcelDto> invoiceLedgerExcelDtoList = invoiceRegistrationProductList.stream().map(item -> {
             InvoiceLedgerExcelDto invoiceLedgerExcelDto = new InvoiceLedgerExcelDto();
             BeanUtils.copyProperties(item, invoiceLedgerExcelDto);
             return invoiceLedgerExcelDto;
@@ -210,11 +214,17 @@
         if(CollectionUtils.isEmpty(invoiceLedgerDto.getFileList())){
             throw new RuntimeException("缂哄皯鏂囦欢淇℃伅");
         }
+        QueryWrapper<InvoiceLedger> ledgerQueryWrapper = new QueryWrapper<>();
+        ledgerQueryWrapper.eq("invoice_registration_product_id", invoiceLedgerDto.getId());
+        InvoiceLedger invoiceLedger = invoiceLedgerMapper.selectOne(ledgerQueryWrapper);
+        if(ObjectUtils.isEmpty(invoiceLedger)){
+            throw new RuntimeException("寮�绁ㄥ彴璐︽湭鐧昏");
+        }
         List<FileVo> fileList = invoiceLedgerDto.getFileList();
         fileList.forEach(fileVo -> {
             InvoiceLedgerFile invoiceLedgerFile = new InvoiceLedgerFile();
             BeanUtils.copyProperties(fileVo, invoiceLedgerFile);
-            invoiceLedgerFile.setInvoiceLedgerId(invoiceLedgerDto.getId());
+            invoiceLedgerFile.setInvoiceLedgerId(invoiceLedger.getId());
             invoiceLedgerFileMapper.insert(invoiceLedgerFile);
         });
     }

--
Gitblit v1.9.3