【springboot+easypoi】导出多个sheet页

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

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

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

一、检查pom是否引入easypoi

        <dependency>
            <groupId>cn.afterturn</groupId>
            <artifactId>easypoi-base</artifactId>
            <version>3.2.0</version>
        </dependency>

二、思路

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

[{
'三年一班'：[Student1,Student2,Student3···]
}，
{
'三年二班'：[Student1,Student2,Student3···]
}，
{
'三年三班'：[Student1,Student2,Student3···]
}
]

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

三、主要工具类

package com.mos.bee.util;

import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.enmus.ExcelType;
import com.mos.bee.entity.dto.QrUserRecordsDownloadDTO;
import org.apache.poi.ss.usermodel.Workbook;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class WorkBookUtils {

    public static Workbook mutiSheet(List<Map<String, Object>> mapListList){
        return ExcelExportUtil.exportExcel(mapListList, ExcelType.XSSF);
    }

    public static Map<String, Object> createOneSheet(ExportParams exportParams, Class<?> clazz, List<?> data){
        Map<String, Object> map = new HashMap<>();
        map.put("title",exportParams);
        map.put("entity", clazz);
        map.put("data",data);
        return map;
    }
    /*
     * 创建一个表格并填充内容
     * 返回map供工作簿使用
     * */
    public static Map<String, Object> createOneSheet(String sheetName,String title,Class<?> clazz,List<?> data){
        ExportParams exportParams = new ExportParams(title,sheetName, ExcelType.XSSF);
        return createOneSheet(exportParams,clazz,data);
    }

    /**
     * 将数据存储进response,调用接口就能下载文件
     */
    public static  void dataToResponse(HttpServletRequest request, HttpServletResponse response, File file, String fileName) throws IOException {
        OutputStream out = null;
        FileInputStream in = null;
        try {
            // 1.读取要下载的内容
            in = new FileInputStream(file);
            // 2. 告诉浏览器下载的方式以及一些设置
            // 解决文件名乱码问题，获取浏览器类型，转换对应文件名编码格式，IE要求文件名必须是utf-8, firefo要求是iso-8859-1编码
            String agent = request.getHeader("user-agent");
            if (agent.contains("FireFox")) {
                fileName = new String(fileName.getBytes("UTF-8"), "iso-8859-1");
            } else {
                fileName = URLEncoder.encode(fileName, "UTF-8");
            }
            // 设置下载文件的mineType，告诉浏览器下载文件类型
            String mineType = request.getServletContext().getMimeType(fileName);
            response.setContentType(mineType);
            // 设置一个响应头，无论是否被浏览器解析，都下载
            response.setHeader("Content-disposition", "attachment; filename=" + fileName);
            // 将要下载的文件内容通过输出流写到浏览器
            out = response.getOutputStream();
            int len = 0;
            byte[] buffer = new byte[1024];
            while ((len = in.read(buffer)) > 0) {
                out.write(buffer, 0, len);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (out != null) {
                out.close();
            }
            if (in != null) {
                in.close();
            }
        }
    }

    /**
     * 设置文件路径 && 保证文件对象的正确打开
     */
    public static  File createDatafile(String fileDownloadTmpPath, String fileName) throws IOException {
        File dir = new File(fileDownloadTmpPath);
        if (!dir.exists()) {
            dir.mkdirs();
        }
        String resource = fileDownloadTmpPath + fileName;
        File myFile = new File(resource);//创建File对象，参数为String类型，表示目录名
        //判断文件是否存在，如不存在则调用createNewFile()创建新目录，否则跳至异常处理代码
        if (!myFile.exists()) {
            myFile.createNewFile();
        }
        return myFile;
    }

}

四、数据处理

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

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

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

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

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

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

List<Map<String, Object>> lists = new ArrayList<>();
        listMap.forEach((k, v) -> {
            Map<String, Object> temp1 = WorkBookUtils.createOneSheet(k,  "这个是标题", StudentDTO.class, v);
            lists.add(temp1);
        });
        Workbook workbook = WorkBookUtils.mutiSheet(lists);
        FileOutputStream fos = new FileOutputStream(saveFile);
        workbook.write(fos);
        fos.close();

4. 剩下就是按部就班的导出功能了，就不再赘述了

五、结语

如果大家有什么疑问，可以关注我的个人公众号【陌与尘埃】，可以回复springboot来获取历史相关文章，或者直接下载个人开源项目【Eboot】直接使用