From 989abb480ea94940c1ebf80d6e98ee49fc52f030 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期六, 06 六月 2020 14:39:35 +0800 Subject: [PATCH] # --- src/main/java/com/zy/asrs/controller/MatCodeController.java | 65 ++++++++++++++++++++++++++------ 1 files changed, 52 insertions(+), 13 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..a75d060 100644 --- a/src/main/java/com/zy/asrs/controller/MatCodeController.java +++ b/src/main/java/com/zy/asrs/controller/MatCodeController.java @@ -1,6 +1,7 @@ package com.zy.asrs.controller; import com.alibaba.excel.EasyExcel; +import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.mapper.EntityWrapper; @@ -13,14 +14,19 @@ 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.BarcodeUtils; +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.imageio.ImageIO; +import javax.servlet.http.HttpServletResponse; +import java.awt.image.BufferedImage; import java.io.IOException; +import java.net.URLEncoder; import java.util.*; @RestController @@ -99,21 +105,54 @@ // 瀵煎嚭 @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)); + @ManagerAuth(value = ManagerAuth.Auth.NONE, memo = "鐗╂枡缂栫爜鏁版嵁瀵煎嚭") + 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) + .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) + .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(); + @ManagerAuth(memo = "鐗╂枡缂栫爜鏁版嵁瀵煎叆") + public R matCodeImport(MultipartFile file) throws IOException, InterruptedException { + Thread.sleep(2000); + EasyExcel.read(file.getInputStream(), MatCodeExcel.class, new MatCodeExcelListener()).sheet().doRead(); + return R.ok(); + } + + // todo resource + // 鐗╂枡鎵撳嵃 + @RequestMapping(value = "/macCode/print/auth") +// @ManagerAuth(memo = "鐗╂枡缂栫爜鎵撳嵃") + public R matCodePrint(HttpServletResponse response) throws IOException { + BufferedImage img = BarcodeUtils.encode("123456789"); + if (!ImageIO.write(img, "jpg", response.getOutputStream())) { + throw new IOException("Could not write an image of format jpg"); + } + response.getOutputStream().flush(); + response.getOutputStream().close(); return R.ok(); } -- Gitblit v1.9.1