对，没有错，又是我，又是easyPOI，又是excel导出。每个程序猿都听说技术是为业务服务的，那么···当需求变了之后我们能做什么呢？
喂，需求大哥，我改还不行吗？刀可以收起来了。
前提
1. 某学校三年级共有三个班：三年一班、三年二班、三年三班
2. 每个班有N个人
要求
以班级为单位，导出所有人的名单，导出到一个Excel文件中，分不同的Sheet页。
一、检查pom是否引入easypoi
cn.afterturn
easypoi-base
3.2.0
二、思路
- 考虑导出时需要组装的数据格式
[{
'三年一班'：[Student1,Student2,Student3···]
}，
{
'三年二班'：[Student1,Student2,Student3···]
}，
{
'三年三班'：[Student1,Student2,Student3···]
}
]
- 创建workbook
- 根据上面组装的对象，根据不同班级，循环创建sheet页
- 下载文件（这一步可以和以前的共用）
三、主要工具类
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> mapListList){
return ExcelExportUtil.exportExcel(mapListList, ExcelType.XSSF);
}
public static Map createOneSheet(ExportParams exportParams, Class clazz, List data){
Map map = new HashMap<>();
map.put("title",exportParams);
map.put("entity", clazz);
map.put("data",data);
return map;
}
/*
* 创建一个表格并填充内容
* 返回map供工作簿使用
* */
public static Map 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;
}
}
四、数据处理
为了杜绝“拿来主义”，此处不再写具体的代码，但是会写一些思路和伪代码。
- 数据库存储格式一般为（主要字段）：
| id | name | class_id | other |
|---|---|---|---|
| id | 学生姓名 | 班级id | 其他字段 |
- 查询出学生列表，然后根据班级分组，jdk8实现方式：
//省略查询步骤
Map> listMap = allStudentList.stream().collect(Collectors.groupingBy(StudentDTO::getClassId));
3.根据分组后的Map创建sheet
List> lists = new ArrayList<>();
listMap.forEach((k, v) -> {
Map 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();
- 剩下就是按部就班的导出功能了，就不再赘述了
五、结语
如果大家有什么疑问，可以关注我的个人公众号【陌与尘埃】，可以回复springboot来获取历史相关文章，或者直接下载个人开源项目【Eboot】直接使用