对,没有错,又是我,又是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}
- 剩下就是按部就班的导出功能了,就不再赘述了
五、结语
如果大家有什么疑问,可以关注我的个人公众号【陌与尘埃】,可以回复springboot来获取历史相关文章,或者直接下载个人开源项目【Eboot】直接使用