`

利用java.util.zip压缩文件

阅读更多

package test;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Enumeration;

import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;
/*
 import org.apache.tools.zip.ZipOutputStream;  
 import org.apache.tools.zip.ZipEntry;  
 import org.apache.tools.zip.ZipFile; */

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * 用ZIP压缩和解压缩文件或目录
 */
public class Zip {

 private final static Log log = LogFactory.getLog(Zip.class);

 /**
  * 压缩文件或者目录
  *
  * @param baseDirName
  *            压缩的根目录
  * @param fileName
  *            根目录下待压缩的文件或文件夹名, 星号*表示压缩根目录下的全部文件。
  * @param targetFileName
  *            目标ZIP文件
  */
 public static void zipFile(String baseDirName, String targetFileName) {
  // 检测根目录是否存在
  if (baseDirName == null) {
   log.info("压缩失败,根目录不存在:" + baseDirName);
   return;
  }
  File file = new File(baseDirName);
  if (!file.exists()) {
   log.info("压缩失败,文件不存在:" + baseDirName);
   return;
  }

  String baseDirPath = file.getAbsolutePath();
  // 目标文件
  File targetFile = new File(targetFileName);

  try {
   // 创建一个zip输出流来压缩数据并写入到zip文件
   ZipOutputStream out = new ZipOutputStream(new FileOutputStream(
     targetFile));
   if (file.isFile()) {
    Zip.fileToZip(baseDirPath, file, out);
   } else {
    Zip.dirToZip(baseDirPath, file, out);
   }
   out.close();
   log.info("压缩文件成功,目标文件名:" + targetFileName);
  } catch (IOException e) {
   log.info("压缩失败:" + e);
   e.printStackTrace();
  }
 }

 /**
  * 解压缩ZIP文件,将ZIP文件里的内容解压到targetDIR目录下
  *
  * @param zipName
  *            待解压缩的ZIP文件名
  * @param targetBaseDirName
  *            目标目录
  */
 public static void upzipFile(String zipFileName, String targetBaseDirName) {
  if (!targetBaseDirName.endsWith(File.separator)) {
   targetBaseDirName += File.separator;
  }
  try {
   // 根据ZIP文件创建ZipFile对象
   ZipFile zipFile = new ZipFile(zipFileName);
   ZipEntry entry = null;
   String entryName = null;
   String targetFileName = null;
   byte[] buffer = new byte[4096];
   int bytes_read;
   // 获取ZIP文件里所有的entry
   Enumeration entrys = zipFile.entries();
   // 遍历所有entry
   while (entrys.hasMoreElements()) {
    entry = (ZipEntry) entrys.nextElement();
    // 获得entry的名字
    entryName = entry.getName();
    System.out.println(entryName);
    targetFileName = targetBaseDirName + entryName;

    if (entry.isDirectory()) {
     // 如果entry是一个目录,则创建目录
     new File(targetFileName).mkdirs();
     continue;
    } else {
     // 如果entry是一个文件,则创建父目录
     new File(targetFileName).getParentFile().mkdirs();
    }

    File targetFile = new File(targetFileName);
    FileOutputStream os = new FileOutputStream(targetFile);
    InputStream is = zipFile.getInputStream(entry);
    while ((bytes_read = is.read(buffer)) != -1) {
     os.write(buffer, 0, bytes_read);
    }
    // 关闭流
    os.close();
    is.close();
   }
   log.info("解压缩文件成功!");
  } catch (IOException err) {
   System.err.println("解压缩文件失败: " + err);
  }
 }

 /**
  * 将目录压缩到ZIP输出流。
  */
 private static void dirToZip(String baseDirPath, File dir,
   ZipOutputStream out) {
  if (dir.isDirectory()) {
   // 列出dir目录下所有文件
   File[] files = dir.listFiles();
   // 如果是空文件夹
   if (files.length == 0) {
    ZipEntry entry = new ZipEntry(getEntryName(baseDirPath, dir));
    // 存储目录信息
    try {
     out.putNextEntry(entry);
     out.closeEntry();
    } catch (IOException e) {
     e.printStackTrace();
    }
    return;
   }
   for (int i = 0; i < files.length; i++) {
    if (files[i].isFile()) {
     // 如果是文件,调用fileToZip方法
     Zip.fileToZip(baseDirPath, files[i], out);
    } else {
     // 如果是目录,递归调用
     Zip.dirToZip(baseDirPath, files[i], out);
    }
   }
  }
 }

 /**
  * 将文件压缩到ZIP输出流
  */
 private static void fileToZip(String baseDirPath, File file,
   ZipOutputStream out) {
  FileInputStream in = null;
  ZipEntry entry = null;
  // 创建复制缓冲区
  byte[] buffer = new byte[4096];
  int bytes_read;
  if (file.isFile()) {
   try {
    // 创建一个文件输入流
    in = new FileInputStream(file);
    // 做一个ZipEntry
    entry = new ZipEntry(getEntryName(baseDirPath, file));
    // 存储项信息到压缩文件
    out.putNextEntry(entry);
    // 复制字节到压缩文件
    while ((bytes_read = in.read(buffer)) != -1) {
     out.write(buffer, 0, bytes_read);
    }
    out.closeEntry();
    in.close();
    log.info("添加文件" + file.getAbsolutePath() + "被到ZIP文件中!");
   } catch (IOException e) {
    e.printStackTrace();
   }
  }
 }

 /**
  * 获取待压缩文件在ZIP文件中entry的名字。即相对于跟目录的相对路径名
  *
  * @param baseDirPath
  * @param file
  * @return
  */
 private static String getEntryName(String baseDirPath, File file) {
  if (!baseDirPath.endsWith(File.separator)) {
   baseDirPath += File.separator;
  }
  String filePath = file.getAbsolutePath();
  // 对于目录,必须在entry名字后面加上"/",表示它将以目录项存储。
  if (file.isDirectory()) {
   filePath += "/";
  }
  int index = filePath.indexOf(baseDirPath);
  return filePath.substring(index + baseDirPath.length());
 }

 public static void main(String[] args) {
  // 压缩C盘下的temp目录,压缩后的文件是C:/temp.zip
  /*
   * String baseDirName = "C:/"; String fileName = "temp/"; String
   * zipFileName = "C:/temp.zip";
   */
  String baseDirName = "E:/plat";
  String zipFileName = "E:/aa.zip";
  Zip.upzipFile("d:/杂志.zip", "d:/");
  Zip.zipFile("e:/123", "e:/123.zip");
  // 将刚创建的ZIP文件解压缩到D盘的temp目录下
  // System.out.println();
  /*
   * String zipFileName = "E:/plat.zip"; String fileName = "E:/plat1";
   * upzipFile(zipFileName,fileName);
   */
  // CompressUtil.zipFile(baseDirName, fileName, zipFileName);
 }
}

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics