From e03c24dfbf3f6dcbef8d44111b244597f91c4224 Mon Sep 17 00:00:00 2001
From: liuliang <401809302@qq.com>
Date: Mon, 25 Jul 2022 11:13:18 +0800
Subject: [PATCH] =?UTF-8?q?=E7=A4=BE=E4=BF=9D=E7=A6=8F=E5=88=A9=E5=AF=BC?=
 =?UTF-8?q?=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/SIImportServiceImpl.java     | 51 +++++++++++--------
 1 file changed, 29 insertions(+), 22 deletions(-)

diff --git a/src/com/engine/salary/service/impl/SIImportServiceImpl.java b/src/com/engine/salary/service/impl/SIImportServiceImpl.java
index 621fd39f9..8b45dfb8f 100644
--- a/src/com/engine/salary/service/impl/SIImportServiceImpl.java
+++ b/src/com/engine/salary/service/impl/SIImportServiceImpl.java
@@ -53,6 +53,7 @@ public class SIImportServiceImpl extends Service implements SIImportService {
     public XSSFWorkbook exportTemplate(InsuranceArchivesListParam param) {
         //工作簿list
         List<List<Object>> excelSheetData = new ArrayList<>();
+        Boolean exportData = param.getExportData();
 
         //工作簿名称
         String sheetName = SalaryI18nUtil.getI18nLabel(85368, "社保福利档案");        //表头
@@ -83,29 +84,35 @@ public class SIImportServiceImpl extends Service implements SIImportService {
 //        List<Map<String, Object>> datas = buildTableData(insuranceArchivesEmployeePOS);
         //表头
         excelSheetData.add(Arrays.asList(headers.toArray(new String[0])));
-        Map<String, Object> dataMap = getSIArchivesService(user).listPage(param);
-        List<Map<String, Object>> records = (List<Map<String, Object>>)dataMap.get("datas");
-        List<WeaTableColumn> columns = (List<WeaTableColumn>)dataMap.get("columns");
-        List<String> rowCloumn = Lists.newArrayList();
-        List<String> names = columns.stream().map(WeaTableColumn::getText).collect(Collectors.toList());
-        headers.forEach(header -> {
-            if (names.contains(header)){
-                columns.forEach(column -> {
-                    //用来判断columns中是否包含header
-                    if (header.equals(column.getText())){
-                        rowCloumn.add(column.getColumn());
-                    }
+        /**
+         * 导出数据
+         */
+        if (exportData != null && exportData){
+            Map<String, Object> dataMap = getSIArchivesService(user).listPage(param);
+            List<Map<String, Object>> records = (List<Map<String, Object>>)dataMap.get("datas");
+            List<WeaTableColumn> columns = (List<WeaTableColumn>)dataMap.get("columns");
+            List<String> rowCloumn = Lists.newArrayList();
+            List<String> names = columns.stream().map(WeaTableColumn::getText).collect(Collectors.toList());
+            headers.forEach(header -> {
+                if (names.contains(header)){
+                    columns.forEach(column -> {
+                        //用来判断columns中是否包含header
+                        if (header.equals(column.getText())){
+                            rowCloumn.add(column.getColumn());
+                        }
+
+                    });
+                }else {
+                    rowCloumn.add(null);
+                }
+            });
+            records.forEach(record -> {
+                List<Object> row = Lists.newArrayList();
+                rowCloumn.forEach(column -> row.add(record.get(column)));
+                excelSheetData.add(row);
+            });
+        }
 
-                });
-            }else {
-                rowCloumn.add(null);
-            }
-        });
-        records.forEach(record -> {
-            List<Object> row = Lists.newArrayList();
-            rowCloumn.forEach(column -> row.add(record.get(column)));
-            excelSheetData.add(row);
-        });
         // 3.表数据
         //工作簿数据
         return ExcelUtil.genWorkbookV2(excelSheetData, sheetName);