From ff2d77915707227283808edc08daeb411687d44f Mon Sep 17 00:00:00 2001
From: futian.liu <liufutianyoo@163.com>
Date: 星期三, 29 十一月 2023 17:29:17 +0800
Subject: [PATCH] 调拨入库导出

---
 consum-base/src/main/java/com/consum/base/controller/LWhFormTransferController.java |   38 ++++++++++++++++++++++++++------------
 1 files changed, 26 insertions(+), 12 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 d7c0e2c..dab1821 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
@@ -1,5 +1,7 @@
 package com.consum.base.controller;
 
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.entity.TemplateExportParams;
 import cn.hutool.core.util.ReflectUtil;
 import com.consum.base.BaseController;
 import com.consum.base.core.utils.CommonUtil;
@@ -14,6 +16,7 @@
 import com.consum.base.pojo.query.TransferQry;
 import com.consum.base.pojo.request.LWhFormTransferParam;
 import com.consum.base.pojo.request.ProcureModelInfoParam;
+import com.consum.base.pojo.request.RecordInfoParam;
 import com.consum.base.pojo.response.FormTransferVO;
 import com.consum.base.pojo.response.FromTransferTemplateInfoVO;
 import com.consum.base.pojo.response.LWHFromTransferExtendVO;
@@ -30,7 +33,6 @@
 import com.consum.model.po.LWhFormTransfer;
 import com.consum.model.po.LWhProcureModelUser;
 import com.consum.model.po.LWhProcureModelUserRecord;
-import com.consum.model.vo.LWhFormOutputVo;
 import com.iplatform.model.po.S_user_core;
 import com.walker.db.page.GenericPager;
 import com.walker.infrastructure.utils.CollectionUtils;
@@ -40,14 +42,19 @@
 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.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.stream.Collectors;
+import javax.servlet.http.HttpServletResponse;
 import org.apache.commons.compress.utils.Lists;
+import org.apache.poi.ss.usermodel.Workbook;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -302,12 +309,18 @@
      * @Date 2023/10/31
      */
     @GetMapping("/export")
-    public ResponseValue export(Long id) {
-        if (id == null) {
-            return ResponseValue.error("璋冩嫧鍗昳d涓虹┖");
+    public void export(Long id, HttpServletResponse response) throws IOException {
+
+        TemplateExportParams params = new TemplateExportParams("import/璋冩嫧鍏ュ簱鍗�.xlsx");
+        Map<String, Object> map = this.lWhFormTransferService.export(id);
+
+        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("璋冩嫧鍏ュ簱鍗�.xlsx", "utf-8"));
+            workbook.write(outputStream);
+            workbook.close();
         }
-        LWhFormOutputVo vo = this.lWhFormTransferService.export(id, this.getCurrentUser());
-        return ResponseValue.success(vo);
     }
 
 
@@ -344,19 +357,20 @@
     @PostMapping("/useInfo/update")
     public ResponseValue infoUpdate() {
 
-        List<ProcureModelInfoParam> param = CommonUtil.getObjFromReqBody(List.class);
-        List<ProcureModelInfoParam> procureModelInfoParam = Lists.newArrayList();
-        procureModelInfoParam.addAll(param);
+        RecordInfoParam param = CommonUtil.getObjFromReqBody(RecordInfoParam.class);
+        RecordInfoParam recordInfoParam = new RecordInfoParam();
+        CommonUtil.copyProperties(param, recordInfoParam);
+        param = recordInfoParam;
 
         S_user_core currentUser = this.getCurrentUser();
         if (currentUser == null) {
             return ResponseValue.error("鐧诲綍鐢ㄦ埛淇℃伅涓嶅瓨鍦�");
         }
-        if (CollectionUtils.isEmpty(procureModelInfoParam)) {
+        if (CollectionUtils.isEmpty(param.getRecordInfoList())) {
             return ResponseValue.error("鍙傛暟閿欒");
         }
 
-        Map<Long, List<ProcureModelInfoParam>> collect = procureModelInfoParam.stream()
+        Map<Long, List<ProcureModelInfoParam>> collect = param.getRecordInfoList().stream()
             .collect(Collectors.groupingBy(ProcureModelInfoParam::getBaseGoodModelId));
         for (Map.Entry<Long, List<ProcureModelInfoParam>> entry : collect.entrySet()) {
             Long baseGoodModelId = entry.getKey();
@@ -375,7 +389,7 @@
             lWhProcureModelUserRecordService.insert(lWhProcureModelUserRecord);
 
             List<LWhProcureModelUser> procureModelUserList = Lists.newArrayList();
-            if (CollectionUtils.isEmpty(procureModelInfoList)) {
+            if (!CollectionUtils.isEmpty(procureModelInfoList)) {
                 for (ProcureModelInfoParam item : procureModelInfoList) {
                     LWhProcureModelUser lWhProcureModelUser = new LWhProcureModelUser();
                     lWhProcureModelUser.setId(IdUtil.generateId());

--
Gitblit v1.9.1