From a78f80c36b64a69b1293289dad908abcb12858a8 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期二, 30 六月 2020 13:39:20 +0800 Subject: [PATCH] # --- src/main/java/com/zy/common/web/FileController.java | 36 +++++++++++++++++++++++++----------- 1 files changed, 25 insertions(+), 11 deletions(-) diff --git a/src/main/java/com/zy/common/web/FileController.java b/src/main/java/com/zy/common/web/FileController.java index 5dbea13..3074912 100644 --- a/src/main/java/com/zy/common/web/FileController.java +++ b/src/main/java/com/zy/common/web/FileController.java @@ -1,36 +1,39 @@ package com.zy.common.web; import com.alibaba.excel.EasyExcel; -import com.zy.common.utils.excel.EnInitData; -import com.zy.common.utils.excel.matcode.MatCodeExcelListener; -import com.zy.common.utils.excel.ExcelUtils; +import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.core.common.R; -import org.springframework.stereotype.Controller; -import org.springframework.web.bind.annotation.GetMapping; -import org.springframework.web.bind.annotation.PostMapping; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; +import com.zy.asrs.entity.MatCode; +import com.zy.asrs.service.MatCodeService; +import com.zy.common.utils.excel.matcode.MatCodeExcel; +import com.zy.common.utils.excel.matcode.MatCodeExcelListener; +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.ArrayList; +import java.util.List; /** * todo * Created by vincent on 2019-11-25 */ -@Controller +@RestController @RequestMapping("excel/") public class FileController { + @Autowired + private MatCodeService matCodeService; /** * 鏂囦欢涓婁紶 */ @PostMapping("upload") @ResponseBody public R upload(MultipartFile file) throws IOException { - EasyExcel.read(file.getInputStream(), EnInitData.class, new MatCodeExcelListener()).sheet().doRead(); + EasyExcel.read(file.getInputStream(), MatCodeExcel.class, new MatCodeExcelListener()).sheet().doRead(); return R.ok(); } @@ -39,13 +42,24 @@ */ @GetMapping("download") public void download(HttpServletResponse response) throws IOException { + 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()); + excels.add(excel); + } // 杩欓噷娉ㄦ剰 鏈夊悓瀛﹀弽搴斾娇鐢╯wagger 浼氬鑷村悇绉嶉棶棰橈紝璇风洿鎺ョ敤娴忚鍣ㄦ垨鑰呯敤postman response.setContentType("application/vnd.ms-excel"); response.setCharacterEncoding("utf-8"); // 杩欓噷URLEncoder.encode鍙互闃叉涓枃涔辩爜 褰撶劧鍜宔asyexcel娌℃湁鍏崇郴 String fileName = URLEncoder.encode("娴嬭瘯", "UTF-8"); response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx"); - EasyExcel.write(response.getOutputStream(), EnInitData.class).sheet("妯℃澘").doWrite(ExcelUtils.data()); + EasyExcel.write(response.getOutputStream(), MatCodeExcel.class).sheet("妯℃澘").doWrite(excels); } } -- Gitblit v1.9.1