Java實(shí)現(xiàn)把文件及文件夾壓縮成zip
最近碰到個(gè)需要下載zip壓縮包的需求,于是我在網(wǎng)上找了下別人寫好的zip工具類。但找了好多篇博客,總是發(fā)現(xiàn)有bug。因此就自己來寫了個(gè)工具類。
這個(gè)工具類的功能為:
(1)可以壓縮文件,也可以壓縮文件夾
(2)同時(shí)支持壓縮多級(jí)文件夾,工具內(nèi)部做了遞歸處理
(3)碰到空的文件夾,也可以壓縮
(4)可以選擇是否保留原來的目錄結(jié)構(gòu),如果不保留,所有文件跑壓縮包根目錄去了,且空文件夾直接舍棄。注意:如果不保留文件原來目錄結(jié)構(gòu),在碰到文件名相同的文件時(shí),會(huì)壓縮失敗。
(5)代碼中提供了2個(gè)壓縮文件的方法,一個(gè)的輸入?yún)?shù)為文件夾路徑,一個(gè)為文件列表,可根據(jù)實(shí)際需求選擇方法。
下面直接上代碼
一、代碼
ZipUtils
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
* @author Nemo
* @version 1.0
* @date 2019/11/5
*/
public class ZipUtils {
private static final int BUFFER_SIZE = 2 * 1024;
/**
* 壓縮成ZIP 方法1
* @param sourceFile 壓縮文件夾路徑
* @param out 壓縮文件輸出流
* @param KeepDirStructure 是否保留原來的目錄結(jié)構(gòu),true:保留目錄結(jié)構(gòu);
* false:所有文件跑到壓縮包根目錄下(注意:不保留目錄結(jié)構(gòu)可能會(huì)出現(xiàn)同名文件,會(huì)壓縮失敗)
* @throws RuntimeException 壓縮失敗會(huì)拋出運(yùn)行時(shí)異常
*/
public static void toZip(File sourceFile, OutputStream out, boolean KeepDirStructure)
throws RuntimeException{
ZipOutputStream zos = null ;
try {
zos = new ZipOutputStream(out);
compress(sourceFile,zos,sourceFile.getName(),KeepDirStructure);
} catch (Exception e) {
throw new RuntimeException("zip error from ZipUtils",e);
}finally{
if(zos != null){
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 壓縮成ZIP 方法2
* @param srcFiles 需要壓縮的文件列表
* @param out 壓縮文件輸出流
* @throws RuntimeException 壓縮失敗會(huì)拋出運(yùn)行時(shí)異常
*/
public static void toZip(List<File> srcFiles , OutputStream out)throws RuntimeException {
long start = System.currentTimeMillis();
ZipOutputStream zos = null ;
try {
zos = new ZipOutputStream(out);
for (File srcFile : srcFiles) {
byte[] buf = new byte[BUFFER_SIZE];
zos.putNextEntry(new ZipEntry(srcFile.getName()));
int len;
FileInputStream in = new FileInputStream(srcFile);
while ((len = in.read(buf)) != -1){
zos.write(buf, 0, len);
}
zos.closeEntry();
in.close();
}
long end = System.currentTimeMillis();
System.out.println("壓縮完成,耗時(shí):" + (end - start) +" ms");
} catch (Exception e) {
throw new RuntimeException("zip error from ZipUtils",e);
}finally{
if(zos != null){
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 遞歸壓縮方法
* @param sourceFile 源文件
* @param zos zip輸出流
* @param name 壓縮后的名稱
* @param KeepDirStructure 是否保留原來的目錄結(jié)構(gòu),true:保留目錄結(jié)構(gòu);
* false:所有文件跑到壓縮包根目錄下(注意:不保留目錄結(jié)構(gòu)可能會(huì)出現(xiàn)同名文件,會(huì)壓縮失敗)
* @throws Exception
*/
private static void compress(File sourceFile, ZipOutputStream zos, String name,
boolean KeepDirStructure) throws Exception{
byte[] buf = new byte[BUFFER_SIZE];
if(sourceFile.isFile()){
// 向zip輸出流中添加一個(gè)zip實(shí)體,構(gòu)造器中name為zip實(shí)體的文件的名字
zos.putNextEntry(new ZipEntry(name));
// copy文件到zip輸出流中
int len;
FileInputStream in = new FileInputStream(sourceFile);
while ((len = in.read(buf)) != -1){
zos.write(buf, 0, len);
}
// Complete the entry
zos.closeEntry();
in.close();
} else {
File[] listFiles = sourceFile.listFiles();
if(listFiles == null || listFiles.length == 0){
// 需要保留原來的文件結(jié)構(gòu)時(shí),需要對(duì)空文件夾進(jìn)行處理
if(KeepDirStructure){
// 空文件夾的處理
zos.putNextEntry(new ZipEntry(name + "/"));
// 沒有文件,不需要文件的copy
zos.closeEntry();
}
}else {
for (File file : listFiles) {
// 判斷是否需要保留原來的文件結(jié)構(gòu)
if (KeepDirStructure) {
// 注意:file.getName()前面需要帶上父文件夾的名字加一斜杠,
// 不然最后壓縮包中就不能保留原來的文件結(jié)構(gòu),即:所有文件都跑到壓縮包根目錄下了
compress(file, zos, name + "/" + file.getName(),KeepDirStructure);
} else {
compress(file, zos, file.getName(),KeepDirStructure);
}
}
}
}
}
public static void main(String[] args) throws Exception {
/** 測試壓縮方法1 */
FileOutputStream fos1 = new FileOutputStream(new File("c:/mytest01.zip"));
ZipUtils.toZip(new File("D:/log"), fos1,true);
/** 測試壓縮方法2 */
List<File> fileList = new ArrayList<>();
fileList.add(new File("D:/Java/jdk1.7.0_45_64bit/bin/jar.exe"));
fileList.add(new File("D:/Java/jdk1.7.0_45_64bit/bin/java.exe"));
FileOutputStream fos2 = new FileOutputStream(new File("c:/mytest02.zip"));
ZipUtils.toZip(fileList, fos2);
}
}
二、注意事項(xiàng)
寫該工具類時(shí),有些注意事項(xiàng)說一下:
(1)支持選擇是否保留原來的文件目錄結(jié)構(gòu),如果不保留,那么空文件夾直接不用處理。
(1)碰到空文件夾時(shí),如果需要保留目錄結(jié)構(gòu),則直接添加個(gè)ZipEntry就可以了,不過就是這個(gè)entry的名字后面需要帶上一斜杠(/)表示這個(gè)是目錄。
(2)遞歸時(shí),不需要把zip輸出流關(guān)閉,zip輸出流的關(guān)閉應(yīng)該是在調(diào)用完遞歸方法后面關(guān)閉
(3)遞歸時(shí),如果是個(gè)文件夾且需要保留目錄結(jié)構(gòu),那么在調(diào)用方法壓縮他的子文件時(shí),需要把文件夾的名字加一斜杠給添加到子文件名字前面,這樣壓縮后才有多級(jí)目錄。
三、如何在javaWeb項(xiàng)目中使用該工具類
這個(gè)工具類在web項(xiàng)目中的使用場景就是多文件下載,我就簡單說個(gè)下載多個(gè)excel表格的案例吧。
代碼中的步驟為:
(1)創(chuàng)建一個(gè)臨時(shí)文件夾
(2)將要下載的文件生成至該臨時(shí)文件夾內(nèi)
(3)當(dāng)所有文件生成完后,獲取HttpServletResponse獲取設(shè)置下載的header
(4)調(diào)用工具類的方法,傳入上面生成的臨時(shí)文件夾路徑及response獲取的輸出流;這樣就下載出來zip包了
(5)遞歸刪除掉上面生成的臨時(shí)文件夾和文件
下面為一個(gè)示例代碼的代碼片段,不是完整代碼,簡單看一下代碼中的步驟
import org.apache.commons.io.FileUtils;
import java.io.*;
/**
* 圖片打包下載
* @author: wangzhouchao
*/
@ApiImplicitParams({
@ApiImplicitParam(name = "id", value = "申請(qǐng)人id", required = true, dataType = "Long", paramType = "query"),
})
@ApiOperation(value = "圖片打包下載", notes = "圖片打包下載")
@RequestMapping(value = "/downloadPictureList", method = RequestMethod.GET)
public void downloadPictureList(TProposerDataVO tProposerDataVO) {
long readyStart = System.currentTimeMillis();
// ************* 1. 獲取到存在數(shù)據(jù)庫中的圖片的url *************
PictureDownloadVO picturesById = tOrderService.getPicturesByProposerDataId(tProposerDataVO.getId());
// 獲取當(dāng)前類的所在項(xiàng)目路徑
File file = null;
try {
file = new File(ResourceUtils.getURL("classpath:").getPath());
} catch (FileNotFoundException e) {
throw new RuntimeException("獲取根目錄失敗,無法獲取文件目錄!");
}
if(!file.exists()) {
file = new File("");
}
String absolutePath = file.getAbsolutePath();
// 要打包的文件夾列表
String order_number = picturesById.getOrder_number();
String country_name = picturesById.getCountry_name();
String visa_type = picturesById.getVisa_type();
String dirName = order_number + country_name + visa_type;
// ************* 2. 創(chuàng)建要壓縮的文件夾 *************
// 根據(jù)訂單號(hào)+國家名稱+簽證類型創(chuàng)建文件夾
File dirOfOrder = new File(absolutePath, dirName);
if(!dirOfOrder.exists()) {
dirOfOrder.mkdirs();
}
ZipOutputStream zos = null;
OutputStream out = null;
long readyEnd = System.currentTimeMillis();
System.out.println("準(zhǔn)備完成,耗時(shí):" + (readyEnd - readyStart) + " ms");
try {
long downStart = System.currentTimeMillis();
System.out.println("開始下載");
TProposerDataVO vo = picturesById.getProposerDataVO();
// ************* 3. 根據(jù)獲取到的圖片的url,把圖片按照想要的文件夾目錄進(jìn)行下載 *************
// 根據(jù)申請(qǐng)人姓名創(chuàng)建文件夾
File proposerFile = new File(dirOfOrder, vo.getReal_name());
if (!proposerFile.exists()) {
proposerFile.mkdirs();
}
// 下載申請(qǐng)人照片
if (StringUtil.checkNotNull(vo.getPhoto_url())) {
System.out.println("開始下載申請(qǐng)人照片");
WordExportUtil.downloadHttpUrl(DOMAIN + vo.getPhoto_url(), proposerFile.toString(), File.separator + "photo.jpg");
}
// 下載申請(qǐng)人護(hù)照首頁
if (StringUtil.checkNotNull(vo.getPassport_home_page_url())) {
System.out.println("開始下載申請(qǐng)人護(hù)照照片");
WordExportUtil.downloadHttpUrl(DOMAIN + vo.getPassport_home_page_url(), proposerFile.toString(), File.separator + "passport.jpg");
}
// 下載申請(qǐng)人戶口本照片
if (StringUtil.checkNotNull(vo.getResidence_booklet_url())) {
System.out.println("開始下載申請(qǐng)人戶口本照片");
String[] booklets = vo.getResidence_booklet_url().split(",");
// 創(chuàng)建戶口本照片文件夾
File bookletsFile = new File(proposerFile, "hukouben");
if (!bookletsFile.exists()) {
bookletsFile.mkdirs();
}
for (int k = 0; k < booklets.length; k++) {
WordExportUtil.downloadHttpUrl(DOMAIN + booklets[k], bookletsFile.toString(), File.separator + "residenceBooklet" + k + ".jpg");
}
}
// 下載申請(qǐng)人身份證照片
if (StringUtil.checkNotNull(vo.getId_card_status()) && vo.getId_card_status() == 0) {
System.out.println("開始下載申請(qǐng)人身份證照片");
// 創(chuàng)建身份證照片文件夾
File idCards = new File(proposerFile, "idCards");
if (!idCards.exists()) {
idCards.mkdirs();
}
if (StringUtil.checkNotNull(vo.getId_card_positive_url())) {
WordExportUtil.downloadHttpUrl(DOMAIN + vo.getId_card_positive_url(), idCards.toString(), File.separator + "idCardPostive.jpg");
}
if (StringUtil.checkNotNull(vo.getId_card_reverse_url())) {
WordExportUtil.downloadHttpUrl(DOMAIN + vo.getId_card_reverse_url(), idCards.toString(), File.separator + "idCardReverse.jpg");
}
}
// 下載申請(qǐng)人婚姻證明照片
if (StringUtil.checkNotNull(vo.getMar_div_card_url())) {
System.out.println("開始下載申請(qǐng)人婚姻證明照片");
WordExportUtil.downloadHttpUrl(DOMAIN + vo.getMar_div_card_url(), proposerFile.toString(), File.separator + "marriage.jpg");
}
// 下載申請(qǐng)人輔助資產(chǎn)照片
if (StringUtil.checkNotNull(vo.getAuxiliary_assets_url())) {
System.out.println("開始下載申請(qǐng)人輔助資產(chǎn)照片");
String[] auxiliarys = vo.getAuxiliary_assets_url().split(",");
// 創(chuàng)建輔助資產(chǎn)照片文件夾
File auxiliarysFile = new File(proposerFile, "fuzhuzichan");
if (!auxiliarysFile.exists()) {
auxiliarysFile.mkdirs();
}
for (int k = 0; k < auxiliarys.length; k++) {
WordExportUtil.downloadHttpUrl(DOMAIN + auxiliarys[k], auxiliarysFile.toString(), File.separator + "auxiliary" + k + ".jpg");
}
}
// 下載申請(qǐng)人居住證照片
if (StringUtil.checkNotNull(vo.getResidence_permit_url())) {
System.out.println("開始下載申請(qǐng)人居住證照片");
String[] residences = vo.getResidence_permit_url().split(",");
// 創(chuàng)建居住證照片文件夾
File residencesFile = new File(proposerFile, "juzhuzheng");
if (!residencesFile.exists()) {
residencesFile.mkdirs();
}
for (int k = 0; k < residences.length; k++) {
WordExportUtil.downloadHttpUrl(DOMAIN + residences[k], residencesFile.toString(), File.separator + "residence" + k + ".jpg");
}
}
// 下載申請(qǐng)人其余補(bǔ)充資料照片
if (StringUtil.checkNotNull(vo.getOther_data_url())) {
System.out.println("開始下載申請(qǐng)人其余補(bǔ)充資料照片");
String[] others = vo.getOther_data_url().split(",");
// 創(chuàng)建其余補(bǔ)充資料照片文件夾
File othersFile = new File(proposerFile, "qitabuchongziliao");
if (!othersFile.exists()) {
othersFile.mkdirs();
}
for (int k = 0; k < others.length; k++) {
WordExportUtil.downloadHttpUrl(DOMAIN + others[k], othersFile.toString(), File.separator + "other" + k + ".jpg");
}
}
// 下載申請(qǐng)人證明資料照片
if (StringUtil.checkNotNull(vo.getProve_url())) {
System.out.println("開始下載申請(qǐng)人證明資料照片");
String[] prove_urls = vo.getProve_url().split(",");
// 創(chuàng)建證明資料照片文件夾
File proveFile = new File(proposerFile, "zhengmingziliao");
if (!proveFile.exists()) {
proveFile.mkdirs();
}
for (int k = 0; k < prove_urls.length; k++) {
WordExportUtil.downloadHttpUrl(DOMAIN + prove_urls[k], proveFile.toString(), File.separator + "prove" + k + ".jpg");
}
}
long downEnd = System.currentTimeMillis();
System.out.println("下載完成,耗時(shí):" + (downEnd - downStart) + " ms");
long zipStart = System.currentTimeMillis();
response.setContentType("application/x-zip-compressed");
response.setHeader("Content-disposition", "attachment;filename=" + StringUtil.getUUID() + ".zip");
out = response.getOutputStream();
zos = new ZipOutputStream(out);
// ************* 4. 把要壓縮的文件夾路徑、壓縮文件輸出流傳入到ZipUtils.toZip方法,對(duì)文件夾進(jìn)行壓縮 *************
// 對(duì)文件夾進(jìn)行壓縮,保留原文件夾路徑
ZipUtils.toZip(dirOfOrder, out, true);
long zipEnd = System.currentTimeMillis();
System.out.println("壓縮完成,耗時(shí):" + (zipEnd - zipStart) + " ms");
out.flush();
} catch (IOException e) {
e.printStackTrace();
} catch (Exception e) {
throw new RuntimeException("zip error from ZipUtils", e);
} finally {
if (zos != null) {
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (out != null) {
try {
zos.close();
out.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
// ************* 5. 刪除壓縮前準(zhǔn)備的中間文件 *************
if (dirOfOrder != null) {
try {
FileUtils.deleteDirectory(dirOfOrder);
System.out.println("中間文件已刪除");
} catch (IOException e) {
e.printStackTrace();
System.out.println("中間文件刪除失敗");
}
}
}
以上就是本文的全部內(nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
詳解基于java的Socket聊天程序——客戶端(附demo)
這篇文章主要介紹了詳解基于java的Socket聊天程序——客戶端(附demo),客戶端設(shè)計(jì)主要分成兩個(gè)部分,分別是socket通訊模塊設(shè)計(jì)和UI相關(guān)設(shè)計(jì)。有興趣的可以了解一下。2016-12-12
如何基于回調(diào)實(shí)現(xiàn)Java的異步調(diào)用
這篇文章主要介紹了如何基于回調(diào)實(shí)現(xiàn)Java的異步調(diào)用,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-06-06
SpringMVC 接收前端傳遞的參數(shù)四種方式小結(jié)
這篇文章主要介紹了SpringMVC 接收前端傳遞的參數(shù)四種方式小結(jié),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2021-10-10
SpringCloud Gateway的熔斷限流配置實(shí)現(xiàn)方法
Spring Cloud Gateway支持通過配置熔斷和限流機(jī)制來保證服務(wù)的穩(wěn)定性和可用性,可通過Hystrix和Resilience4j兩種方式實(shí)現(xiàn)。Hystrix通過注解和配置文件實(shí)現(xiàn)熔斷限流,Resilience4j通過編程式配置實(shí)現(xiàn)2023-04-04
詳解Spring Security中獲取當(dāng)前登錄用戶的詳細(xì)信息的幾種方法
本文主要介紹了詳解Spring Security中獲取當(dāng)前登錄用戶的詳細(xì)信息的幾種方法,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-05-05
Java實(shí)戰(zhàn)之網(wǎng)上書店管理系統(tǒng)的實(shí)現(xiàn)
本文將利用Java語言實(shí)現(xiàn)網(wǎng)上書店管理系統(tǒng)。其功能一般包括:圖書信息管理、用戶信息管理、圖書購買、圖書訂單查看、圖書添加、圖書維護(hù)等等,感興趣的可以了解一下2022-06-06

