java生成zip压缩包文件

生成压缩包工具生成代码

package com.util;

import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.FileCopyUtils;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.nio.file.Files;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.Locale;
import java.util.zip.Adler32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * 生成压缩包
 */
@Slf4j
public class ZipCompressor {

    @SneakyThrows
    public static void getZip(String zipName, File zipFile, HttpServletRequest request, HttpServletResponse response, File... files) {
        try (
                // 作用是为任何OutputStream产生校验和
                FileOutputStream fos = new FileOutputStream(zipFile);
                // 第一个参数是制定产生校验和的输出流，第二个参数是指定Checksum的类型 （Adler32（较快）和CRC32两种）
                CheckedOutputStream cos = new CheckedOutputStream(fos, new Adler32());
                // 用于将数据压缩成zip文件格式
                ZipOutputStream zos = new ZipOutputStream(cos)
        ) {
            // 循环根据路径从OSS获得对象，存入临时文件zip中
            for (File file : files) {
                compressFile(zos, file, file.getName());
            }
            zos.close();

            String header = request.getHeader(CommonConstants.USER_AGENT).toUpperCase(Locale.ROOT);
            // IE下载文件名空格变+号问题
            zipName = URLEncoder.encode(zipName, CommonConstants.UTF_8);
            zipName = zipName.replace("+", "%20");

            // 设置响应类型，以附件形式下载文件
            response.reset();
            response.setContentType("text/plain");
            response.setContentType("application/octet-stream; charset=utf-8");
            response.setHeader("Location", zipName);
            response.setHeader("Cache-Control", "max-age=0");
            response.setHeader("Content-Disposition", "attachment; filename=" + zipName);
            response.addHeader("Access-Control-Allow-Origin", "*");
            response.addHeader("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE");
            response.addHeader("Access-Control-Allow-Headers", "Content-Type");
            response.addHeader("Access-Control-Expose-Headers", "Content-Disposition");
            // 下载流封装
            try (FileInputStream fis = new FileInputStream(zipFile);
                 BufferedInputStream bis = new BufferedInputStream(fis);
                 BufferedOutputStream bos = new BufferedOutputStream(response.getOutputStream())) {
                // 下载zip文件
                FileCopyUtils.copy(bis, bos);

                // 删除临时zip文件
                boolean delete = zipFile.delete();
            }
        } catch (Exception exception) {
            log.error("getZip error: ", exception);
        }
    }

    private static void compressFile(ZipOutputStream zos, File sourceFile, String baseName) throws IOException {
        if (!sourceFile.exists()) {
            return;
        }
        // 若路径为目录（文件夹）
        BasicFileAttributes basicFileAttributes = Files.readAttributes(sourceFile.toPath(), BasicFileAttributes.class);
        if (basicFileAttributes.isDirectory()) {
            // 取出文件夹中的文件（或子文件夹）
            File[] fileList = sourceFile.listFiles();
            // 若文件夹为空，则创建一个目录进入点
            assert fileList != null;
            if (fileList.length == 0) {
                // 文件名称后跟File.separator表示这是一个文件夹
                zos.putNextEntry(new ZipEntry(baseName + File.separator));
                // 若文件夹非空，则递归调用compressFile,对文件夹中的每个文件或每个文件夹进行压缩
            } else {
                for (File file : fileList) {
                    compressFile(zos, file, baseName + File.separator + file.getName());
                }
            }
            // 若为文件,则先创建目录进入点,再将文件写入zip文件中
        } else {
            ZipEntry zipEntry = new ZipEntry(baseName);
            // 设置ZipEntry的最后修改时间为源文件的最后修改时间
            zipEntry.setTime(basicFileAttributes.lastModifiedTime().toMillis());
            zos.putNextEntry(zipEntry);
            try (FileInputStream fis = new FileInputStream(sourceFile)) {
                copyStream(fis, zos);
            } catch (Exception exception) {
                log.error("输入流关闭异常：", exception);
            }
        }
    }

    /**
     * 流拷贝
     *
     * @param inputStream  输入流
     * @param outputStream 输出流
     */
    private static void copyStream(InputStream inputStream, OutputStream outputStream) throws IOException {
        int bufferLength = 1024 * 100;
        int count;
        byte[] buffer = new byte[bufferLength];
        while ((count = inputStream.read(buffer, 0, bufferLength)) != -1) {
            outputStream.write(buffer, 0, count);
        }
        outputStream.flush();
    }
}

应用代码

java

@RequestMapping(value = "/getZip", method = RequestMethod.POST)
    public void getZip(@RequestParam("name") String name, HttpServletRequest request, HttpServletResponse response) {
        if (StringUtils.isBlank(name)) {
            log.info("getZip error: name is not empty");
            return;
        }
        File file = new File("文件夹路径");
        File subFile = new File(file, "子文件夹名称");
        String zipName = eventName + ".zip";
        try {
            // 创建临时文件
            File zipFile = FileUtil.createTempFile(eventName, ".zip", true);
            // 下载文件为zip压缩包
            ZipCompressor.getZip(zipName, zipFile, request, response, subFile);
            log.info("getZip end.");
        } catch (Exception exception) {
            log.error("getZip error: ", exception);
        } finally {
            // 删除生成的文件和文件夹
            if (file.exists()) {
            try {
                FileUtils.forceDelete(file);
            } catch (Exception exception) {
                log.error("forceDelete error ", exception);
            }
        }
        }
    }

react

jsonSchema2pojo = () => {
    const formData = new FormData();
    formData.append('name', "名称");
    const xhr = new XMLHttpRequest();
    xhr.open('POST', apiPrefix + '/getZip', true);
    xhr.responseType = 'blob';
    xhr.onload = function () {
      if (xhr.status === 200) {
        let filename
        var type = xhr.getResponseHeader('content-type')
        // 后台返回了名字则用后台的名字
        var contentDisposition = xhr.getResponseHeader('content-disposition')
        if (!filename && contentDisposition) {
          var idx = contentDisposition.indexOf('filename=')
          if (idx > -1) {
            filename = decodeURIComponent(contentDisposition.substring(idx + 9))
          }
        }
        var blob = xhr.response
        if (window.navigator && window.navigator.msSaveOrOpenBlob) {
          window.navigator.msSaveBlob(blob, filename)
        } else {
          var downloadElement = document.createElement('a')
          var href = window.URL.createObjectURL(blob, {
            type: type
          })
          // 创建下载的链接
          downloadElement.href = href
          // 下载后文件名
          downloadElement.download = filename
          document.body.appendChild(downloadElement)
          // 点击下载
          downloadElement.click()
          // 下载完成移除元素
          document.body.removeChild(downloadElement)
          // 释放掉blob对象
          window.URL.revokeObjectURL(href)
        }
      }
    }
    xhr.send(formData);
  }