From 1c3c68d3dc45e4d519088de17001284858de5eb9 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期六, 06 六月 2020 10:50:02 +0800 Subject: [PATCH] #导出 --- src/main/java/com/zy/asrs/controller/MatCodeController.java | 41 +++++++++++++++++++++++++++++------------ 1 files changed, 29 insertions(+), 12 deletions(-) diff --git a/src/main/java/com/zy/asrs/controller/MatCodeController.java b/src/main/java/com/zy/asrs/controller/MatCodeController.java index e5fd20f..b36c779 100644 --- a/src/main/java/com/zy/asrs/controller/MatCodeController.java +++ b/src/main/java/com/zy/asrs/controller/MatCodeController.java @@ -13,14 +13,16 @@ import com.core.common.R; import com.zy.asrs.entity.MatCode; import com.zy.asrs.service.MatCodeService; -import com.zy.common.utils.excel.EnInitData; -import com.zy.common.utils.excel.EnInitDataListener; +import com.zy.common.utils.excel.matcode.MatCodeExcel; +import com.zy.common.utils.excel.matcode.MatCodeExcelListener; import com.zy.common.web.BaseController; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; +import javax.servlet.http.HttpServletResponse; import java.io.IOException; +import java.net.URLEncoder; import java.util.*; @RestController @@ -99,21 +101,36 @@ // 瀵煎嚭 @RequestMapping(value = "/matCode/export/auth") - @ManagerAuth - public R export(@RequestBody JSONObject param){ - List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class); - EntityWrapper<MatCode> wrapper = new EntityWrapper<>(); - Map<String, Object> map = excludeTrash(param.getJSONObject("matCode")); - convert(map, wrapper); - List<MatCode> list = matCodeService.selectList(wrapper); - return R.ok(exportSupport(list, fields)); + public void export(@RequestParam(required = false) String fileName, + @RequestParam(required = false) Integer rowCount, + HttpServletResponse response) throws Exception { + List<MatCode> list = matCodeService.selectList(new EntityWrapper<>()); + List<MatCodeExcel> excels = new ArrayList<>(); + for (MatCode matCode : list) { + MatCodeExcel excel = new MatCodeExcel(); + excel.setMatNo(matCode.getMatNo()); + excel.setBarcode(matCode.getBarcode()); + excel.setMatName(matCode.getMatName()); + excel.setStr1(matCode.getStr1()); + excel.setStr2(matCode.getStr2()); + excels.add(excel); + if (rowCount != null && excels.size() >= rowCount){ + break; + } + } + response.setContentType("application/vnd.ms-excel"); + response.setCharacterEncoding("utf-8"); + fileName = URLEncoder.encode(Cools.isEmpty(fileName)?"鐗╂枡缂栫爜":fileName, "UTF-8"); + response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx"); + EasyExcel.write(response.getOutputStream(), MatCodeExcel.class).sheet("琛�1").doWrite(excels); } // 瀵煎叆 @RequestMapping(value = "/matCode/import/auth") @ManagerAuth - public R matCodeImport(MultipartFile file) throws IOException { - EasyExcel.read(file.getInputStream(), EnInitData.class, new EnInitDataListener()).sheet().doRead(); + public R matCodeImport(MultipartFile file) throws IOException, InterruptedException { + Thread.sleep(2000); + EasyExcel.read(file.getInputStream(), MatCodeExcel.class, new MatCodeExcelListener()).sheet().doRead(); return R.ok(); } -- Gitblit v1.9.1