文件压缩下载

工具类FileUtil

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * 文件工具类
 *
 * @author: 张定辉
 * @date: 2024/5/26 17:45
 * @description: 文件工具类
 */
@Slf4j
public class FileUtil {
    private FileUtil(){}

    private static final int BUFFER_SIZE = 2 * 1024;

    /**
     * 文件夹压缩
     * @param srcDir 文件夹路劲
     * @param out 输出流，可以是一个文件输出流，也可以是HttpServletResponse的输出流
     * @param KeepDirStructure 是否保留目录结构
     */
    @SneakyThrows
    public static void toZip(String srcDir, OutputStream out, boolean KeepDirStructure) {
        long start = System.currentTimeMillis();
        try(ZipOutputStream zos =new ZipOutputStream(out)){
            File sourceFile = new File(srcDir);
            compress(sourceFile, zos, sourceFile.getName(), KeepDirStructure);
            long end = System.currentTimeMillis();
            log.info("压缩完成，耗时: {} ms", end - start);
        }
    }

    /**
     * 递归压缩
     * @param sourceFile       源文件
     * @param zos              zip输出流
     * @param name             压缩后的名称
     * @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
     *                         false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
     */
    @SneakyThrows
    private static void compress(File sourceFile, ZipOutputStream zos, String name, boolean KeepDirStructure) {
        byte[] buf = new byte[BUFFER_SIZE];

        if (sourceFile.isFile()) {
            // 向zip输出流中添加一个zip实体，构造器中name为zip实体的文件的名字
            zos.putNextEntry(new ZipEntry(name));
            // copy文件到zip输出流中
            int len;
            try(FileInputStream in = new FileInputStream(sourceFile)){
                while ((len = in.read(buf)) != -1) {
                    zos.write(buf, 0, len);
                }
                // Complete the entry
                zos.closeEntry();
            }
        } else {
            File[] listFiles = sourceFile.listFiles();
            if (listFiles == null || listFiles.length == 0) {
                // 需要保留原来的文件结构时,需要对空文件夹进行处理
                if (KeepDirStructure) {
                    // 空文件夹的处理
                    zos.putNextEntry(new ZipEntry(name + "/"));
                    // 没有文件，不需要文件的copy
                    zos.closeEntry();
                }
            } else {
                for (File file : listFiles) {
                    // 判断是否需要保留原来的文件结构
                    if (KeepDirStructure) {
                        // 注意：file.getName()前面需要带上父文件夹的名字加一斜杠,
                        // 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了
                        compress(file, zos, name + "/" + file.getName(), KeepDirStructure);
                    } else {
                        compress(file, zos, file.getName(), KeepDirStructure);
                    }

                }
            }
        }
    }
}

Service代码

    @SneakyThrows
    private void  uoloadZip(String folderPath){
        String zipPath=folderPath+".zip";
        //压缩附件和元数据
        FileUtil.toZip(folderPath, Files.newOutputStream(Paths.get(zipPath)),true);
        HttpServletResponse response = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getResponse();
        response.setContentType("applicaation/octet-stream");
        response.addHeader("Content-disposition", "attachment;filename=" +UUID.randomUUID());
        IOUtils.copy(new FileInputStream(zipPath),response.getOutputStream());
    }