利用Java进行zip文件压缩与解压缩

可能存在的业务情况：

1、用户上传了压缩包，需校验压缩包中的文件是否合格。

2、用户上传压缩包，对压缩包中的文件进行批量水印处理

解决思路：

1、读取原压缩包文件，解压缩至临时目录

2、对临时目录中的解压缩文件进行校验/水印处理

3、对临时目录中处理过的文件进行压缩

4、删除临时目录及其下的文件

需要参考前面的校验PDF的帖子

测试结果如下：

测试代码：

@Test
    public void testZip() {
        String filePath = "D:\pdfTest\合格压缩文件.zip";
        // 获取原文所在目录
        // 服务器上时，文件路径为“/”，此处测试需要换成filePath中的“\”
        //String oldFilePath = filePath.substring(0, filePath.lastIndexOf("/"));
        String oldFilePath = filePath.substring(0, filePath.lastIndexOf("\"));
        System.out.println("原文件路径：" + oldFilePath);
        // 临时目录，原压缩文件解压目录
        String destDirPath = oldFilePath + "\tmp\";
        System.out.println("临时路径：" + destDirPath);
        // 将原压缩文件解压到临时目录
        ZipUtil.unzipFile(filePath, destDirPath);

        // 临时目录文件对象
        File destDir = new File(destDirPath);
        // 获取临时目录下的所有文件
        File[] files = destDir.listFiles();
        // 定义变量，保存校验结果
        List<Integer> list = new ArrayList<>();
        // 遍历文件，进行校验
        for (File file: files) {
            String absolutePath = file.getAbsolutePath();
            System.out.println(absolutePath);
            int i = CheckPdfHelper.checkPdf(absolutePath);
            list.add(i);
            // 压缩包中存在不合格PDF文件时
            if (i != 0) {
                break;
            }
        }
        // 判断是否包含不合格PDF文件
        if (list.contains(1)) {
            System.out.println("压缩文件中包含不合格PDF文件");
            // 删除解压缩的文件和临时目录
            ZipUtil.deletefile(destDirPath);
            // 不合格时，不生成新的压缩包文件
            return;
        } else {
            System.out.println("压缩文件PDF文件均符合要求");
        }

        // 获取原压缩文件后缀
        int pos = filePath.lastIndexOf('.');
        String suffix = filePath.substring(pos + 1);
        // 新生成压缩文件路径
        String newFilePath = filePath.substring(0, pos) + ".PSW." + suffix;
        System.out.println("新的压缩文件路径：" + newFilePath);

        // 将检验成功的文件压缩成一个新的压缩包
        ZipUtil.zipFile(newFilePath, files);
        // 删除临时目录
        ZipUtil.deletefile(destDirPath);
    }

ZipUtil工具类：

package com.alphajuns.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;

import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;

/**
 * @ClassName ZipUtil
 * @Description 压缩或解压缩zip：由于直接使用java.util.zip工具包下的类，会出现中文乱码问题，所以使用ant.jar中的org.apache.tools.zip下的工具类
 * @Author AlphaJunS
 * @Date 2020/3/8 11:30
 * @Version 1.0
 */
public class ZipUtil {

    /**
     * @Author AlphaJunS
     * @Date 11:32 2020/3/8
     * @Description
      * @param zip 压缩目的地址
     * @param srcFiles 压缩的源文件
     * @return void
     */
    public static void zipFile( String zip , File[] srcFiles ) {
        try {
            if( zip.endsWith(".zip") || zip.endsWith(".ZIP") ){
                FileOutputStream fos = new FileOutputStream(new File(zip));
                ZipOutputStream _zipOut = new ZipOutputStream(fos) ;
                _zipOut.setEncoding("GBK");
                for( File _f : srcFiles ){
                    handlerFile(zip , _zipOut , _f , "");
                }
                fos.close();
                _zipOut.close();
            }else{
                System.out.println("target file[" + zip + "] is not .zip type file");
            }
        } catch (FileNotFoundException e) {
        } catch (IOException e) {
        }
    }

    /**
     * @Author AlphaJunS
     * @Date 11:33 2020/3/8
     * @Description
     * @param zip 压缩的目的地址
     * @param zipOut
     * @param srcFile 被压缩的文件信息
     * @param path 在zip中的相对路径
     * @return void
     */
    private static void handlerFile(String zip , ZipOutputStream zipOut , File srcFile , String path) throws IOException {
        System.out.println(" begin to compression file[" + srcFile.getName() + "]");
        if( !"".equals(path) && ! path.endsWith(File.separator)){
            path += File.separator ;
        }
        if( ! srcFile.getPath().equals(zip) ){
            if( srcFile.isDirectory() ){
                File[] _files = srcFile.listFiles() ;
                if( _files.length == 0 ){
                    zipOut.putNextEntry(new ZipEntry( path + srcFile.getName() + File.separator));
                    zipOut.closeEntry();
                }else{
                    for( File _f : _files ){
                        handlerFile( zip ,zipOut , _f , path + srcFile.getName() );
                    }
                }
            }else{
                InputStream _in = new FileInputStream(srcFile) ;
                zipOut.putNextEntry(new ZipEntry(path + srcFile.getName()));
                int len = 0 ;
                byte[] _byte = new byte[1024];
                while( (len = _in.read(_byte)) > 0  ){
                    zipOut.write(_byte, 0, len);
                }
                _in.close();
                zipOut.closeEntry();
            }
        }
    }

    /**
     * @Author AlphaJunS
     * @Date 11:34 2020/3/8
     * @Description 解压缩ZIP文件，将ZIP文件里的内容解压到targetDIR目录下
      * @param zipPath 待解压缩的ZIP文件名
     * @param descDir 目标目录
     * @return java.util.List<java.io.File>
     */
    public static List<File> unzipFile(String zipPath, String descDir) {
        return unzipFile(new File(zipPath) , descDir) ;
    }

    /**
     * @Author AlphaJunS
     * @Date 11:36 2020/3/8
     * @Description 对.zip文件进行解压缩
     * @param zipFile 解压缩文件
     * @param descDir 压缩的目标地址，如：D:\测试 或 /mnt/d/测试
     * @return java.util.List<java.io.File>
     */
    @SuppressWarnings("rawtypes")
    public static List<File> unzipFile(File zipFile, String descDir) {
        List<File> _list = new ArrayList<File>() ;
        try {
            ZipFile _zipFile = new ZipFile(zipFile , "GBK") ;
            for( Enumeration entries = _zipFile.getEntries() ; entries.hasMoreElements() ; ){
                ZipEntry entry = (ZipEntry)entries.nextElement() ;
                File _file = new File(descDir + File.separator + entry.getName()) ;
                if( entry.isDirectory() ){
                    _file.mkdirs() ;
                }else{
                    File _parent = _file.getParentFile() ;
                    if( !_parent.exists() ){
                        _parent.mkdirs() ;
                    }
                    InputStream _in = _zipFile.getInputStream(entry);
                    OutputStream _out = new FileOutputStream(_file) ;
                    int len = 0 ;
                    byte[] _byte = new byte[1024];
                    while( (len = _in.read(_byte)) > 0){
                        _out.write(_byte, 0, len);
                    }
                    _in.close();
                    _out.flush();
                    _out.close();
                    _list.add(_file) ;
                }
            }
        } catch (IOException e) {
        }
        return _list ;
    }

    /**
     * @Author AlphaJunS
     * @Date 11:36 2020/3/8
     * @Description 对临时生成的文件夹和文件夹下的文件进行删除
     * @param delpath
     * @return void
     */
    public static void deletefile(String delpath) {
        try {
            File file = new File(delpath);
            if (!file.isDirectory()) {
                file.delete();
            } else if (file.isDirectory()) {
                String[] fileList = file.list();
                for (int i = 0; i < fileList.length; i++) {
                    File delfile = new File(delpath + File.separator + fileList[i]);
                    if (!delfile.isDirectory()) {
                        delfile.delete();
                    } else if (delfile.isDirectory()) {
                        deletefile(delpath + File.separator + fileList[i]);
                    }
                }
                file.delete();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}