使用Hutool生成多个excel文件合并成zip压缩包下载

1
2
3
4
5
6
7
8
9
@Getter
@Setter
public class Student {
	private String id;
	private String name;
	private Integer age;
	private String sex;
}

1
2
public interface StudentMappper extends Mapper<Student>{}

1
2
3
4
5
6
public class StudentExport {
	private String name;
	private Integer age;
	private String sex;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
@Service
public class StudentService {
	@Autowired
	private StudentMappper studentMappper;
	public List<Student> getAll(){
		return studentMappper.selectAll();
	}
	//创建单个excel的方法，多个excel的话，加几个这种方法就行
	public List<Object> createExcel() {
		List<Object> result=new ArrayList<Object>();
		List<Student> sList=getAll();
		List<StudentExport> exportList=new ArrayList<StudentExport>();
		for (Student student : sList) {
			StudentExport export=new StudentExport();
			export.setAge(student.getAge());
			export.setName(student.getName());
			export.setSex(student.getSex());
			exportList.add(export);
		}
		ExcelWriter writer=new ExcelUtil().getWriter();
		//添加表头对应的数据列
		writer.addHeaderAlias("name", "名字");
		writer.addHeaderAlias("age", "年龄");
		writer.addHeaderAlias("sex", "性别");
		writer.write(exportList,true);
		//格式化文件名字模板
        String fileName = String.format("%s-%s.xls", "student", DateUtil.format(new Date(), "yyyyMMdd"));
        //写入流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        writer.flush(out);
        //关闭
        writer.close();
        result.add(out);
        result.add(fileName);
		return result;
	}

//循环导入excel的流，准备在Controller层生成zip包
	public void writeZos(List<ByteArrayOutputStream> bosList, ZipOutputStream zos,List<String> excelName) throws IOException {
		for (int i = 0; i < bosList.size(); i++) {
			//将多个excel都转成字节流写入
			zos.putNextEntry(new ZipEntry(excelName.get(i)));
			byte[] excelStream=bosList.get(i).toByteArray();
			zos.write(excelStream);
			//记得关闭
			zos.closeEntry();
		}
	}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
public class StudentApi {
	
	@Autowired
	private StudentService studentService;
	
	public void createZip(HttpServletResponse response) throws IOException {
		List<ByteArrayOutputStream> bosList=new ArrayList<ByteArrayOutputStream>();
		List<String> excelName =new ArrayList<String>();
		//这里循环取出service里填充的单个excel流和单个excelName,如果多个循环即可
		bosList.add((ByteArrayOutputStream) studentService.createExcel().get(0));
		excelName.add((String) studentService.createExcel().get(1));
		//创建HttpServerResponse的输出流
		OutputStream out=response.getOutputStream();
		//创建写入流
		BufferedInputStream bis;
		//创建要写入的文件
		File file=new File("student.zip");
		//通过ZipOutputStream定义要写入的对象
		ZipOutputStream zos=new ZipOutputStream(new FileOutputStream(file));
		//调取service层写入循环excle的方法，将流输入
		studentService.writeZos(bosList, zos, excelName);
		zos.close();
		//定义返回类型
		response.setContentType("text/html; charset=UTF-8"); 
        response.setContentType("application/octet-stream"); 
        response.setHeader("Content-Disposition", "attachment;filename=".concat(String.valueOf(URLEncoder.encode("student.zip", "UTF-8"))));
        bis = new BufferedInputStream(new FileInputStream("student.zip"));
        //定义byte，长度就是要转成zip文件的byte长度，避免浪费资源
        byte[] buffer=new byte[bis.available()];
        bis.read(buffer);
        out.flush();
        out.write(buffer);
	}
}