From 79a57e0844bdd3b5f06ebaac807a0896624be70f Mon Sep 17 00:00:00 2001
From: futian.liu <liufutianyoo@163.com>
Date: 星期五, 01 十二月 2023 10:24:06 +0800
Subject: [PATCH] 导出结构调整

---
 consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java |   50 +++++++++++++++++++++++++++++++++-----------------
 1 files changed, 33 insertions(+), 17 deletions(-)

diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java
index af3e13f..7769dae 100644
--- a/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java
+++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java
@@ -13,6 +13,7 @@
 import com.consum.base.pojo.RecordUserInfoVO;
 import com.consum.base.pojo.UseRecordSkuVO;
 import com.consum.base.pojo.dto.UseRecordDTO;
+import com.consum.base.pojo.excel.TransferExcelTemplate;
 import com.consum.base.pojo.query.TransferQry;
 import com.consum.base.pojo.request.LWhFormTransferParam;
 import com.consum.base.pojo.request.ProcureModelInfoParam;
@@ -41,10 +42,7 @@
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
-import java.io.IOException;
-import java.io.OutputStream;
 import java.lang.reflect.Field;
-import java.net.URLEncoder;
 import java.util.ArrayList;
 import java.util.Comparator;
 import java.util.HashMap;
@@ -301,29 +299,47 @@
         return ResponseValue.success();
     }
 
-    /**
-     * @Description 瀵煎嚭璋冩嫧鍑哄簱鍗�
-     * @Author 鍗㈠簡闃�
-     * @Date 2023/10/31
-     */
+
+    @ApiOperation(value = "璋冩嫧鍗曞鍑�", notes = "璋冩嫧鍗曞鍑�")
+    @ApiImplicitParams({
+        @ApiImplicitParam(name = "id", value = "璋冩嫧鍗昳d", required = true, dataType = "Long"),
+        @ApiImplicitParam(name = "type", value = "瀵煎嚭绫诲瀷 1 鍏ュ簱 2 鍑哄簱", required = true, dataType = "Integer"),
+    })
     @GetMapping("/list/export")
-    public void export(Long id, Integer type, HttpServletResponse response) throws IOException {
+    public void export(Long id, Integer type, HttpServletResponse response) throws Exception {
 
         TemplateExportParams params;
+        String fileName;
         if (type == 1) {
             params = new TemplateExportParams("import/璋冩嫧鍏ュ簱鍗�.xls");
+            fileName = "璋冩嫧鍏ュ簱鍗�";
         } else {
             params = new TemplateExportParams("import/璋冩嫧鍑哄簱鍗�.xls");
+            fileName = "璋冩嫧鍑哄簱鍗�";
         }
-        Map<String, Object> map = this.lWhFormTransferService.export(id, type);
+        params.setHeadingStartRow(2);
+        List<TransferExcelTemplate> export = this.lWhFormTransferService.export(id, type);
 
-        Workbook workbook = ExcelExportUtil.exportExcel(params, map);
-        try (OutputStream outputStream = response.getOutputStream()) {
-            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
-            response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode("璋冩嫧鍏ュ簱鍗�.xls", "utf-8"));
-            workbook.write(outputStream);
-            workbook.close();
-        }
+        int countNum = export.stream().filter(item -> item.getNum() != null).mapToInt(TransferExcelTemplate::getNum).sum();
+        int totalAmount = export.stream().filter(item -> item.getTotalAmount() != null).mapToInt(TransferExcelTemplate::getTotalAmount).sum();
+        Optional<TransferExcelTemplate> first = export.stream().findFirst();
+        TransferExcelTemplate entity = first.get();
+        String businessFormCode = entity.getBusinessFormCode();
+        Long createTime = entity.getCreateTime();
+        String operatorName = entity.getOperatorName();
+        String tenantName = entity.getTenantName();
+
+        Map<String, Object> map = new HashMap<>();
+        map.put("code", businessFormCode);
+        map.put("date", DateUtils.toShowDate(createTime));
+        map.put("tenantName", tenantName);
+        map.put("name", operatorName);
+        map.put("countNum", countNum);
+        map.put("totalAmount", totalAmount);
+
+        Workbook workbook = ExcelExportUtil.exportExcel(params, TransferExcelTemplate.class, export, map);
+        downLoadExcel(fileName, response, workbook);
+
     }
 
 

--
Gitblit v1.9.1