在工作上回遇到很多需要进行文件批量下载的需求,将多个文件合并成一个zip压缩包进行下载
之前也是参考了其他大佬的博客才有个人使用的总结 在此表示感谢
直接上代码
/**
* req 请求头
* response 返回体
* fileInfoList 文件信息集合
*/
public void downForZip(HttpServletRequest req, HttpServletResponse response,List<FileInfo> fileInfoList) {
// 创建临时文件
File zipFile = null;
FileInputStream fis = null;
BufferedInputStream buff = null;
try {
//临时文件名称
zipFile = File.createTempFile("test", ".zip");
FileOutputStream fot = new FileOutputStream(zipFile);
// 为任何OutputStream产生校验,第一个参数是制定产生校验和的输出流,第二个参数是指定Checksum的类型 (Adler32(较快)和CRC32两种)
CheckedOutputStream cos = new CheckedOutputStream(fot, new Adler32());
// 用于将数据压缩成Zip文件格式
ZipOutputStream zos = new ZipOutputStream(cos);
for (FileInfo file: fileInfoList) {
//根据文件信息获取输入流。这个看实际的业务目的就是获取当前文件的输入流
InputStream inputStream = fileService.getFileInpuStream(file.getId(), file.getKey());
if (null == inputStream) {
break;
}
// 对于每一个要被存放到压缩包的文件,都必须调用ZipOutputStream对象的putNextEntry()方法,确保压缩包里面文件不同名
//多个文件名称重复时zos.putNextEntry()会报错!! 可以再文件名称后面加编号等操作
zos.putNextEntry(new ZipEntry(file.getFileName()));
int bytesRead = 0;
// 向压缩文件中输出数据
while ((bytesRead = inputStream.read()) != -1) {
zos.write(bytesRead);
}
inputStream.close();
// 当前文件写完,写入下一个文件
zos.closeEntry();
}
zos.close();
ServletOutputStream os = response.getOutputStream();
//下载文件,使用spring框架中的FileCopyUtils工具
response.setCharacterEncoding("GB2312");
response.setContentType(req.getSession().getServletContext().getMimeType(file.getFileName()));
//设置响应头,attachment表示以附件的形式下载,inline表示在线打开
response.setHeader("content-disposition", "attachment;fileName=" + URLEncoder.encode("批量下载.zip", "UTF-8"));
fis = new FileInputStream(zipFile);
buff = new BufferedInputStream(fis);
FileCopyUtils.copy(buff, os);
} catch (Exception e1) {
throw new BusinessException("批量下载失败");
// 关闭流
} finally {
try {
if (null != fis) {
fis.close();
}
if (null != buff) {
buff.close();
}
} catch (IOException e) {
log.error( "流关闭异常");
}
// 删除临时文件
if (null != zipFile) {
zipFile.delete();
}
}
}