From 43a322430e7e6dcea64242bec6918037da160986 Mon Sep 17 00:00:00 2001
From: futian.liu <liufutianyoo@163.com>
Date: 星期一, 04 十二月 2023 10:50:23 +0800
Subject: [PATCH] 分转元

---
 consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java |   65 +++++++++++++++-----------------
 1 files changed, 30 insertions(+), 35 deletions(-)

diff --git a/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java b/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java
index 4cceb53..5ff38e9 100644
--- a/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java
+++ b/consum-base/src/main/java/com/consum/base/controller/LWhFormOutputController.java
@@ -9,6 +9,7 @@
 import com.consum.base.core.utils.MapperUtil;
 import com.consum.base.core.utils.PageUtil;
 import com.consum.base.pojo.LWhFormOutputParam;
+import com.consum.base.pojo.excel.OutputExcelTemplate;
 import com.consum.base.pojo.query.LWhFormOutputQry;
 import com.consum.base.pojo.response.FormOutputGoodsVO;
 import com.consum.base.pojo.response.FormOutputTemplateInfoVO;
@@ -23,20 +24,18 @@
 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.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.ByteArrayOutputStream;
-import java.io.IOException;
 import java.io.InputStream;
-import java.io.OutputStream;
 import java.lang.reflect.Field;
-import java.net.URLEncoder;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
 import org.apache.commons.compress.utils.Lists;
@@ -161,6 +160,7 @@
         LWhFormOutput lWhFormOutput = lWhFormOutputService.get(new LWhFormOutput(id));
         LWhFormOutputExtendVO lWhFormOutputExtendVO = new LWhFormOutputExtendVO();
         BeanUtils.copyProperties(lWhFormOutput, lWhFormOutputExtendVO);
+        lWhFormOutputExtendVO.setDoc(lWhFormOutput.getOutputDoc());
 
         String sql = "SELECT bgt.id, CATEGORY_ID, CATEGORY_NAME, GOODS_NAME "
             + "FROM l_wh_procure_model pm LEFT JOIN base_goods_models bgm ON bgm.id = pm.BASE_GOODS_MODELS_ID "
@@ -208,51 +208,46 @@
 
 
     @ApiOperation(value = "鍑哄簱鍗曞鍑�", notes = "鍑哄簱鍗曞鍑�")
+    @ApiImplicitParam(name = "id", value = "鍑哄簱鍗昳d", required = true, dataType = "Long", paramType = "query")
     @GetMapping("/list/export")
-    public void export(Long id, HttpServletResponse response) throws IOException {
-        ClassPathResource classPathResource = new ClassPathResource("import/閲囪喘閫�璐у崟.xls");
+    public void export(Long id, HttpServletResponse response) throws Exception {
+        FinSysTenantUser sysInfo = this.getSysInfo();
+        if (sysInfo == null) {
+            throw new RuntimeException("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
+        }
+        ClassPathResource classPathResource = new ClassPathResource("import/鍑哄簱鍗�.xls");
         // 鑾峰彇鏂囦欢杈撳叆娴�
         InputStream inputStream = classPathResource.getInputStream();
         Workbook wb = new HSSFWorkbook(inputStream);
         TemplateExportParams params = new TemplateExportParams();
         params.setTemplateWb(wb);
+        params.setHeadingStartRow(2);
 
-        FinSysTenantUser sysInfo = this.getSysInfo();
-        String userName = sysInfo.getUserName();
+        List<OutputExcelTemplate> exportList = lWhFormOutputService.getExportList(id);
 
-        Map<String, Object> map = lWhFormOutputService.getExportList(id, userName);
-        if (CollectionUtils.isEmpty(map)) {
+        if (CollectionUtils.isEmpty(exportList)) {
             throw new RuntimeException("鏁版嵁涓虹┖");
         }
+        int countNum = exportList.stream().filter(item -> item.getNum() != null).mapToInt(OutputExcelTemplate::getNum).sum();
+        double totalAmount = exportList.stream().filter(item -> item.getTotalAmount() != null).mapToDouble(OutputExcelTemplate::getAmount).sum();
+        Optional<OutputExcelTemplate> first = exportList.stream().findFirst();
+        OutputExcelTemplate templateExcelExport = first.get();
+        String businessFormCode = templateExcelExport.getBusinessFormCode();
+        Long createTime = templateExcelExport.getCreateTime();
+        String operatorName = templateExcelExport.getOperatorName();
 
-        Workbook workbook = ExcelExportUtil.exportExcel(params, map);
-        downLoadExcel("111", response, workbook);
+        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);
 
-//        try (OutputStream outputStream = response.getOutputStream()) {
-//            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
-//            response.setHeader("Content-disposition", "attachment;filename=" + URLEncoder.encode("閲囪喘閫�璐у崟1.xlsx", "utf-8"));
-//            workbook.write(outputStream);
-//            workbook.close();
-//        }
+        Workbook workbook = ExcelExportUtil.exportExcel(params, OutputExcelTemplate.class, exportList, map);
+        downLoadExcel("鍑哄簱鍗�", response, workbook);
+
     }
 
-    private void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) throws IOException {
-        try (OutputStream out = response.getOutputStream()) {
-            response.setCharacterEncoding("UTF-8");
-            response.setHeader("content-Type", "application/vnd.ms-excel");
-            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName + ".xls", "UTF-8"));
-            ByteArrayOutputStream baos = new ByteArrayOutputStream();
-            workbook.write(baos);
-            response.setHeader("Content-Length", String.valueOf(baos.size()));
-            out.write(baos.toByteArray());
-        } catch (Exception e) {
-            try {
-                throw new Exception(e.getMessage());
-            } catch (Exception e1) {
-                e1.printStackTrace();
-            }
-        }
-    }
 
 }
 

--
Gitblit v1.9.1