文件打包实现

实现逻辑
1、使用java.util.zip.ZipOutputStream构造一个压缩流（zip）
2、将一个或多个文件构造成对应的压缩流的条目（java.util.zip.Entry）
3、将压缩流输出即可。
4、若是多个文件打包则需要考虑文件同名问题。
依赖

    <dependency>
        <groupId>commons-io</groupId>
        <artifactId>commons-io</artifactId>
        <version>2.6</version>
    </dependency>

源码

package com.zhicheng.utils;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.ArrayUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ZipUitls {

    private Logger           logger = LoggerFactory.getLogger(ZipUitls.class);

    /**
     * 批量压缩文件
     * 1、一般获取文件都是在该方法中直接下载，不会直接是使用map传入
     * @param fileIns
     * @param fileName
     * @return
     * @throws IOException
     */
    public static InputStream compress2ZipBatch(Map<File,InputStream> fileIns) throws IOException{
        if(fileIns.isEmpty()) {
            throw new IllegalArgumentException("the param fileIns can not be empty!");
        }
        //处理文件名相同情况
        Map<String,Integer> fileNameMap = new HashMap<String,Integer>();
        // 文件缓存输出流
        ByteArrayOutputStream byteCache = new ByteArrayOutputStream();
        // 压缩流
        ZipOutputStream zipOutStream = new ZipOutputStream(byteCache);
        try {
            for(Entry<File,InputStream> entry:fileIns.entrySet()) {
                addZipEntry(zipOutStream, entry.getValue(), dealSameFileName(fileNameMap,entry.getKey().getName()));
            }
        }finally {
            IOUtils.closeQuietly(zipOutStream);
        }
        return new ByteArrayInputStream(byteCache.toByteArray());
    }

    /**
     * 压缩单个文件
     * @param in
     * @param fileName
     * @return
     * @throws IOException
     */
    public static InputStream compress2Zip(InputStream in,String fileName) throws IOException {
        // 文件输出流
        ByteArrayOutputStream byteCache = new ByteArrayOutputStream();
        // 压缩流
        ZipOutputStream zipOutStream = new ZipOutputStream(byteCache);
        addZipEntry(zipOutStream,in,fileName);
        return new ByteArrayInputStream(byteCache.toByteArray());
    }

    //添加条目
    private static void addZipEntry(ZipOutputStream zos, InputStream in, String fileName) throws IOException{
        try {
            zos.putNextEntry(new ZipEntry(fileName));
            zos.write(IOUtils.toByteArray(in));
            zos.closeEntry();
        }finally {
            IOUtils.closeQuietly(in);
        }
    }

    /**
     * 处理同一包中同名文件
     * 相同的文件名后&后缀前递增加(n)
     * @param fileNameMap
     * @param fileName
     * @return
     */
    private static String dealSameFileName(Map<String,Integer> fileNameMap,String fileName) {
        String _fileName = fileName.substring(0,fileName.lastIndexOf("."));
        String suffix = fileName.substring(fileName.lastIndexOf(".")+1);
        if(fileNameMap.containsKey(fileName)) {
            Integer size = fileNameMap.get(fileName);
            fileNameMap.put(fileName, size.intValue()+1);
            return _fileName+"("+size.intValue()+")."+suffix;
        }else {
            fileNameMap.put(fileName, 1);
            return fileName;
        }
    }

    /**
     * 测试
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws IOException {
        File[] files = new File("D:\\Oss_file\\pdfTest").listFiles();
        if(ArrayUtils.isNotEmpty(files)) {
            Map<File,InputStream> insMap = new HashMap<File,InputStream>();
            for(File file:files) {
                if(file.isFile()) {
                    insMap.put(file,new FileInputStream(file));
                }
            }
            OutputStream out = new FileOutputStream("D:\\Oss_file\\pdfTest\\test\\file.zip");
            IOUtils.copy(compress2ZipBatch(insMap), out);
            IOUtils.closeQuietly(out);
        }
        System.out.println("执行完成!");

    }
}