前端请求:
{"ids":["1080431045908779009","1080434473292681218"]}
Controller:
@PostMapping
public void downloadFiles(HttpServletRequest request, HttpServletResponse response, @RequestBody Map<String, Object> map){
List<String> ids = (List<String>) map.get("ids");
System.out.println(ids);
}
Service:
/**
* 批量下载文件
*
* @param request http请求
* @param response http响应
* @param ids 文件id数组
* @param fileName 打包文件名称
*/
@Override
public void downloadFiles(HttpServletRequest request, HttpServletResponse response, List<String> ids, String fileName) {
//响应头的设置
response.reset();
response.setCharacterEncoding("utf-8");
response.setContentType("multipart/form-data");
//设置压缩包的名字
//解决不同浏览器压缩包名字含有中文时乱码的问题
String downloadName = fileName + ".zip";
String agent = request.getHeader("USER-AGENT");
try {
if (agent.contains("MSIE") || agent.contains("Trident")) {
downloadName = java.net.URLEncoder.encode(downloadName, "UTF-8");
} else {
downloadName = new String(downloadName.getBytes("UTF-8"), "ISO-8859-1");
}
} catch (Exception e) {
logger.error(e.getMessage());
}
response.setHeader("Content-Disposition", "attachment;filename=\"" + downloadName + "\"");
//设置压缩流:直接写入response,实现边压缩边下载
ZipOutputStream zipos = null;
try {
zipos = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()));
//设置压缩方法
zipos.setMethod(ZipOutputStream.DEFLATED);
} catch (Exception e) {
logger.error(e.getMessage());
}
//循环将文件写入压缩流
DataOutputStream os = null;
for (int i = 0; i < ids.size(); i++) {
FileInfo fileInfo = getById(ids.get(i));
//拼接文件路径
String filePath = baseFilePath + fileInfo.getFolder() + "\\" + fileInfo.getSaveName();
File file = new File(filePath);
if (!file.exists()) {
continue;
}
try {
//添加ZipEntry,并ZipEntry中写入文件流
//这里,加上i是防止要下载的文件有重名的导致下载失败
if (zipos != null) {
zipos.putNextEntry(new ZipEntry(fileInfo.getName()));
os = new DataOutputStream(zipos);
InputStream is = new FileInputStream(file);
byte[] b = new byte[1000];
int length = 0;
while ((length = is.read(b)) != -1) {
os.write(b, 0, length);
}
is.close();
zipos.closeEntry();
//将该文件的downloadCount加1
increaseOneDownloadCount(fileInfo.getId());
}
} catch (IOException e) {
logger.error(e.getMessage());
}
}
//关闭流
if (os != null) {
try {
os.flush();
os.close();
} catch (IOException e) {
logger.error(e.getMessage());
}
}
if (zipos != null) {
try {
zipos.close();
} catch (IOException e) {
logger.error(e.getMessage());
}
}
}