spring mvc 批量下载压缩文件夹中文件

一、工具类

package com.juanpi.service.util;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import javax.servlet.http.HttpServletResponse;

public class CustomFileUtil {
    /**
     *
     * @param inputFileName
     *            输入一个文件夹
     * @param zipFileName
     *            输出一个压缩文件夹，打包后文件名字
     * @throws Exception
     */
    public static void zip(String inputFileName, String zipFileName) throws Exception {
        zip(zipFileName, new File(inputFileName));
    }

    private static void zip(String zipFileName, File inputFile) throws Exception {
        ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipFileName));
        zip(out, inputFile, "");
        out.close();
    }

    private static void zip(ZipOutputStream out, File f, String base) throws Exception {
        if (f.isDirectory()) { // 判断是否为目录
            File[] fl = f.listFiles();
            out.putNextEntry(new ZipEntry(base + "/"));
            base = base.length() == 0 ? "" : base + "/";
            for (int i = 0; i < fl.length; i++) {
                zip(out, fl[i], base + fl[i].getName());
            }
        } else { // 压缩目录中的所有文件
            out.putNextEntry(new ZipEntry(base));
            FileInputStream in = new FileInputStream(f);
            int b;
            while ((b = in.read()) != -1) {
                out.write(b);
            }
            in.close();
        }
    }

    /**
     * 下载文件
     *
     * @param file
     * @param response
     */
    public static void downloadFile(File file, HttpServletResponse response, boolean isDelete) {
        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.setContentType("application/octet-stream");
            response.setHeader("Content-Disposition",
                    "attachment;filename=" + new String(file.getName().getBytes("UTF-8"), "ISO-8859-1"));
            toClient.write(buffer);
            toClient.flush();
            toClient.close();
            if (isDelete) {
                file.delete(); // 是否将生成的服务器端文件删除
            }
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

}

二、restful接口

/**
 * 批量打包下载文件生成zip文件下载
 *
 * @param session
 */
@RequestMapping(value = "/loadFiles", method = RequestMethod.GET)
public void downloadFiles(HttpServletRequest request, HttpServletResponse response,String param)
        throws ServletException, IOException {
    //String inputFileName = realPath; 可配置文件中配置磁盘路径,路径传参数拼接，得到文件夹下的子文件夹
    String inputFileName = "C:/test/";
    if (domain != null && !domain.trim().equals("")) {
        inputFileName = realPath + domain.toUpperCase();
    }
    logger.info("download file directory is :" + inputFileName);
    String zipFileName = realPath + "temp.zip"; // 压缩后的zip文件 可随意定一个磁盘路径或者相对路径
    try {
        CustomFileUtil.zip(inputFileName, zipFileName);
    } catch (Exception ex) {
        logger.error(ex.getMessage());
    }
    File temp = new File(zipFileName);
    CustomFileUtil.downloadFile(temp, response, true);
}