最近接觸到一個需求要求壓碩訓出檔案，于是乎便要致力于研究一下工具類啦，故也誕生了此篇文章，
下面代碼中，溪源也將import匯入的依賴也貼出來了，避免大家引入錯誤，
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
* @author wx
* @date 2020/10/29 5:19 下午
*/
public class FileZipUtil {
private static void handlerFile(ZipOutputStream zip, File file, String dir) throws Exception {
//如果當前的是檔案夾，則進行進一步處理
if (file.isDirectory()) {
//得到檔案串列資訊
File[] fileArray = file.listFiles();
if (fileArray == null) {
return;
}
//將檔案夾添加到下一級打包目錄
zip.putNextEntry(new ZipEntry(dir + "/"));
dir = dir.length() == 0 ? "" : dir + "/";
//遞回將檔案夾中的檔案打包
for (File f : fileArray) {
handlerFile(zip, f, dir + f.getName());
}
} else {
//當前的是檔案，打包處理
//檔案輸入流
BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
ZipEntry entry = new ZipEntry(dir);
zip.putNextEntry(entry);
zip.write(FileUtils.readFileToByteArray(file));
IOUtils.closeQuietly(bis);
zip.flush();
zip.closeEntry();
}
}
private static byte[] createZip(String sourceFilePath) throws Exception{
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
ZipOutputStream zip = new ZipOutputStream(outputStream);
//將目標檔案打包成zip匯出
File file = new File(sourceFilePath);
handlerFile(zip, file,"");
IOUtils.closeQuietly(zip);
return outputStream.toByteArray();
}
public static void exportZip(HttpServletResponse response, String sourceFilePath) {
//檔案名以時間戳作為前綴
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
String filePrefix = sdf.format(new Date());
String downloadName = filePrefix + ".zip";
//將檔案進行打包下載
try {
OutputStream out = response.getOutputStream();
//接收壓縮包位元組
byte[] data = createZip(sourceFilePath);
response.reset();
response.addHeader("Access-Control-Allow-Origin", "*");
response.setHeader("Access-Control-Expose-Headers", "*");
response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + downloadName);
response.addHeader("Content-Length", "" + data.length);
response.setContentType("application/octet-stream;charset=UTF-8");
IOUtils.write(data, out);
out.flush();
out.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}
客戶端呼叫方法：
@GetMapping("/exportFile")
public Result exportFile(HttpServletResponse response) {
//第二個引數為：要壓縮檔案的地址
FileZipUtil.exportZip(response, "/Users/Downloads");
}
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/203074.html
標籤：其他
上一篇：Java實作單鏈表的簡單操作