【File操作】Java实现导出ZIP压缩包

怼烎@ 2022-11-21 11:58 128阅读 0赞

最近接触到一个需求要求压缩导出文件，于是乎便要致力于研究一下工具类啦，故也诞生了此篇文章。  
下面代码中，溪源也将`import`导入的依赖也贴出来了，避免大家引入错误。

import org.apache.commons.io.FileUtils;
    import org.apache.commons.io.IOUtils;
    
    import javax.servlet.http.HttpServletResponse;
    import java.io.*;
    import java.text.SimpleDateFormat;
    import java.util.Date;
    import java.util.zip.ZipEntry;
    import java.util.zip.ZipOutputStream;
    
    /** * @author wx * @date 2020/10/29 5:19 下午 */
    public class FileZipUtil { 
    
        private static void handlerFile(ZipOutputStream zip, File file, String dir) throws Exception { 
            //如果当前的是文件夹，则进行进一步处理
            if (file.isDirectory()) { 
                //得到文件列表信息
                File[] fileArray = file.listFiles();
                if (fileArray == null) { 
                    return;
                }
                //将文件夹添加到下一级打包目录
                zip.putNextEntry(new ZipEntry(dir + "/"));
                dir = dir.length() == 0 ? "" : dir + "/";
                //递归将文件夹中的文件打包
                for (File f : fileArray) { 
                    handlerFile(zip, f, dir + f.getName());
                }
            } else { 
                //当前的是文件，打包处理
                //文件输入流
                BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
                ZipEntry entry = new ZipEntry(dir);
                zip.putNextEntry(entry);
                zip.write(FileUtils.readFileToByteArray(file));
                IOUtils.closeQuietly(bis);
                zip.flush();
                zip.closeEntry();
            }
        }
    
        private static byte[] createZip(String sourceFilePath) throws Exception{ 
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
            ZipOutputStream zip = new ZipOutputStream(outputStream);
            //将目标文件打包成zip导出
            File file = new File(sourceFilePath);
            handlerFile(zip, file,"");
            IOUtils.closeQuietly(zip);
            return outputStream.toByteArray();
        }
    
    
        public static void exportZip(HttpServletResponse response, String sourceFilePath) { 
        //文件名以时间戳作为前缀
            SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
            String filePrefix = sdf.format(new Date());
            String downloadName = filePrefix + ".zip";
            //将文件进行打包下载
            try { 
                OutputStream out = response.getOutputStream();
                //接收压缩包字节
                byte[] data = createZip(sourceFilePath);
                response.reset();
                response.addHeader("Access-Control-Allow-Origin", "*");
                response.setHeader("Access-Control-Expose-Headers", "*");
                response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + downloadName);
                response.addHeader("Content-Length", "" + data.length);
                response.setContentType("application/octet-stream;charset=UTF-8");
                IOUtils.write(data, out);
                out.flush();
                out.close();
            } catch (Exception e) { 
                e.printStackTrace();
            }
        }
    
    }

客户端调用方法：

@GetMapping("/exportFile")
        public Result exportFile(HttpServletResponse response) { 
            //第二个参数为：要压缩文件的地址
            FileZipUtil.exportZip(response, "/Users/Downloads");
        }