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