From cab8f5284fd5bef35584195246f2f5dc7ad4fd38 Mon Sep 17 00:00:00 2001
From: wujianwei <wjw@11.com>
Date: 星期三, 24 十二月 2025 15:26:24 +0800
Subject: [PATCH] 修改导入模板

---
 service/src/main/java/com/ruoyi/cwgl/service/impl/ReceivableFeeManagementServiceImpl.java |  259 ++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 243 insertions(+), 16 deletions(-)

diff --git a/service/src/main/java/com/ruoyi/cwgl/service/impl/ReceivableFeeManagementServiceImpl.java b/service/src/main/java/com/ruoyi/cwgl/service/impl/ReceivableFeeManagementServiceImpl.java
index 311e4e0..4df87cb 100644
--- a/service/src/main/java/com/ruoyi/cwgl/service/impl/ReceivableFeeManagementServiceImpl.java
+++ b/service/src/main/java/com/ruoyi/cwgl/service/impl/ReceivableFeeManagementServiceImpl.java
@@ -1,13 +1,22 @@
 package com.ruoyi.cwgl.service.impl;
 
+import java.io.File;
+import java.io.OutputStream;
+import java.lang.reflect.Field;
+import java.nio.file.Files;
+import java.nio.file.Paths;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
 import java.util.*;
 
+import com.ruoyi.common.annotation.Excel;
 import com.ruoyi.common.enums.SystemDataNoEnum;
 import com.ruoyi.common.exception.ServiceException;
 import com.ruoyi.common.utils.DateUtils;
 import com.ruoyi.common.utils.SecurityUtils;
+import com.ruoyi.common.utils.file.DownloadExportUtil;
+import com.ruoyi.common.utils.file.DownloadExportUtil.ExprotStatus;
+import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.cwgl.domain.ReceivableFeeDetail;
 import com.ruoyi.cwgl.domain.ReceivableFeeManagementLog;
 import com.ruoyi.cwgl.domain.vo.ReceivableFeeStatisticsVo;
@@ -26,6 +35,9 @@
 import com.ruoyi.common.annotation.DataSource;
 import com.ruoyi.common.enums.DataSourceType;
 import com.ruoyi.common.core.service.BaseService;
+import com.ruoyi.common.config.RuoYiConfig;
+import com.ruoyi.common.exception.UtilException;
+import com.ruoyi.common.core.redis.RedisCache;
 
 import com.ruoyi.cwgl.mapper.ReceivableFeeManagementMapper;
 import com.ruoyi.cwgl.domain.ReceivableFeeManagement;
@@ -35,6 +47,8 @@
 import com.ruoyi.common.core.text.Convert;
 
 import javax.annotation.Resource;
