Terry.Li-彬

虚其心，可解天下之问；专其心，可治天下之学；静其心，可悟天下之理；恒其心，可成天下之业。
  BlogJava :: 首页 :: 新随笔 :: 联系 :: 聚合  :: 管理 ::
  143 随笔 :: 344 文章 :: 130 评论 :: 0 Trackbacks
压缩/解压缩zip包处理

 

package com.newegg.lab.util;

import java.io.BufferedOutputStream;
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.Enumeration;

import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

//使用apache的api不存在压缩文件名乱码问题（该jar包在ant中）
//import org.apache.tools.zip.ZipEntry;
//import org.apache.tools.zip.ZipFile;
//import org.apache.tools.zip.ZipOutputStream;

/** *//**
 * 压缩/解压缩zip包处理类
 *
 * @author terry.b.li
 * @date 2--2
 */
public class ZipUtil {

    /** *//**
     * 压缩
     *
     * @param zipFileName
     *            压缩产生的zip包文件名--带路径,如果为null或空则默认按文件名生产压缩文件名
     * @param relativePath
     *            相对路径，默认为空
     * @param directory
     *            文件或目录的绝对路径
     * @throws FileNotFoundException
     * @throws IOException
     * @author terry.b.li
     */
    public static void zip(String zipFileName, String relativePath,
            String directory) throws FileNotFoundException, IOException {
        String fileName = zipFileName;
        if (fileName == null || fileName.trim().equals("")) {
            File temp = new File(directory);
            if (temp.isDirectory()) {
                fileName = directory + ".zip";
            } else {
                if (directory.indexOf(".") > 0) {
                    fileName = directory.substring(0, directory
                            .lastIndexOf("."))
                            + "zip";
                } else {
                    fileName = directory + ".zip";
                }
            }
        }
        ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(fileName));
        try {
            zip(zos, relativePath, directory);
        } catch (IOException ex) {
            throw ex;
        } finally {
            if (null != zos) {
                zos.close();
            }
        }
    }

    /** *//**
     * 压缩
     *
     * @param zos
     *            压缩输出流
     * @param relativePath
     *            相对路径
     * @param absolutPath
     *            文件或文件夹绝对路径
     * @throws IOException
     * @author terry.b.li
     */
    private static void zip(ZipOutputStream zos, String relativePath,
            String absolutPath) throws IOException {
        File file = new File(absolutPath);
        if (file.isDirectory()) {
            File[] files = file.listFiles();
            for (int i = 0; i < files.length; i++) {
                File tempFile = files[i];
                if (tempFile.isDirectory()) {
                    String newRelativePath = relativePath + tempFile.getName()
                            + File.separator;
                    createZipNode(zos, newRelativePath);
                    zip(zos, newRelativePath, tempFile.getPath());
                } else {
                    zipFile(zos, tempFile, relativePath);
                }
            }
        } else {
            zipFile(zos, file, relativePath);
        }
    }

    /** *//**
     * 压缩文件
     *
     * @param zos
     *            压缩输出流
     * @param file
     *            文件对象
     * @param relativePath
     *            相对路径
     * @throws IOException
     * @author terry.b.li
     */
    private static void zipFile(ZipOutputStream zos, File file,
            String relativePath) throws IOException {
        ZipEntry entry = new ZipEntry(relativePath + new String(file.getName().getBytes("ISO-8859-1")));
        zos.putNextEntry(entry);
        zos.setComment("this is zip file luckystar");
        InputStream is = null;
        try {
            is = new FileInputStream(file);
            int BUFFERSIZE = 2 <<10;
            int length =0;
            byte[] buffer = new byte[BUFFERSIZE];
            while ((length = is.read(buffer, 0, BUFFERSIZE)) >= 0) {
                zos.write(buffer, 0, length);
            }
            zos.flush();
            zos.closeEntry();
        } catch (IOException ex) {
            throw ex;
        } finally {
            if (null != is) {
                is.close();
            }
        }
    }

    /** *//**
     * 创建目录
     *
     * @param zos
     *            zip输出流
     * @param relativePath
     *            相对路径
     * @throws IOException
     * @author terry.b.li
     */
    private static void createZipNode(ZipOutputStream zos, String relativePath)
            throws IOException {
        ZipEntry zipEntry = new ZipEntry(relativePath);
        zos.putNextEntry(zipEntry);
        zos.closeEntry();
    }

    /** *//**
     * 解压缩zip包
     *
     * @param zipFilePath
     *            zip文件路径
     * @param targetPath
     *            解压缩到的位置，如果为null或空字符串则默认解压缩到跟zip包同目录跟zip包同名的文件夹下
     * @throws IOException
     * @author terry.b.li
     */
    public static void unzip(String zipFilePath, String targetPath)
            throws IOException {
        OutputStream os = null;
        InputStream is = null;
        ZipFile zipFile = null;
        try {
            zipFile = new ZipFile(zipFilePath);
            String directoryPath = "";
            if (null == targetPath || "".equals(targetPath)) {
                directoryPath = zipFilePath.substring(0, zipFilePath
                        .lastIndexOf("."));
            } else {
                directoryPath = targetPath;
            }
//            Enumeration entryEnum = zipFile.getEntries();
            Enumeration entryEnum = zipFile.entries();
            if (null != entryEnum) {
                ZipEntry zipEntry = null;
                while (entryEnum.hasMoreElements()) {
                    zipEntry = (ZipEntry) entryEnum.nextElement();
                    if (zipEntry.isDirectory()) {
                        directoryPath = directoryPath + File.separator
                                + zipEntry.getName();
                        System.out.println(directoryPath);
                        continue;
                    }
                    if (zipEntry.getSize() > 0) {
                        // 文件
                        File targetFile = FileUtil.buildFile(directoryPath
                                + File.separator + zipEntry.getName(), false);
                        os = new BufferedOutputStream(new FileOutputStream(
                                targetFile));
                        is = zipFile.getInputStream(zipEntry);
                        byte[] buffer = new byte[406];
                        int readLen = 0;
                        while ((readLen = is.read(buffer, 0, 4096)) >= 0) {
                            os.write(buffer, 0, readLen);
                        }

                        os.flush();
                        os.close();
                    } else {
                        // 空目录
                        FileUtil.buildFile(directoryPath + File.separator
                                + zipEntry.getName(), true);
                    }
                }
            }
        } catch (IOException ex) {
            throw ex;
        } finally {
            if(null != zipFile){
                zipFile = null;
            }
            if (null != is) {
                is.close();
            }
            if (null != os) {
                os.close();
            }
        }
    }

    public static void main(String[] args) {
        try {
            ZipUtil.zip("c:\\test.zip", "", "C:\\Documents and Settings\\Administrator\\桌面\\新建文件夹");
            System.out.println("OK");
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
}

posted on 2008-12-07 11:03 礼物 阅读(733) 评论(0)  编辑  收藏 所属分类: java

只有注册用户登录后才能发表评论。
网站导航: