package com.wdy.tools.utils.pressuitl;
 
import java.io.BufferedInputStream;
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.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
 
import org.apache.commons.logging.Log;
 
import com.wdy.tools.utils.LogUtil;
 
/**
 * 壓縮/解壓縮工具類（zip格式）
 * 
 * @author wdy
 * @date 2016-08-23
 */
public class PressUtil {
 
 private static final Log log = LogUtil.getLog(PressUtil.class);
 
 
 
 public static void main(String[] args) {
// PressUtil.ZipMultiFile("d:\\nwp_trans\\nwp_h\\", "d:\\nwp_trans\\nwp_h\\wdy.zip");
 String sourceFilePath = "d:\\nwp_trans\\nwp_h\\"; 
   String zipFilePath = "d:\\nwp_trans\\nwp_h\\"; 
   String fileName = "wdy"; 
   boolean flag = PressUtil.fileToZip(sourceFilePath, zipFilePath, fileName);
   if(flag){ 
   log.info("文件打包成功!"); 
   }else{ 
   log.info("文件打包失敗!"); 
   } 
 }
 
 /** 
  * 將存放在sourceFilePath目錄下的源文件，打包成fileName名稱的zip文件，并存放到zipFilePath路徑下 
  * @param sourceFilePath :待壓縮的文件路徑 
  * @param zipFilePath :壓縮后存放路徑 
  * @param fileName :壓縮后文件的名稱(不包括擴展名) 
  * @return 
  */ 
 @SuppressWarnings("resource")
 public static boolean fileToZip(String sourceFilePath,String zipFilePath,String fileName){ 
  boolean flag = false; 
  File sourceFile = new File(sourceFilePath); 
  FileInputStream fis = null; 
  BufferedInputStream bis = null; 
  FileOutputStream fos = null; 
  ZipOutputStream zos = null; 
   
  if(sourceFile.exists() == false){ 
   log.info("待壓縮的文件目錄："+sourceFilePath+"不存在.");
  }else{ 
   try { 
    File zipFile = new File(zipFilePath + "/" + fileName +".zip"); 
    if(zipFile.exists()){ 
     log.info(zipFilePath + "目錄下存在名字為:" + fileName +".zip" +"打包文件.");
    }else{ 
     File[] sourceFiles = sourceFile.listFiles(); 
     if(null == sourceFiles || sourceFiles.length<1){ 
      log.info("待壓縮的文件目錄：" + sourceFilePath + "里面不存在文件，無需壓縮.");
     }else{ 
      fos = new FileOutputStream(zipFile); 
      zos = new ZipOutputStream(new BufferedOutputStream(fos)); 
      byte[] bufs = new byte[1024*10]; 
      for(int i=0;i<sourceFiles.length;i++){ 
       //創建ZIP實體，并添加進壓縮包 
       ZipEntry zipEntry = new ZipEntry(sourceFiles[i].getName()); 
       zos.putNextEntry(zipEntry); 
       //讀取待壓縮的文件并寫進壓縮包里 
       fis = new FileInputStream(sourceFiles[i]); 
       bis = new BufferedInputStream(fis, 1024*10); 
       int read = 0; 
       while((read=bis.read(bufs, 0, 1024*10)) != -1){ 
        zos.write(bufs,0,read); 
       } 
      } 
      flag = true; 
     } 
    } 
   } catch (FileNotFoundException e) { 
    e.printStackTrace(); 
    throw new RuntimeException(e); 
   } catch (IOException e) { 
    e.printStackTrace(); 
    throw new RuntimeException(e); 
   } finally{ 
    //關閉流 
    try { 
     if(null != bis) bis.close(); 
     if(null != zos) zos.close(); 
    } catch (IOException e) { 
     e.printStackTrace(); 
     throw new RuntimeException(e); 
    } 
   } 
  } 
  return flag; 
 } 
 
 /**
 * 壓縮單個文件
 * @param filePath 要被壓縮的文件的全路徑，包含文件名d:/hello.txt
 * @param zipPath 壓縮后的全路徑，包含文件名d:/hello.zip
 */
 public static void zipOneFile(String filePath, String zipPath) {
 try {
 File file = new File(filePath);
 File zipFile = new File(zipPath);
 InputStream input = new FileInputStream(file);
 ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(zipFile));
 zipOut.putNextEntry(new ZipEntry(file.getName()));
 int temp = 0;
 while ((temp = input.read()) != -1) {
 zipOut.write(temp);
 }
 input.close();
 zipOut.close();
 } catch (Exception e) {
 e.printStackTrace();
 }
 }
}