Java中往zip壓縮包追加文件

有個需求,從某個接口下載的一個zip壓縮包,往裏面添加一個說明文件。搜索了一下,沒有找到往zip直接添加文件的方法,最終解決方法是先解壓、再壓縮。

具體過程如下:

1、一個zip文件的壓縮和解壓工具類

壓縮和解壓工具類來自https://www.iteye.com/blog/songfeng-123-2243016,但是原文代碼因爲用的是Java自帶的java.util.zip,有中文亂碼的bug,所以需要修改部分代碼,並且修改爲引用org.apache.tools.zip.*,pom.xml加入依賴包,如下:

        <dependency>
            <groupId>org.apache.ant</groupId>
            <artifactId>ant</artifactId>
            <version>1.10.7</version>
        </dependency>

工具類代碼:

package com.example.demo;

import java.io.*;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.zip.ZipException;

import org.apache.tools.zip.*;

public class ZipUtil {

    private static int BUFFERSIZE = 1024;

    /**
     * 壓縮
     *
     * @param paths
     * @param fileName
     */
    public static void zip(List<String> paths, String fileName) {
        ZipOutputStream zos = null;
        try {
            zos = new ZipOutputStream(new FileOutputStream(fileName));
            for (String filePath : paths) {
                // 遞歸壓縮文件
                File file = new File(filePath);
                String relativePath = file.getName();
                if (file.isDirectory()) {
                    relativePath += File.separator;
                }
                zipFile(file, relativePath, zos);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (zos != null) {
                    zos.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    public static void zipFile(File file, String relativePath, ZipOutputStream zos) {
        InputStream is = null;
        try {
            if (!file.isDirectory()) {
                ZipEntry zp = new ZipEntry(relativePath);
                zos.putNextEntry(zp);
                is = new FileInputStream(file);
                byte[] buffer = new byte[BUFFERSIZE];
                int length = 0;
                while ((length = is.read(buffer)) >= 0) {
                    zos.write(buffer, 0, length);
                }
                zos.setEncoding("gbk");//解決文件名中文亂碼
                zos.flush();
                zos.closeEntry();
            } else {
                String tempPath = null;
                for (File f : file.listFiles()) {
                    tempPath = relativePath + f.getName();
                    if (f.isDirectory()) {
                        tempPath += File.separator;
                    }
                    zipFile(f, tempPath, zos);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (is != null) {
                    is.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 解壓縮
     *
     * @param fileName
     * @param path
     */
    public static List<String> unzip(String fileName, String path) {
        FileOutputStream fos = null;
        InputStream is = null;
        List<String> filePaths = new ArrayList<String>();
        try {
            ZipFile zf = new ZipFile(new File(fileName));
            Enumeration en = zf.getEntries();
            while (en.hasMoreElements()) {
                ZipEntry zn = (ZipEntry) en.nextElement();
                if (!zn.isDirectory()) {
                    is = zf.getInputStream(zn);
                    File f = new File(path + zn.getName());
                    File file = f.getParentFile();
                    file.mkdirs();
                    fos = new FileOutputStream(path + zn.getName());
                    int len = 0;
                    byte bufer[] = new byte[BUFFERSIZE];
                    while (-1 != (len = is.read(bufer))) {
                        fos.write(bufer, 0, len);
                    }
                    fos.close();
                    filePaths.add(path + zn.getName());
                }
            }
        } catch (ZipException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != is) {
                    is.close();
                }
                if (null != fos) {
                    fos.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return filePaths;
    }
 }

2、測試

有如下目錄結構:
D:\測試\文檔.zip
D:\測試\說明.pdf
把“說明.pdf”添加到“文檔.zip”裏面,生成一個新壓縮包“文檔(新).zip”。

package com.example.demo;

import java.io.File;
import java.util.List;

public class ZipUtilTest {
    public static void main(String[] args) {
        //解壓
        List<String> files = ZipUtil.unzip("D:/測試/文檔.zip", "D:/測試/");
        //集合添加文件
        files.add("D:/測試/說明.pdf");
        //壓縮
        ZipUtil.zip(files,"D:/測試/文檔(新).zip");
        //保留說明.pdf
        files.remove(files.size()-1);
        //刪除上面解壓出來的文件
        for(String f : files){
            File file = new File(f);
            if(file.exists()){
                file.delete();
            }
        }
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章