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/LWhFormProcureController.java |   46 +++++++++++++++++++++++++++++++---------------
 1 files changed, 31 insertions(+), 15 deletions(-)

diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java
index f621b84..15235d3 100644
--- a/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java
+++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormProcureController.java
@@ -14,6 +14,7 @@
 import com.consum.base.pojo.LWhFormProcureGoodsInfoParam;
 import com.consum.base.pojo.LWhFormProcureParam;
 import com.consum.base.pojo.LWhProcureModelParam;
+import com.consum.base.pojo.excel.ProcureExcelTemplate;
 import com.consum.base.pojo.query.FormProcureQry;
 import com.consum.base.pojo.response.FormProcureVO;
 import com.consum.base.pojo.response.FromProcureTemplateInfoVO;
@@ -38,21 +39,20 @@
 import com.iplatform.model.po.S_user_core;
 import com.walker.db.page.GenericPager;
 import com.walker.infrastructure.utils.CollectionUtils;
+import com.walker.infrastructure.utils.DateUtils;
 import com.walker.infrastructure.utils.NumberGenerator;
 import com.walker.web.ResponseValue;
 import io.swagger.annotations.Api;
 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.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.Set;
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
@@ -421,25 +421,41 @@
     }
 
     @ApiOperation(value = "閲囪喘鍗曞鍑�", notes = "閲囪喘鍗曞鍑�")
+    @ApiImplicitParams({
+        @ApiImplicitParam(name = "id", value = "閲囪喘鍗昳d", required = true, dataType = "Long", paramType = "query")
+    })
     @GetMapping("/list/export")
-    public void export(Long id, HttpServletResponse response) throws IOException {
+    public void export(Long id, HttpServletResponse response) throws Exception {
         TemplateExportParams params = new TemplateExportParams(
             "import/閲囪喘鍏ュ簱鍗�.xls");
-
+        params.setHeadingStartRow(2);
         FinSysTenantUser sysInfo = this.getSysInfo();
-        String userName = sysInfo.getUserName();
+        if (sysInfo == null) {
+            throw new RuntimeException("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
+        }
 
-        Map<String, Object> map = lWhFormProcureService.getExportList(id, userName);
-        if (CollectionUtils.isEmpty(map)) {
+        List<ProcureExcelTemplate> exportList = lWhFormProcureService.getExportList(id);
+        if (CollectionUtils.isEmpty(exportList)) {
             throw new RuntimeException("鏁版嵁涓虹┖");
         }
-        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 = exportList.stream().filter(item -> item.getNum() != null).mapToInt(ProcureExcelTemplate::getNum).sum();
+        int totalAmount = exportList.stream().filter(export -> export.getTotalAmount() != null).mapToInt(ProcureExcelTemplate::getTotalAmount).sum();
+        Optional<ProcureExcelTemplate> first = exportList.stream().findFirst();
+        ProcureExcelTemplate templateExcelExport = first.get();
+        String businessFormCode = templateExcelExport.getBusinessFormCode();
+        Long createTime = templateExcelExport.getCreateTime();
+        String operatorName = templateExcelExport.getOperatorName();
+
+        Map<String, Object> map = new HashMap<>();
+        map.put("code", businessFormCode);
+        map.put("date", DateUtils.toShowDate(createTime));
+        map.put("name", operatorName);
+        map.put("countNum", countNum);
+        map.put("totalAmount", totalAmount);
+
+        Workbook workbook = ExcelExportUtil.exportExcel(params, ProcureExcelTemplate.class, exportList, map);
+        downLoadExcel("閲囪喘鍏ュ簱鍗�", response, workbook);
+
     }
 }

--
Gitblit v1.9.1