1. 生成压缩包需要引入ZipOutputStream类
//获取目标文件输入流
InputStream is = Files.newInputStream("目标文件路径");
//构建生成压缩文件的格式
zipOutputStream.putNextEntry(new ZipEntry("生成压缩包内的路径"));
//将输入流写入压缩包
int readLen;
byte[] buffer = new byte[1024];
while ((readLen = is.read(buffer)) != -1) {
zipOutputStream.write(buffer, 0, readLen);
}
2. 最近项目中写的DEMO代码
- 记录需要压缩的文件名和文件输入流
public Map<String, InputStream> record(String path, String preDir) throws IOException {
Map<String, InputStream> fileMap = new HashMap<>();
Files.walkFileTree(Paths.get(path), new SimpleFileVisitor<Path>() {
//同名文件加前缀
private int count = 2;
@Override
public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
//文件为空则存null值区分
if (Files.list(dir).count() == 0) {
fileMap.put(null, null);
}
return FileVisitResult.CONTINUE;
}
@Override
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
InputStream is = Files.newInputStream(file);
if (fileMap.containsKey(preDir + File.separator + file.getFileName().toString())) {
StringBuffer sb = new StringBuffer(preDir + File.separator + count + "_");
sb.append(file.getFileName().toString());
fileMap.put(sb.toString(), is);
count++;
} else {
StringBuffer sb2 = new StringBuffer(preDir + File.separator);
sb2.append(file.getFileName().toString());
fileMap.put(sb2.toString(), is);
}
return FileVisitResult.CONTINUE;
}
@Override
public FileVisitResult visitFileFailed(Path file, IOException exc) throws IOException {
logger.error("文件:{},访问失败:{}", file.getFileName().toString(), exc.getMessage(), exc);
return FileVisitResult.CONTINUE;
}
});
return fileMap;
}
- 整合SpringMVC把文件响应给页面
@RequestMapping("/downzip")
public void getFile(HttpServletResponse response) {
response.setContentType("application/octet-stream");
try {
response.setHeader("Content-disposition", "attachment;filename="
+ new String("预约ID.zip".getBytes("UTF-8"), "ISO8859-1")
);
} catch (UnsupportedEncodingException e) {
logger.error("响应头出错:{}", e.getMessage(), e);
}
response.setHeader("Accept-Ranges", "bytes");
Map<String, InputStream> videoMap = new HashMap<>();
Map<String, InputStream> videopicMap = new HashMap<>();
Map<String, InputStream> videoAttchMap = new HashMap<>();
try {
//视频文件
videoMap = record("E:\\video\\999\\", "video");
//图片文件
videopicMap = record("E:\\videopic\\999", "videopic");
//附件文件
videoAttchMap = record("E:\\videoAttch\\999", "videoAttch");
}catch (IOException e){
}
List<Map<String, InputStream>> mapList = new ArrayList<>(3);
mapList.add(videoMap);
mapList.add(videopicMap);
mapList.add(videoAttchMap);
zipMultipleFiles2(response, mapList);
}
- 遍历记录好的文件输入流集合
/**
* 将存储的流添加压缩包并响应给调用者
* @param response
* @param mapList
*/
public void zipMultipleFiles(HttpServletResponse response, List<Map<String, InputStream>> mapList) {
try (ZipOutputStream zipOutputStream = new ZipOutputStream(response.getOutputStream())) {
// 遍历每一个文件，进行压缩
for (Map<String, InputStream> map : mapList) {
Iterator<Map.Entry<String, InputStream>> it = map.entrySet().iterator();
while (it.hasNext()) {
Map.Entry<String, InputStream> entry = it.next();
String path = entry.getKey();
InputStream is = entry.getValue();
if (Objects.equals(path, null) || Objects.equals(is, null)) {
continue;
}
zipOutputStream.putNextEntry(new ZipEntry(path));
int readLen;
byte[] buffer = new byte[1024];
while ((readLen = is.read(buffer)) != -1) {
zipOutputStream.write(buffer, 0, readLen);
}
entry.getValue().close();
zipOutputStream.closeEntry();
}
}
} catch (IOException e) {
logger.error("出现错误:{}", e.getMessage(), e);
}
}
-
压缩包效果
压缩包效果图
