使用JDK的zip编写打包工具类

 

JDK自带的zip AIP在java.util.zip包下面，主要有以下几个类：

java.util.zip.ZipEntry
java.util.zip.ZipInputStream
java.util.zip.ZipOutputStream

 

本文编写的zip工具类有以下功能：打包（单个文件、目录）、解包、查看zip包文件

 

工具类代码如下

package org.net5ijy.util.zip;

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.util.ArrayList;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;

/**
 * ZIP工具类
 * 
 */
public class ZipUtil {

    /**
     * 解包时的读写大小：10MB
     */
    private static final int SIZE = 1024 * 1024 * 10;

    /**
     * 打包
     * 
     * @param src
     *            - 需要打包的文件或目录，不可以是根目录，为根目录时抛出IOException
     * @param zipFile
     *            - 打包后的.zip文件
     * @throws IOException
     */
    public static void zip(File src, File zipFile) throws IOException {

        // 不可以压缩根目录
        if (src.getParent() == null) {
            throw new IOException("不可以压缩根目录");
        }
        // 获取基目录
        String base = src.getParentFile().getAbsolutePath();

        ZipOutputStream out = null;
        BufferedOutputStream bos = null;

        try {

            out = new ZipOutputStream(new FileOutputStream(zipFile));

            // 创建缓冲输出流
            bos = new BufferedOutputStream(out);

            // 调用函数
            zip(out, bos, src, base);

        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            // 关闭流
            if (bos != null) {
                bos.close();
            }
            if (out != null) {
                out.close();
            }
        }
    }

    private static void zip(ZipOutputStream out, BufferedOutputStream bos,
            File sourceFile, String base) throws IOException {

        // 如果路径为目录（文件夹）
        if (sourceFile.isDirectory()) {
            out.putNextEntry(new ZipEntry(sourceFile.getAbsolutePath().replace(
                    base + File.separator, "")
                    + "/"));
            // 取出文件夹中的文件（或子文件夹）
            File[] flist = sourceFile.listFiles();
            if (flist.length > 0) {
                for (int i = 0; i < flist.length; i++) {
                    zip(out, bos, flist[i], base);
                }
            }
        } else {// 如果不是目录（文件夹），即为文件，则先写入目录进入点，之后将文件写入zip文件中
            out.putNextEntry(new ZipEntry(sourceFile.getAbsolutePath().replace(
                    base + File.separator, "")));
            BufferedInputStream bis = null;
            try {
                // 文件输入流
                bis = new BufferedInputStream(new FileInputStream(sourceFile));
                int b = bis.read();
                // 将源文件写入到zip文件中
                while (b > -1) {
                    bos.write(b);
                    b = bis.read();
                }
                bos.flush();
            } catch (Exception e) {
                throw e;
            } finally {
                if (bis != null) {
                    bis.close();
                }
            }
        }
    }

    /**
     * 解包
     * 
     * @param zipFile
     *            - 需要解包的.zip文件
     * @param targetFolder
     *            - 目标目录
     * @throws IOException
     */
    public static void unzip(File zipFile, File targetFolder)
            throws IOException {

        // 如果源文件是null 或者 不是一个文件
        if (zipFile == null || !zipFile.isFile()
                || !zipFile.getName().endsWith(".zip")) {
            throw new IOException("请选择正确的.zip文件");
        }

        // 如果目标目录为null 或者 不是一个目录
        if (targetFolder == null || !targetFolder.isDirectory()) {
            throw new IOException("请选择正确的解压目录");
        }

        // 定义zip文件输入流
        ZipInputStream zin = null;

        try {

            // 从zipFile创建对应的ZipInputStream输入流
            zin = new ZipInputStream(new FileInputStream(zipFile));

            // 获取下一个条目
            for (ZipEntry entry = zin.getNextEntry(); entry != null; entry = zin
                    .getNextEntry()) {

                // 获取文件或者目录名
                String name = entry.getName();
                // 目录
                if (name.endsWith("/")) {
                    File dir = new File(targetFolder, name);
                    dir.mkdirs();
                } else {// 文件
                    unzip(targetFolder, zin, name);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            if (zin != null) {
                zin.close();
            }
        }
    }

    private static void unzip(File targetFolder, ZipInputStream zin, String name)
            throws IOException {

        // 定义输出流
        FileOutputStream out = null;

        try {

            // 创建输出流
            out = new FileOutputStream(new File(targetFolder.getAbsolutePath()
                    + File.separator + name));

            byte[] buf = new byte[SIZE];
            int len = zin.read(buf);

            while (len > -1) {
                out.write(buf, 0, len);
                len = zin.read(buf);
            }
            out.flush();

        } catch (Exception e) {
            throw e;
        } finally {
            if (out != null) {
                out.close();
            }
        }
    }

    /**
     * 获取.zip文件中的条目集合
     * 
     * @param zipFile
     *            - 待查看的.zip文件
     * @return 条目的集合
     * @throws IOException
     */
    public static List<ZipEntry> zipEntryList(File zipFile) throws IOException {

        // 如果源文件是null 或者 不是一个文件
        if (zipFile == null || !zipFile.isFile()) {
            throw new IOException("请选择正确的.zip文件");
        }

        // 定义zip文件输入流
        ZipInputStream zin = null;

        List<ZipEntry> l = new ArrayList<ZipEntry>();

        try {
            // 从zipFile创建对应的ZipInputStream输入流
            zin = new ZipInputStream(new FileInputStream(zipFile));
            // 获取下一个条目
            ZipEntry entry = zin.getNextEntry();

            while (entry != null) {
                l.add(entry);
                entry = zin.getNextEntry();
            }
        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            if (zin != null) {
                zin.close();
            }
        }
        return l;
    }

    /**
     * 获取.zip文件中的条目名称集合
     * 
     * @param zipFile
     *            - 待查看的.zip文件
     * @return 条目名称集合
     * @throws IOException
     */
    public static List<String> zipEntryNameList(File zipFile)
            throws IOException {
        List<String> l = new ArrayList<String>();
        List<ZipEntry> entryList = zipEntryList(zipFile);
        for (ZipEntry zipEntry : entryList) {
            l.add(zipEntry.getName());
        }
        return l;
    }
}

 

测试代码

import static org.net5ijy.util.zip.ZipUtil.*;

public class ZipUtilTest {

    public static void main(String[] args) throws IOException {

        zip(new File(
                "D:\\workspace\\SpringCloud\\zip-demo\\src\\org\\net5ijy\\util\\zip\\ZipUtil.java"),
                new File("d:\\zip01.zip"));

        zip(new File("D:\\workspace\\SpringCloud\\spring-data-demo2\\"),
                new File("d:\\zip02.zip"));

        unzip(new File("d:\\zip02.zip"), new File("D:\\a\\"));

        List<String> nameList = zipEntryNameList(new File("d:\\zip02.zip"));
        for (String name : nameList) {
            System.out.println(name);
        }
    }
}