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

柔情只为你懂 2023-09-28 16:31 162阅读 0赞

需求是多个文件需要同时打包成zip压缩文件并下载到本地，首先我需要的是知道下载文件的路径。我有一个专门的sys\_file表，表中有对应的文件路径。通过前段传来的文件ids查询出文件的路径。下面是代码：

### 1.Mapper层 ###

<select id="selectScriptPathList" parameterType="String" resultType="map">
        SELECT i.file_path filePath, i.file_name fileName FROM sys_file_info i
        WHERE i.id IN
        <foreach item="id" collection="array" open="(" separator="," close=")">
            #{
        id}
        </foreach>
    </select>

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

### 2.Service层 ###

@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层 ###

/**
     * 下载文件 支持多个下载
     * 格式 zip包
     *
     * @param ids
     */
    @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();
            }
        }
    }