Java下載壓縮包(zip)

package com.golden.crm.web.util;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;


/**
 * 下載zip工具類
 */
public class ZipUtil {

    private static final Logger logger = LoggerFactory.getLogger(ZipUtil.class);

    /**
     * 生成Zip文件
     */
    public static void generateZipFile(HttpServletResponse response, List<String> fileUrl, String fileName) throws IOException {

        ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream());

        for (int i = 0; i < fileUrl.size(); i++) {

            zipOut.putNextEntry(new ZipEntry(System.currentTimeMillis() + ".jpg"));

            InputStream stream = this.getInputStreamByUrl(fileUrl.get(i));

            if (null == stream) {
                continue;
            }

            int temp = 0;
            while ((temp = stream.read()) != -1) {
                zipOut.write(temp);
            }
            stream.close();
        }

        zipOut.closeEntry();
        zipOut.close();
    }

    /**
     * 經過url讀取圖片信息
     */
    public static InputStream getInputStreamByUrl(String url) {

        InputStream ins = null;

        try {

            URL url_ = new URL(url);
            HttpURLConnection conn = (HttpURLConnection) url_.openConnection();
            conn.setConnectTimeout(3 * 1000);
            conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");

            ins = conn.getInputStream();

        } catch (IOException e) {
            logger.info("圖片讀取失敗!-->" + e);
        }

        return ins;
    }

    /**
     * 導出Zip文件
     */
    public static void exportZipFile(String fileName, List<String> fileUrl, HttpServletResponse response, HttpServletRequest request) {

        try {

            // 瀏覽器處理亂碼問題
            String userAgent = request.getHeader("User-Agent");

            // filename.getBytes("UTF-8")處理safari的亂碼問題
            byte[] bytes = userAgent.contains("MSIE") ? fileName.getBytes() : fileName.getBytes("UTF-8");

            // 各瀏覽器基本都支持ISO編碼
            fileName = new String(bytes, "ISO-8859-1");

            // 文件名外的雙引號處理firefox的空格截斷問題
            response.setHeader("Content-disposition", String.format("attachment; filename=\"%s\"", fileName));
            response.setContentType("application/x-msdownload");
            response.setCharacterEncoding("UTF-8");
            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);

            this.generateZipFile(response, fileUrl, fileName);

        } catch (Exception e) {
            logger.info("下載失敗!-->" + e);
        }
    }

}
相關文章
相關標籤/搜索