Java实现把文件及文件夹压缩成zip
最近碰到个需要下载zip压缩包的需求,于是我在网上找了下别人写好的zip工具类。但找了好多篇博客,总是发现有bug。因此就自己来写了个工具类。
这个工具类的功能为:
(1)可以压缩文件,也可以压缩文件夹
(2)同时支持压缩多级文件夹,工具内部做了递归处理
(3)碰到空的文件夹,也可以压缩
(4)可以选择是否保留原来的目录结构,如果不保留,所有文件跑压缩包根目录去了,且空文件夹直接舍弃。注意:如果不保留文件原来目录结构,在碰到文件名相同的文件时,会压缩失败。
(5)代码中提供了2个压缩文件的方法,一个的输入参数为文件夹路径,一个为文件列表,可根据实际需求选择方法。
下面直接上代码
一、代码
ZipUtils
importjava.io.File;
importjava.io.FileInputStream;
importjava.io.FileOutputStream;
importjava.io.IOException;
importjava.io.OutputStream;
importjava.util.ArrayList;
importjava.util.List;
importjava.util.zip.ZipEntry;
importjava.util.zip.ZipOutputStream;
/**
*@authorNemo
*@version1.0
*@date2019/11/5
*/
publicclassZipUtils{
privatestaticfinalintBUFFER_SIZE=2*1024;
/**
*压缩成ZIP方法1
*@paramsourceFile压缩文件夹路径
*@paramout压缩文件输出流
*@paramKeepDirStructure是否保留原来的目录结构,true:保留目录结构;
*false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败)
*@throwsRuntimeException压缩失败会抛出运行时异常
*/
publicstaticvoidtoZip(FilesourceFile,OutputStreamout,booleanKeepDirStructure)
throwsRuntimeException{
ZipOutputStreamzos=null;
try{
zos=newZipOutputStream(out);
compress(sourceFile,zos,sourceFile.getName(),KeepDirStructure);
}catch(Exceptione){
thrownewRuntimeException("ziperrorfromZipUtils",e);
}finally{
if(zos!=null){
try{
zos.close();
}catch(IOExceptione){
e.printStackTrace();
}
}
}
}
/**
*压缩成ZIP方法2
*@paramsrcFiles需要压缩的文件列表
*@paramout压缩文件输出流
*@throwsRuntimeException压缩失败会抛出运行时异常
*/
publicstaticvoidtoZip(ListsrcFiles,OutputStreamout)throwsRuntimeException{
longstart=System.currentTimeMillis();
ZipOutputStreamzos=null;
try{
zos=newZipOutputStream(out);
for(FilesrcFile:srcFiles){
byte[]buf=newbyte[BUFFER_SIZE];
zos.putNextEntry(newZipEntry(srcFile.getName()));
intlen;
FileInputStreamin=newFileInputStream(srcFile);
while((len=in.read(buf))!=-1){
zos.write(buf,0,len);
}
zos.closeEntry();
in.close();
}
longend=System.currentTimeMillis();
System.out.println("压缩完成,耗时:"+(end-start)+"ms");
}catch(Exceptione){
thrownewRuntimeException("ziperrorfromZipUtils",e);
}finally{
if(zos!=null){
try{
zos.close();
}catch(IOExceptione){
e.printStackTrace();
}
}
}
}
/**
*递归压缩方法
*@paramsourceFile源文件
*@paramzoszip输出流
*@paramname压缩后的名称
*@paramKeepDirStructure是否保留原来的目录结构,true:保留目录结构;
*false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败)
*@throwsException
*/
privatestaticvoidcompress(FilesourceFile,ZipOutputStreamzos,Stringname,
booleanKeepDirStructure)throwsException{
byte[]buf=newbyte[BUFFER_SIZE];
if(sourceFile.isFile()){
//向zip输出流中添加一个zip实体,构造器中name为zip实体的文件的名字
zos.putNextEntry(newZipEntry(name));
//copy文件到zip输出流中
intlen;
FileInputStreamin=newFileInputStream(sourceFile);
while((len=in.read(buf))!=-1){
zos.write(buf,0,len);
}
//Completetheentry
zos.closeEntry();
in.close();
}else{
File[]listFiles=sourceFile.listFiles();
if(listFiles==null||listFiles.length==0){
//需要保留原来的文件结构时,需要对空文件夹进行处理
if(KeepDirStructure){
//空文件夹的处理
zos.putNextEntry(newZipEntry(name+"/"));
//没有文件,不需要文件的copy
zos.closeEntry();
}
}else{
for(Filefile:listFiles){
//判断是否需要保留原来的文件结构
if(KeepDirStructure){
//注意:file.getName()前面需要带上父文件夹的名字加一斜杠,
//不然最后压缩包中就不能保留原来的文件结构,即:所有文件都跑到压缩包根目录下了
compress(file,zos,name+"/"+file.getName(),KeepDirStructure);
}else{
compress(file,zos,file.getName(),KeepDirStructure);
}
}
}
}
}
publicstaticvoidmain(String[]args)throwsException{
/**测试压缩方法1*/
FileOutputStreamfos1=newFileOutputStream(newFile("c:/mytest01.zip"));
ZipUtils.toZip(newFile("D:/log"),fos1,true);
/**测试压缩方法2*/
ListfileList=newArrayList<>();
fileList.add(newFile("D:/Java/jdk1.7.0_45_64bit/bin/jar.exe"));
fileList.add(newFile("D:/Java/jdk1.7.0_45_64bit/bin/java.exe"));
FileOutputStreamfos2=newFileOutputStream(newFile("c:/mytest02.zip"));
ZipUtils.toZip(fileList,fos2);
}
}
二、注意事项
写该工具类时,有些注意事项说一下:
(1)支持选择是否保留原来的文件目录结构,如果不保留,那么空文件夹直接不用处理。
(1)碰到空文件夹时,如果需要保留目录结构,则直接添加个ZipEntry就可以了,不过就是这个entry的名字后面需要带上一斜杠(/)表示这个是目录。
(2)递归时,不需要把zip输出流关闭,zip输出流的关闭应该是在调用完递归方法后面关闭
(3)递归时,如果是个文件夹且需要保留目录结构,那么在调用方法压缩他的子文件时,需要把文件夹的名字加一斜杠给添加到子文件名字前面,这样压缩后才有多级目录。
三、如何在javaWeb项目中使用该工具类
这个工具类在web项目中的使用场景就是多文件下载,我就简单说个下载多个excel表格的案例吧。
代码中的步骤为:
(1)创建一个临时文件夹
(2)将要下载的文件生成至该临时文件夹内
(3)当所有文件生成完后,获取HttpServletResponse获取设置下载的header
(4)调用工具类的方法,传入上面生成的临时文件夹路径及response获取的输出流;这样就下载出来zip包了
(5)递归删除掉上面生成的临时文件夹和文件
下面为一个示例代码的代码片段,不是完整代码,简单看一下代码中的步骤
importorg.apache.commons.io.FileUtils;
importjava.io.*;
/**
*图片打包下载
*@author:wangzhouchao
*/
@ApiImplicitParams({
@ApiImplicitParam(name="id",value="申请人id",required=true,dataType="Long",paramType="query"),
})
@ApiOperation(value="图片打包下载",notes="图片打包下载")
@RequestMapping(value="/downloadPictureList",method=RequestMethod.GET)
publicvoiddownloadPictureList(TProposerDataVOtProposerDataVO){
longreadyStart=System.currentTimeMillis();
//*************1.获取到存在数据库中的图片的url*************
PictureDownloadVOpicturesById=tOrderService.getPicturesByProposerDataId(tProposerDataVO.getId());
//获取当前类的所在项目路径
Filefile=null;
try{
file=newFile(ResourceUtils.getURL("classpath:").getPath());
}catch(FileNotFoundExceptione){
thrownewRuntimeException("获取根目录失败,无法获取文件目录!");
}
if(!file.exists()){
file=newFile("");
}
StringabsolutePath=file.getAbsolutePath();
//要打包的文件夹列表
Stringorder_number=picturesById.getOrder_number();
Stringcountry_name=picturesById.getCountry_name();
Stringvisa_type=picturesById.getVisa_type();
StringdirName=order_number+country_name+visa_type;
//*************2.创建要压缩的文件夹*************
//根据订单号+国家名称+签证类型创建文件夹
FiledirOfOrder=newFile(absolutePath,dirName);
if(!dirOfOrder.exists()){
dirOfOrder.mkdirs();
}
ZipOutputStreamzos=null;
OutputStreamout=null;
longreadyEnd=System.currentTimeMillis();
System.out.println("准备完成,耗时:"+(readyEnd-readyStart)+"ms");
try{
longdownStart=System.currentTimeMillis();
System.out.println("开始下载");
TProposerDataVOvo=picturesById.getProposerDataVO();
//*************3.根据获取到的图片的url,把图片按照想要的文件夹目录进行下载*************
//根据申请人姓名创建文件夹
FileproposerFile=newFile(dirOfOrder,vo.getReal_name());
if(!proposerFile.exists()){
proposerFile.mkdirs();
}
//下载申请人照片
if(StringUtil.checkNotNull(vo.getPhoto_url())){
System.out.println("开始下载申请人照片");
WordExportUtil.downloadHttpUrl(DOMAIN+vo.getPhoto_url(),proposerFile.toString(),File.separator+"photo.jpg");
}
//下载申请人护照首页
if(StringUtil.checkNotNull(vo.getPassport_home_page_url())){
System.out.println("开始下载申请人护照照片");
WordExportUtil.downloadHttpUrl(DOMAIN+vo.getPassport_home_page_url(),proposerFile.toString(),File.separator+"passport.jpg");
}
//下载申请人户口本照片
if(StringUtil.checkNotNull(vo.getResidence_booklet_url())){
System.out.println("开始下载申请人户口本照片");
String[]booklets=vo.getResidence_booklet_url().split(",");
//创建户口本照片文件夹
FilebookletsFile=newFile(proposerFile,"hukouben");
if(!bookletsFile.exists()){
bookletsFile.mkdirs();
}
for(intk=0;k
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。