【springboot+easypoi】导出多个sheet页

小尘哥

发布于 2020-07-17 14:07:21

1.8K0

发布于 2020-07-17 14:07:21

文章被收录于专栏：小尘哥的专栏小尘哥的专栏

对，没有错，又是我，又是easyPOI，又是excel导出。每个程序猿都听说技术是为业务服务的，那么···当需求变了之后我们能做什么呢？

喂，需求大哥，我改还不行吗？刀可以收起来了。

前提 1. 某学校三年级共有三个班：三年一班、三年二班、三年三班 2. 每个班有N个人 要求以班级为单位，导出所有人的名单，导出到一个Excel文件中，分不同的Sheet页。

一、检查pom是否引入easypoi

1        <dependency>
2            <groupId>cn.afterturn</groupId>
3            <artifactId>easypoi-base</artifactId>
4            <version>3.2.0</version>
5        </dependency>

二、思路

考虑导出时需要组装的数据格式

 1[{
 2'三年一班'：[Student1,Student2,Student3···]
 3}，
 4{
 5'三年二班'：[Student1,Student2,Student3···]
 6}，
 7{
 8'三年三班'：[Student1,Student2,Student3···]
 9}
10]

创建workbook
根据上面组装的对象，根据不同班级，循环创建sheet页
下载文件（这一步可以和以前的共用）

三、主要工具类

 1package com.mos.bee.util;
 2
 3import cn.afterturn.easypoi.excel.ExcelExportUtil;
 4import cn.afterturn.easypoi.excel.entity.ExportParams;
 5import cn.afterturn.easypoi.excel.entity.enmus.ExcelType;
 6import com.mos.bee.entity.dto.QrUserRecordsDownloadDTO;
 7import org.apache.poi.ss.usermodel.Workbook;
 8
 9import javax.servlet.http.HttpServletRequest;
10import javax.servlet.http.HttpServletResponse;
11import java.io.*;
12import java.net.URLEncoder;
13import java.util.ArrayList;
14import java.util.HashMap;
15import java.util.List;
16import java.util.Map;
17
18public class WorkBookUtils {
19
20    public static Workbook mutiSheet(List<Map<String, Object>> mapListList){
21        return ExcelExportUtil.exportExcel(mapListList, ExcelType.XSSF);
22    }
23
24    public static Map<String, Object> createOneSheet(ExportParams exportParams, Class<?> clazz, List<?> data){
25        Map<String, Object> map = new HashMap<>();
26        map.put("title",exportParams);
27        map.put("entity", clazz);
28        map.put("data",data);
29        return map;
30    }
31    /*
32     * 创建一个表格并填充内容
33     * 返回map供工作簿使用
34     * */
35    public static Map<String, Object> createOneSheet(String sheetName,String title,Class<?> clazz,List<?> data){
36        ExportParams exportParams = new ExportParams(title,sheetName, ExcelType.XSSF);
37        return createOneSheet(exportParams,clazz,data);
38    }
39
40    /**
41     * 将数据存储进response,调用接口就能下载文件
42     */
43    public static  void dataToResponse(HttpServletRequest request, HttpServletResponse response, File file, String fileName) throws IOException {
44        OutputStream out = null;
45        FileInputStream in = null;
46        try {
47            // 1.读取要下载的内容
48            in = new FileInputStream(file);
49            // 2. 告诉浏览器下载的方式以及一些设置
50            // 解决文件名乱码问题，获取浏览器类型，转换对应文件名编码格式，IE要求文件名必须是utf-8, firefo要求是iso-8859-1编码
51            String agent = request.getHeader("user-agent");
52            if (agent.contains("FireFox")) {
53                fileName = new String(fileName.getBytes("UTF-8"), "iso-8859-1");
54            } else {
55                fileName = URLEncoder.encode(fileName, "UTF-8");
56            }
57            // 设置下载文件的mineType，告诉浏览器下载文件类型
58            String mineType = request.getServletContext().getMimeType(fileName);
59            response.setContentType(mineType);
60            // 设置一个响应头，无论是否被浏览器解析，都下载
61            response.setHeader("Content-disposition", "attachment; filename=" + fileName);
62            // 将要下载的文件内容通过输出流写到浏览器
63            out = response.getOutputStream();
64            int len = 0;
65            byte[] buffer = new byte[1024];
66            while ((len = in.read(buffer)) > 0) {
67                out.write(buffer, 0, len);
68            }
69        } catch (IOException e) {
70            e.printStackTrace();
71        } finally {
72            if (out != null) {
73                out.close();
74            }
75            if (in != null) {
76                in.close();
77            }
78        }
79    }
80
81    /**
82     * 设置文件路径 && 保证文件对象的正确打开
83     */
84    public static  File createDatafile(String fileDownloadTmpPath, String fileName) throws IOException {
85        File dir = new File(fileDownloadTmpPath);
86        if (!dir.exists()) {
87            dir.mkdirs();
88        }
89        String resource = fileDownloadTmpPath + fileName;
90        File myFile = new File(resource);//创建File对象，参数为String类型，表示目录名
91        //判断文件是否存在，如不存在则调用createNewFile()创建新目录，否则跳至异常处理代码
92        if (!myFile.exists()) {
93            myFile.createNewFile();
94        }
95        return myFile;
96    }
97
98}

四、数据处理

为了杜绝“拿来主义”，此处不再写具体的代码，但是会写一些思路和伪代码。

1. 数据库存储格式一般为（主要字段）：

student_id	student_name	class_id	other
学生id	学生姓名	班级id	其他字段

2. 查询出学生列表，然后根据班级分组，jdk8实现方式：

1//省略查询步骤
2 Map<String, List<StudentDTO>> listMap = allStudentList.stream().collect(Collectors.groupingBy(StudentDTO::getClassId));

3.根据分组后的Map创建sheet

 1public File fillData(listMap){
 2        List<Map<String, Object>> lists = new ArrayList<>();
 3        listMap.forEach((k, v) -> {
 4            Map<String, Object> temp1 = WorkBookUtils.createOneSheet(k,  "这个是标题", StudentDTO.class, v);
 5            lists.add(temp1);
 6        });
 7        Workbook workbook = WorkBookUtils.mutiSheet(lists);
 8        FileOutputStream fos = new FileOutputStream(saveFile);
 9        workbook.write(fos);
10        fos.close();
11}