[Java知识库]Java将多个文件打包成ZIP并下载

Java将多个文件打包成ZIP并下载

需求是多个文件需要同时打包成zip压缩文件并下载到本地，首先我需要的是知道下载文件的路径。我有一个专门的sys_file_info表，表中有对应的文件路径。业务表中的文件地址放的就是文件表的id值。下面是代码：

1.Mapper层

<select id="selectScriptPathList" parameterType="String" resultType="map">
    SELECT i.file_path filePath, i.file_name fileName FROM ves_data_poc p
    LEFT JOIN sys_file_info i ON p.script_url = i.file_id
    WHERE i.file_path IS NOT NULL
    AND p.id IN
    <foreach item="id" collection="array" open="(" separator="," close=")">
        #{id}
    </foreach>
</select>

List<Map<String, String>> selectScriptPathList(String[] ids);

2.Service层

idStr 是前端传来的多个id值用逗号连接的字符串

/**
 * 下载ZIP包
 * @param idStr
 * @param response
 */
void download(String idStr, HttpServletResponse response);

@Override
public void download(String idStr, HttpServletResponse response) {
    if (StringUtils.isNotEmpty(idStr)) {
        String[] ids = idStr.split(",");
        List<Map<String, String>> filePaths = vesDataPocMapper.selectScriptPathList(ids);
        String zipName = "POC_" + ((int) (Math.random() * 10000)) + ".zip";
        String zipPath = Global.getProfile() + "/" + zipName;
        CompressUtil.compress(filePaths, zipPath, false);
        File pocZipFile = new File(zipPath);
        CompressUtil.downloadZip(response, zipName, pocZipFile);
    }
}

3.Controller层

/**
 * 下载poc文件 支持多个下载
 * 格式 zip包
 *
 * @param ids
 */
@Log(title = "数据管理-POC", businessType = BusinessType.EXPORT)
@GetMapping("/download")
public void download(String ids, HttpServletResponse response) {
    vesDataPocService.download(ids, response);
}

4.工具类

import org.apache.commons.lang3.StringUtils;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;


public class CompressUtil {

    /**
     * 生成zip压缩文件
     * @param filePaths
     * @param zipFilePath
     * @param keepDirStructure
     */
    public static void compress(List<Map<String, String>> filePaths, String zipFilePath, Boolean keepDirStructure) {
        byte[] buf = new byte[1024];
        File zipFile = new File(zipFilePath);
        try {
            ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFile));
            for (int i = 0; i < filePaths.size(); i++) {
                String relativePath = filePaths.get(i).get("filePath");
                String relativeName = filePaths.get(i).get("fileName");
                if (StringUtils.isEmpty(relativePath)) {
                    continue;
                }
                File sourceFile = new File(relativePath);
                if (sourceFile == null || !sourceFile.exists()) {
                    continue;
                }
                FileInputStream fis = new FileInputStream(sourceFile);
                if (keepDirStructure != null && keepDirStructure) {
                    zos.putNextEntry(new ZipEntry(relativePath));
                } else {
                    zos.putNextEntry(new ZipEntry(i + "_" + relativeName));
                }
                int len;
                while ((len = fis.read(buf)) > 0) {
                    zos.write(buf, 0, len);
                }
                zos.closeEntry();
                // zos.close();
            }
            zos.close();
            if (!zipFile.exists()) {
                zipFile.createNewFile();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 下载zip
     *
     * @param response
     * @param zipName  浏览器header中zip名称
     * @param zipFile  zipFile文件
     */
    public static void downloadZip(HttpServletResponse response, String zipName, File zipFile) {
        //下载文件
        try {
            response.setCharacterEncoding("utf-8");
            response.setContentType("application/zip");
            response.setHeader("Content-Disposition", "attachment;FileName=" + zipName);
            ServletOutputStream out = response.getOutputStream();
            int len = 0;
            byte[] buffer = new byte[1024];
            FileInputStream fis = new FileInputStream(zipFile);
            while ((len = fis.read(buffer)) > 0) {
                out.write(buffer, 0, len);
                out.flush();
            }
            out.close();
            fis.close();
            response.flushBuffer();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}