将多个文件打包成ZIP压缩包输出

工具类一(将多个远端文件读取并压缩)

import java.io.BufferedInputStream;

import java.io.BufferedOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.net.HttpURLConnection;

import java.net.URL;

import java.util.Map;

import java.util.Map.Entry;

import java.util.Set;

import java.util.zip.ZipEntry;

import java.util.zip.ZipOutputStream;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServletResponse;

public class ZipUtil {

/**

* 压缩文件

* @param files 键值对-》(文件名:文件链接)

* @param outputStream

* @throws Exception

* @throws IOException

*/

public static void zipPersonPhotoFile(Map files, ZipOutputStream outputStream) {

    try {

    Set> entrySet = files.entrySet();

        for (Entry file:entrySet) {

            try {

                zipFile(getImgIs(file.getValue()),file.getKey(), outputStream);

            } catch (Exception e) {

                continue;

            }

        }

    } catch (Exception e) {

        e.printStackTrace();

    } finally {

    try {

outputStream.close();

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

    }

}

/**

* 将文件写入到zip文件中

* @param inputFile

* @param outputstream

* @throws Exception

*/

public static void zipFile(InputStream is, String fileName, ZipOutputStream outputstream) throws IOException, ServletException {

    try {

        if (is != null) {

                BufferedInputStream bInStream = new BufferedInputStream(is);

                ZipEntry entry = new ZipEntry(fileName);

                outputstream.putNextEntry(entry);

                int len = 0 ;

                byte[] buffer = new byte[10 * 1024];

                while ((len = is.read(buffer)) > 0) {

                outputstream.write(buffer, 0, len);

                outputstream.flush();

                }

                outputstream.closeEntry();//Closes the current ZIP entry and positions the stream for writing the next entry

                bInStream.close();//关闭

                is.close();

        } else {

            throw new ServletException("文件不存在!");

        }

    } catch (IOException e) {

        throw e;

    }

}

    /**

    * 获取文件流

    */

public static InputStream getImgIs(String imgURL) throws IOException{

//new一个URL对象

URL url = new URL(imgURL);

//打开链接 

        HttpURLConnection conn = (HttpURLConnection)url.openConnection(); 

        //设置请求方式为"GET" 

        conn.setRequestMethod("GET");

        //超时响应时间为5秒 

        conn.setConnectTimeout(5 * 1000); 

        //通过输入流获取图片数据 

        return conn.getInputStream();

}

/**

* 下载打包的文件

*

* @param file

* @param response

*/

public static void downloadZip(File file, HttpServletResponse response) {

    try {

        // 以流的形式下载文件。

        BufferedInputStream fis = new BufferedInputStream(new FileInputStream(file.getPath()));

        byte[] buffer = new byte[fis.available()];

        fis.read(buffer);

        fis.close();

        // 清空response

        response.reset();

        OutputStream toClient = new BufferedOutputStream(response.getOutputStream());

        response.setCharacterEncoding("UTF-8");

        response.setContentType("application/octet-stream");

        response.setHeader("Content-Disposition", "attachment;filename=" + file.getName());

        toClient.write(buffer);

        toClient.flush();

        toClient.close();

        file.delete();

    } catch (IOException ex) {

        ex.printStackTrace();

    }

}

}

/****************************

测试代码

ZipUtil.zipPersonPhotoFile(map, new ZipOutputStream(new FileOutputStream(new File(zipFilePath))));


/**response.setContentType("application/x-download");

response.addHeader("Content-Disposition", "attachment;filename="+new String("中文".getBytes("utf-8"), "ISO8859-1" )+".zip");*/

//response.setContentType("application/octet-stream");

//直接将压缩后的文件包以流的方式返出

ZipUtil.zipPersonPhotoFile(files, new ZipOutputStream(response.getOutputStream()));

/***********************************

工具类二(将本地某个文件/文件夹压缩)

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.Map;

import java.util.Map.Entry;

import java.util.Set;

import java.util.zip.ZipEntry;

import java.util.zip.ZipOutputStream;

import javax.servlet.ServletException;

import org.apache.commons.io.FileUtils;

public class ZipUtils {

  private ZipUtils() {

    throw new IllegalStateException("Utility class");

  }

  /**

  * 将文件/目录进行压缩

  * @param sourceFile 原文件/目录

  * @param targetZipFile 压缩后目标文件

  * @throws IOException

  */

  public static void zipFiles(File sourceFile, File targetZipFile) throws IOException {

    ZipOutputStream outputStream = null;

    try {

      outputStream = new ZipOutputStream(new FileOutputStream(targetZipFile));

      addEntry("", sourceFile, outputStream);

    } catch (Exception e) {

      throw new IOException(e);

    } finally {

      outputStream.close();

    }

  }

  /**

  * 将文件写入到zip文件中

  * @param source

  * @param outputstream

  * @throws IOException

  * @throws ServletException

  */

  private static void addEntry(String base, File source, ZipOutputStream outputstream)

      throws IOException, ServletException {

    FileInputStream is = null;

    try {

      String entry = base + source.getName();

      if (source.isDirectory()) {

        for (File file : source.listFiles()) {

          // 递归导入文件

          addEntry(entry + File.separator, file, outputstream);

        }

      } else {

        is = FileUtils.openInputStream(source);

        if (is != null) {

          outputstream.putNextEntry(new ZipEntry(entry));

          int len = 0;

          byte[] buffer = new byte[10 * 1024];

          while ((len = is.read(buffer)) > 0) {

            outputstream.write(buffer, 0, len);

            outputstream.flush();

          }

          outputstream.closeEntry();

        }

      }

    } catch (IOException e) {

      throw e;

    } finally {

      if (is != null) {

        is.close();

      }

    }

  }

}

你可能感兴趣的:(将多个文件打包成ZIP压缩包输出)