+import org.apache.poi.ss.usermodel.*;
+import org.apache.poi.xssf.streaming.SXSSFWorkbook;
 
 /**
  * 搴旀敹璐圭敤绠$悊Service涓氬姟灞傚鐞�
@@ -66,6 +80,9 @@
 
     @Autowired
     private IReceivableFeeManagementLogService logService;
+
+    @Autowired
+    private RedisCache redisCache;
 
     /**
      * 鏌ヨ搴旀敹璐圭敤绠$悊
@@ -571,14 +588,18 @@
      * @return 缁撴灉
      */
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public String importReceivableFee(List<ReceivableFeeManagement> receivableFeeList, String operName) {
         if (receivableFeeList == null || receivableFeeList.isEmpty()) {
-            return "瀵煎叆鏁版嵁涓嶈兘涓虹┖锛�";
+            throw new ServiceException("瀵煎叆搴旀敹璐圭敤鏁版嵁涓嶈兘涓虹┖锛�");
         }
         int successNum = 0;
         int failureNum = 0;
         StringBuilder successMsg = new StringBuilder();
         StringBuilder failureMsg = new StringBuilder();
+        
+        // 鑾峰彇褰撳墠鏃ユ湡
+        Date now = DateUtils.getNowDate();
         
         for (ReceivableFeeManagement receivableFee : receivableFeeList) {
             try {
@@ -587,7 +608,7 @@
                 receivableFee.setSystemNo(noByKey);
                 
                 // 璁剧疆鍒涘缓鏃堕棿鍜屽垱寤轰汉
-                receivableFee.setCreateTime(DateUtils.getNowDate());
+                receivableFee.setCreateTime(now);
                 receivableFee.setCreateBy(operName);
                 
                 // 璁$畻搴旀敹閲戦瀛楃涓叉弿杩�
@@ -596,19 +617,24 @@
                     receivableFee.setReceivableAmountStr(receivableAmountStr);
                 }
                 
-                // 鏂板璁板綍锛堜笉鍋氬瓨鍦ㄦ�у垽鏂級
-                int result = insertReceivableFeeManagement(receivableFee);
+                // 鎻掑叆涓昏〃鏁版嵁
+                receivableFeeManagementMapper.insertReceivableFeeManagement(receivableFee);
                 
-                if (result > 0) {
-                    successNum++;
-                    successMsg.append("<br/>" + successNum + ". 绯荤粺缂栧彿 " + receivableFee.getSystemNo() + " 瀵煎叆鎴愬姛");
-                } else {
-                    failureNum++;
-                    failureMsg.append("<br/>" + failureNum + ". 绯荤粺缂栧彿 " + receivableFee.getSystemNo() + " 瀵煎叆澶辫触");
+                // 澶勭悊鏄庣粏鏁版嵁锛堝鏋滄湁锛�
+                if (receivableFee.getReceivableFeeDetailList() != null && !receivableFee.getReceivableFeeDetailList().isEmpty()) {
+                    for (ReceivableFeeDetail detail : receivableFee.getReceivableFeeDetailList()) {
+                        detail.setReceivableFeeId(receivableFee.getId());
+                        detail.setCreateBy(operName);
+                        detail.setCreateTime(now);
+                        receivableFeeDetailService.insertReceivableFeeDetail(detail);
+                    }
                 }
+                
+                successNum++;
+                successMsg.append("<br/>" + successNum + ".搴旀敹璐圭敤 " + receivableFee.getSystemNo() + " 瀵煎叆鎴愬姛");
             } catch (Exception e) {
                 failureNum++;
-                String msg = "<br/>" + failureNum + ". 绯荤粺缂栧彿 " + (receivableFee.getSystemNo() != null ? receivableFee.getSystemNo() : "鏈煡") + " 瀵煎叆澶辫触锛�";
+                String msg = "<br/>" + failureNum + ".搴旀敹璐圭敤 " + receivableFee.getSystemNo() + " 瀵煎叆澶辫触锛�";
                 failureMsg.append(msg + e.getMessage());
                 logger.error(msg, e);
             }
@@ -624,7 +650,7 @@
     }
 
     /**
-     * 涓嬭浇瀵煎叆妯℃澘
+     * 涓嬭浇瀵煎叆妯℃澘锛堝寘鍚玈heet1涓昏〃鍜孲heet2鏄庣粏琛級
      * 
      * @param exportKey 瀵煎嚭鍔熻兘鐨勫敮涓�鏍囪瘑
      */
@@ -632,9 +658,210 @@
     @Async
     @Override
     public void importTemplate(String exportKey) {
-        super.export(ReceivableFeeManagement.class, exportKey, "receivableFeeManagementTemplate", (pageNum) -> {
-            // 杩斿洖绌哄垪琛紝鍙鍑烘ā鏉跨粨鏋�
-            return new ArrayList<ReceivableFeeManagement>();
-        });
+        String fileName = ExcelUtil.encodeFileName("搴旀敹璐圭敤瀵煎叆妯℃澘");
+
+        // 璁剧疆褰撳墠浠诲姟涓�"涓嬭浇涓�"鐘舵��
+        DownloadExportUtil.deleteDownloadFile(redisCache, exportKey, ExprotStatus.XZZ.getStatus());
+        
+        try {
+            // 鍒涘缓澶歋heet瀵煎叆妯℃澘
+            createMultiSheetImportTemplate(exportKey,fileName);
+            
+            // 璁剧疆涓嬭浇瀹屾垚鐘舵��
+            DownloadExportUtil.setDownloadFile(redisCache, exportKey, fileName);
+            logger.info("瀵煎叆妯℃澘瀵煎嚭瀹屾垚: {}, file: {}", exportKey, fileName);
+        } catch (Exception e) {
+            logger.error("瀵煎叆妯℃澘瀵煎嚭澶辫触: {}, error: {}", exportKey, e.getMessage(), e);
+            DownloadExportUtil.deleteDownloadFile(redisCache, exportKey, ExprotStatus.XZYC.getStatus()); // 璁剧疆澶辫触鐘舵��
+            throw e;
+        }
+    }
+    
+    /**
+     * 鍒涘缓澶歋heet瀵煎叆妯℃澘
+     * 鍖呭惈Sheet1锛氬簲鏀惰垂鐢ㄤ富琛ㄦā鏉�
+     * 鍖呭惈Sheet2锛氬簲鏀惰垂鐢ㄦ槑缁嗚〃妯℃澘
+     *
+     * @param exportKey 瀵煎嚭鍔熻兘鐨勫敮涓�鏍囪瘑
+     * @param fileName
+     */
+    private void createMultiSheetImportTemplate(String exportKey, String fileName) {
+
+        
+        try {
+            // 鍒涘缓鏂扮殑宸ヤ綔绨匡紝鍖呭惈涓や釜Sheet
+            SXSSFWorkbook workbook = new SXSSFWorkbook(1000);
+            
+            // 鍒涘缓涓昏〃Sheet
+            Sheet mainSheet = workbook.createSheet("搴旀敹璐圭敤涓昏〃");
+            createMainSheetTemplateWithExcelUtil(mainSheet);
+            
+            // 鍒涘缓鏄庣粏琛⊿heet
+            Sheet detailSheet = workbook.createSheet("搴旀敹璐圭敤鏄庣粏");
+            createDetailSheetTemplateWithExcelUtil(detailSheet);
+            
+            // 淇濆瓨宸ヤ綔绨垮埌鏂囦欢
+            try (OutputStream out = Files.newOutputStream(Paths.get(getAbsoluteFile(fileName)))) {
+                workbook.write(out);
+            }
+            
+            workbook.close();
+            
+        } catch (Exception e) {
+            logger.error("鍒涘缓澶歋heet瀵煎叆妯℃澘澶辫触: {}", e.getMessage(), e);
+            throw new UtilException("瀵煎叆妯℃澘鐢熸垚澶辫触锛�");
+        }
+    }
+    
+    /**
+     * 浣跨敤ExcelUtil鍒涘缓涓昏〃Sheet妯℃澘
+     * 
+     * @param sheet Sheet瀵硅薄
+     */
+    private void createMainSheetTemplateWithExcelUtil(Sheet sheet) {
+        try {
+            // 浣跨敤ExcelUtil鍒涘缓涓昏〃妯℃澘
+            ExcelUtil<ReceivableFeeManagement> mainSheetUtil = new ExcelUtil<>(ReceivableFeeManagement.class);
+            
+            // 鍒濆鍖朎xcelUtil
+            mainSheetUtil.initialize("搴旀敹璐圭敤涓昏〃", "搴旀敹璐圭敤涓昏〃瀵煎叆妯℃澘锛堣鍕夸慨鏀瑰簭鍙峰垪锛�", Excel.Type.EXPORT);
+            
+            // 閫氳繃鍙嶅皠鑾峰彇workbook瀛楁
+            Field workbookField = ExcelUtil.class.getDeclaredField("workbook");
+            workbookField.setAccessible(true);
+            SXSSFWorkbook excelUtilWorkbook = (SXSSFWorkbook) workbookField.get(mainSheetUtil);
+            
+            // 鑾峰彇ExcelUtil鐢熸垚鐨凷heet
+            Sheet generatedSheet = excelUtilWorkbook.getSheetAt(0);
+            
+            // 澶嶅埗鏍囬琛屽埌鐩爣Sheet
+            copyRow(generatedSheet.getRow(0), sheet.createRow(0));
+            
+            // 澶嶅埗琛ㄥご琛屽埌鐩爣Sheet
+            copyRow(generatedSheet.getRow(1), sheet.createRow(1));
+            
+            // 娣诲姞璇存槑琛�
+            Row instructionRow = sheet.createRow(2);
+            Cell instructionCell = instructionRow.createCell(0);
+            instructionCell.setCellValue("璇存槑锛氬簭鍙风敤浜庡叧鑱斾富琛ㄥ拰鏄庣粏琛紝璇风‘淇濅富琛ㄥ拰鏄庣粏琛ㄤ腑鐨勫簭鍙蜂竴鑷�");
+            
+            // 璁剧疆鍒楀
+            for (int i = 0; i < generatedSheet.getRow(1).getLastCellNum(); i++) {
+                sheet.setColumnWidth(i, generatedSheet.getColumnWidth(i));
+            }
+            
+        } catch (Exception e) {
+            logger.error("浣跨敤ExcelUtil鍒涘缓涓昏〃妯℃澘澶辫触: {}", e.getMessage(), e);
+            throw new UtilException("涓昏〃妯℃澘鐢熸垚澶辫触锛�");
+        }
+    }
+    
+    /**
+     * 浣跨敤ExcelUtil鍒涘缓鏄庣粏琛⊿heet妯℃澘
+     * 
+     * @param sheet Sheet瀵硅薄
+     */
+    private void createDetailSheetTemplateWithExcelUtil(Sheet sheet) {
+        try {
+            // 浣跨敤ExcelUtil鍒涘缓鏄庣粏琛ㄦā鏉�
+            ExcelUtil<ReceivableFeeDetail> detailSheetUtil = new ExcelUtil<>(ReceivableFeeDetail.class);
+            
+            // 鍒濆鍖朎xcelUtil
+            detailSheetUtil.initialize("搴旀敹璐圭敤鏄庣粏", "搴旀敹璐圭敤鏄庣粏琛ㄥ鍏ユā鏉匡紙搴忓彿涓庝富琛ㄥ搴旓級", Excel.Type.EXPORT);
+            
+            // 閫氳繃鍙嶅皠鑾峰彇workbook瀛楁
+            Field workbookField = ExcelUtil.class.getDeclaredField("workbook");
+            workbookField.setAccessible(true);
+            SXSSFWorkbook excelUtilWorkbook = (SXSSFWorkbook) workbookField.get(detailSheetUtil);
+            
+            // 鑾峰彇ExcelUtil鐢熸垚鐨凷heet
+            Sheet generatedSheet = excelUtilWorkbook.getSheetAt(0);
+            
+            // 澶嶅埗鏍囬琛屽埌鐩爣Sheet
+            copyRow(generatedSheet.getRow(0), sheet.createRow(0));
+            
+            // 澶嶅埗琛ㄥご琛屽埌鐩爣Sheet
+            copyRow(generatedSheet.getRow(1), sheet.createRow(1));
+            
+            // 娣诲姞璇存槑琛�
+            Row instructionRow = sheet.createRow(2);
+            Cell instructionCell = instructionRow.createCell(0);
+            instructionCell.setCellValue("璇存槑锛氬簭鍙峰繀椤讳笌涓昏〃涓殑搴忓彿涓�鑷达紝鐢ㄤ簬鍏宠仈涓昏〃鍜屾槑缁嗚〃鏁版嵁");
+            
+            // 璁剧疆鍒楀
+            for (int i = 0; i < generatedSheet.getRow(1).getLastCellNum(); i++) {
+                sheet.setColumnWidth(i, generatedSheet.getColumnWidth(i));
+            }
+            
+        } catch (Exception e) {
+            logger.error("浣跨敤ExcelUtil鍒涘缓鏄庣粏琛ㄦā鏉垮け璐�: {}", e.getMessage(), e);
+            throw new UtilException("鏄庣粏琛ㄦā鏉跨敓鎴愬け璐ワ紒");
+        }
+    }
+    
+    /**
+     * 澶嶅埗琛屽唴瀹�
+     * 
+     * @param sourceRow 婧愯
+     * @param targetRow 鐩爣琛�
+     */
+    private void copyRow(Row sourceRow, Row targetRow) {
+        if (sourceRow == null) return;
+        
+        targetRow.setHeight(sourceRow.getHeight());
+        
+        for (int i = 0; i < sourceRow.getLastCellNum(); i++) {
+            Cell sourceCell = sourceRow.getCell(i);
+            if (sourceCell != null) {
+                Cell targetCell = targetRow.createCell(i);
+                copyCell(sourceCell, targetCell);
+            }
+        }
+    }
+    
+    /**
+     * 澶嶅埗鍗曞厓鏍煎唴瀹瑰拰鏍峰紡
+     * 
+     * @param sourceCell 婧愬崟鍏冩牸
+     * @param targetCell 鐩爣鍗曞厓鏍�
+     */
+    private void copyCell(Cell sourceCell, Cell targetCell) {
+        // 澶嶅埗鍗曞厓鏍兼牱寮�
+        targetCell.setCellStyle(sourceCell.getCellStyle());
+        
+        // 澶嶅埗鍗曞厓鏍煎��
+        switch (sourceCell.getCellType()) {
+            case STRING:
+                targetCell.setCellValue(sourceCell.getStringCellValue());
+                break;
+            case NUMERIC:
+                targetCell.setCellValue(sourceCell.getNumericCellValue());
+                break;
+            case BOOLEAN:
+                targetCell.setCellValue(sourceCell.getBooleanCellValue());
+                break;
+            case FORMULA:
+                targetCell.setCellFormula(sourceCell.getCellFormula());
+                break;
+            default:
+                targetCell.setCellValue(sourceCell.getStringCellValue());
+        }
+    }
+    
+    /**
+     * 鑾峰彇鏂囦欢鐨勭粷瀵硅矾寰�
+     * 
+     * @param fileName 鏂囦欢鍚�
+     * @return 缁濆璺緞
+     */
+    private String getAbsoluteFile(String fileName) {
+        String downloadPath = RuoYiConfig.getDownloadPath();
+        File desc = new File(downloadPath + File.separator + fileName);
+        if (!desc.exists()) {
+            if (!desc.getParentFile().exists()) {
+                desc.getParentFile().mkdirs();
+            }
+        }
+        return desc.getAbsolutePath();
     }
 }
\ No newline at end of file

--
Gitblit v1.8.0