springMVC实现基本文件夹压缩下载功能

将文件夹压缩后下载:

@Slf4j
public class Test {

    private static final String BASE_PATH = "/root/doc/";

    private static final String TEMP_PATH = "/root/temp/";
    
    private static final String FILE_NAME = "测试";
    
    @RequestMapping(value = "download", method = RequestMethod.GET, produces = {MediaType.APPLICATION_JSON_VALUE})
    public ResponseEntity<byte[]> download() throws Exception {
        fileToZip(BASE_PATH, TEMP_PATH, FILE_NAME);
        File file = new File(TEMP_PATH + "//" + FILE_NAME + ".zip");
        HttpHeaders headers = new HttpHeaders();
        // 为了解决中文名称乱码问题
        String fileName = new String(FILE_NAME.getBytes("UTF-8"), "iso-8859-1");
        headers.setContentDispositionFormData("attachment", fileName + ".zip");
        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
        return new ResponseEntity<>(FileUtils.readFileToByteArray(file), headers, HttpStatus.CREATED);
    }

    private static void fileToZip(String sourceFilePath, String zipFilePath, String fileName) throws Exception {
        File sourceFile = new File(sourceFilePath);
        if (!sourceFile.exists()) {
            log.info("{} >>>> is not exists", sourceFilePath);
            throw new IotBusinessException(ExceptionDefinition.FILE_PATH_NOT_EXISTS.code, ExceptionDefinition.FILE_PATH_NOT_EXISTS.key, "filePathNotExists");
        }
        File file = new File(zipFilePath);
        if (!file.exists()) {
            file.mkdirs();
        }
        File zipFile = new File(zipFilePath + "/" + fileName + ".zip");
        if (zipFile.exists()) {
            // zip文件存在, 将原有文件删除
            zipFile.delete();
        }
        // zip文件不存在, 打包
        pushZip(sourceFile, sourceFilePath, zipFile);
    }

    private static void pushZip(File sourceFile, String sourceFilePath, File zipFile) throws Exception {
        FileInputStream fis;
        BufferedInputStream bis = null;
        FileOutputStream fos;
        ZipOutputStream zos = null;
        try {
            File[] sourceFiles = sourceFile.listFiles();
            if (null == sourceFiles || sourceFiles.length < 1) {
                log.info("{} >>>> is null", sourceFilePath);
            } else {
                fos = new FileOutputStream(zipFile);
                zos = new ZipOutputStream(new BufferedOutputStream(fos));
                byte[] bytes = new byte[1024 * 10];
                for (int i = 0; i < sourceFiles.length; i++) {
                    // 创建ZIP实体，并添加进压缩包  
                    ZipEntry zipEntry = new ZipEntry(sourceFiles[i].getName());
                    zos.putNextEntry(zipEntry);
                    // 读取待压缩的文件并写进压缩包里  
                    fis = new FileInputStream(sourceFiles[i]);
                    bis = new BufferedInputStream(fis, 10240);
                    int read = 0;
                    while ((read = bis.read(bytes, 0, 10240)) != -1) {
                        zos.write(bytes, 0, read);
                    }
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            try {
                if (null != bis) {
                    bis.close();
                }
                if (null != zos) {
                    zos.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}