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

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

1.Mapper层

    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
    
        #{id}
    

List> selectscriptPathList(String[] ids);

2.Service层

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

void download(String idStr, HttpServletResponse response);

@Override
public void download(String idStr, HttpServletResponse response) {
    if (StringUtils.isNotEmpty(idStr)) {
        String[] ids = idStr.split(",");
        List> 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层

@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 {

    
    public static void compress(List> 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();
        }
    }

    
    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();
        }
    }
}