java实现文件压缩成zip的工具类

package com.tax.core.util;

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;

/**

* ziputils

* @author gdl

* @date 2017年11月20日

* @version v1.0

*/

public class ziputils {

private static final int buffer_size = 2 * 1024 ;

/**

* 压缩成zip 方法1

* @param srcdir 压缩文件夹路径

* @param out 压缩文件输出流

* @param keepdirstructure 是否保留原来的目录结构,true:保留目录结构;

* false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)

* @throws runtimeexception 压缩失败会抛出运行时异常

*/

public static void tozip(string srcdir, outputstream out, boolean keepdirstructure)

throws runtimeexception{

long start = system.currenttimemillis();

zipoutputstream zos = null ;

try {

zos = new zipoutputstream(out);

file sourcefile = new file(srcdir);

compress(sourcefile,zos,sourcefile.getname(),keepdirstructure);

long end = system.currenttimemillis();

system.out.println( "压缩完成，耗时：" + (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();

}

}

}

}

/**

* 压缩成zip 方法2

* @param srcfiles 需要压缩的文件列表

* @param out 压缩文件输出流

* @throws runtimeexception 压缩失败会抛出运行时异常

*/

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( "压缩完成，耗时：" + (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 是否保留原来的目录结构,true:保留目录结构;

* false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)

* @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输出流中添加一个zip实体，构造器中name为zip实体的文件的名字

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 ){

// 需要保留原来的文件结构时,需要对空文件夹进行处理

if (keepdirstructure){

// 空文件夹的处理

zos.putnextentry( new zipentry(name + "/" ));

// 没有文件，不需要文件的copy

zos.closeentry();

}

} else {

for (file file : listfiles) {

// 判断是否需要保留原来的文件结构

if (keepdirstructure) {

// 注意：file.getname()前面需要带上父文件夹的名字加一斜杠,

// 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了

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( "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);

}

}

if (userlist.size() > 0 ){

/** 下面为下载zip压缩包相关流程 */

httpservletrequest request = servletactioncontext.getrequest();

filewriter writer;

/** 1.创建临时文件夹 */

string rootpath = request.getsession().getservletcontext().getrealpath( "/" );

file temdir = new file(rootpath + "/" + uuid.randomuuid().tostring().replaceall( "-" , "" ));

if (!temdir.exists()){

temdir.mkdirs();

}

/** 2.生成需要下载的文件，存放在临时文件夹内 */

// 这里我们直接来10个内容相同的文件为例，但这个10个文件名不可以相同

for ( int i = 0 ; i < 10 ; i++) {

datamap.put( "userlist" , userlist);

map<string, string> endmap = new hashmap<>();

endmap.put( "user" , "老王" );

endmap.put( "time" , "2017-10-10 10:50:55" );

datamap.put( "endmap" , endmap);

configuration cfg = new configuration(configuration.version_2_3_22);

cfg.setservletcontextfortemplateloading(servletactioncontext.getservletcontext(), "/ftl" );

template template = cfg.gettemplate( "exportexcel.ftl" );

writer = new filewriter(temdir.getpath()+ "/excel" + i + ".xls" );

template.process(datamap, writer);

writer.flush();

writer.close();

}

/** 3.设置response的header */

httpservletresponse response = servletactioncontext.getresponse();

response.setcontenttype( "application/zip" );

response.setheader( "content-disposition" , "attachment; filename=excel.zip" );

/** 4.调用工具类，下载zip压缩包 */

// 这里我们不需要保留目录结构

ziputils.tozip(temdir.getpath(), response.getoutputstream(), false );

/** 5.删除临时文件和文件夹 */

// 这里我没写递归，直接就这样删除了

file[] listfiles = temdir.listfiles();

for ( int i = 0 ; i < listfiles.length; i++) {

listfiles[i].delete();

}

temdir.delete();

}