1、编写工具类
public class ZipUtils {
public static void createZipFile(HttpServletResponse response, List<File> files, List<String> fileNames, String zipName) {
ZipOutputStream zipStream = null;
response.setContentType("application/x-zip-compressed");
response.setHeader("Cache-Control", "max-age=0");
try {
response.setHeader("Content-Disposition ", "attachment; filename=" + URLEncoder.encode(zipName+".zip", "UTF-8") );
OutputStream os = response.getOutputStream();
zipStream = new ZipOutputStream(os);
for (int i = 0; i < files.size(); i++) {
File file = files.get(i);
compressZip(file, zipStream, fileNames.get(i));
}
} catch (Exception e) {
e.printStackTrace();
}finally {
try {
if (zipStream != null) {
zipStream.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
private static void compressZip(File file, ZipOutputStream zipStream, String alias) throws Exception{
FileInputStream input = null;
try {
input = new FileInputStream(file);
zip(input, zipStream, alias+"."+file.getName().substring(file.getName().lastIndexOf(".")+1));
} catch (Exception e) {
e.printStackTrace();
}finally {
try {
if(input != null)
input.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
private static void zip(InputStream input, ZipOutputStream zipStream, String zipEntryName) throws Exception{
byte[] bytes = null;
BufferedInputStream bufferStream = null;
try {
if(input == null) {
throw new Exception("获取压缩的数据项失败! 数据项名为:" + zipEntryName);
}
ZipEntry zipEntry = new ZipEntry(zipEntryName);
zipStream.putNextEntry(zipEntry);
bytes = new byte[1024 * 5];
bufferStream = new BufferedInputStream(input);
int read = 0;
while ((read = bufferStream.read(bytes)) != -1) {
zipStream.write(bytes, 0, read);
}
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
if (null != bufferStream)
bufferStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
2、编写测试类
File file1 = new File("D:\\file\\policy.sql");
File file2 = new File("D:\\file\\example.sql");
List<File> files = new ArrayList<>();
List<String> fileName = new ArrayList<>();
files.add(file1);
fileName.add("测试1");
files.add(file2);
fileName.add("测试2");
String zipName = "测试";
ZipUtils.createZipFile(response,files,fileName,zipName);