壓縮文件工具類

public class ZipUtils {

	private static final int BUFFER_SIZE = 2 * 1024;

	private static Logger logger = LoggerFactory.getLogger(ZipUtils.class);

	/**
	 * 壓縮成ZIP
	 *
	 * @param srcDir
	 *            壓縮文件夾路徑
	 * @param out
	 *            壓縮文件輸出流
	 * @param KeepDirStructure
	 *            是否保留原來的目錄結構,true:保留目錄結構;
	 *            false:所有文件跑到壓縮包根目錄下(注意:不保留目錄結構可能會出現同名文件,會壓縮失敗)
	 * @throws RuntimeException
	 *             壓縮失敗會拋出運行時異常
	 */
	public static void toZip(String srcDir, OutputStream out,
			boolean KeepDirStructure) throws RuntimeException {
		long start = System.currentTimeMillis();
		ZipOutputStream zos = null;
		try {
			zos = new ZipOutputStream(out);
			File sourceFile = new File(srcDir);
			compress(sourceFile, zos, sourceFile.getName(), KeepDirStructure);
			long end = System.currentTimeMillis();
			System.out.println("壓縮完成,耗時:" + (end - start) + " ms");
		} catch (Exception e) {
			throw new RuntimeException("zip error from ZipUtils", e);
		} finally {
			if (zos != null) {
				try {
					zos.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	/**
	 * 壓縮成ZIP
	 *
	 * @param srcFiles
	 *            需要壓縮的文件列表
	 * @param out
	 *            壓縮文件輸出流
	 * @throws RuntimeException
	 *             壓縮失敗會拋出運行時異常
	 */
	public static void toZip(List<File> srcFiles, OutputStream out)
			throws IOException {

		long start = System.currentTimeMillis();

		ZipOutputStream zos = null;

		try {

			zos = new ZipOutputStream(out);

			for (File srcFile : srcFiles) {

				byte[] buf = new byte[BUFFER_SIZE];

				zos.putNextEntry(new ZipEntry(srcFile.getName()));

				int len;

				FileInputStream in = new FileInputStream(srcFile);

				while ((len = in.read(buf)) != -1) {

					zos.write(buf, 0, len);
				}

				zos.closeEntry();

				in.close();
			}

			long end = System.currentTimeMillis();

			logger.debug("Zip compress took:[{}] ms", (end - start));
		} finally {
			if (zos != null) {
				try {
					zos.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	/**
	 * 遞歸壓縮方法
	 *
	 * @param sourceFile
	 *            源文件
	 * @param zos
	 *            zip輸出流
	 * @param name
	 *            壓縮後的名稱
	 * @param KeepDirStructure
	 *            是否保留原來的目錄結構,true:保留目錄結構;
	 *            false:所有文件跑到壓縮包根目錄下(注意:不保留目錄結構可能會出現同名文件,會壓縮失敗)
	 * @throws Exception
	 */
	private static void compress(File sourceFile, ZipOutputStream zos,
			String name, boolean KeepDirStructure) throws Exception {
		byte[] buf = new byte[BUFFER_SIZE];
		if (sourceFile.isFile()) {
			// 向zip輸出流中添加一個zip實體,構造器中name爲zip實體的文件的名字
			zos.putNextEntry(new ZipEntry(name));
			// copy文件到zip輸出流中
			int len;
			FileInputStream in = new FileInputStream(sourceFile);
			while ((len = in.read(buf)) != -1) {
				zos.write(buf, 0, len);
			}
			// Complete the entry
			zos.closeEntry();
			in.close();
		} else {
			File[] listFiles = sourceFile.listFiles();
			if (listFiles == null || listFiles.length == 0) {
				// 需要保留原來的文件結構時,需要對空文件夾進行處理
				if (KeepDirStructure) {
					// 空文件夾的處理
					zos.putNextEntry(new ZipEntry(name + "/"));
					// 沒有文件,不需要文件的copy
					zos.closeEntry();
				}
			} else {
				for (File file : listFiles) {
					// 判斷是否需要保留原來的文件結構
					if (KeepDirStructure) {
						// 注意:file.getName()前面需要帶上父文件夾的名字加一斜槓,
						// 不然最後壓縮包中就不能保留原來的文件結構,即:所有文件都跑到壓縮包根目錄下了
						compress(file, zos, name + "/" + file.getName(),
								KeepDirStructure);
					} else {
						compress(file, zos, file.getName(), KeepDirStructure);
					}
				}

			}

		}
	}

	/**
	 * 解壓zip格式的壓縮文件到指定位置
	 *
	 * @param zipFileName
	 *            壓縮文件
	 * @param extPlace
	 *            解壓目錄
	 * @throws Exception
	 */
	public static boolean unZipFiles(String zipFileName, String extPlace) {
		System.setProperty("sun.zip.encoding",
				System.getProperty("sun.jnu.encoding"));
		try {
			(new File(extPlace)).mkdirs();
			File f = new File(zipFileName);
			ZipFile zipFile = new ZipFile(zipFileName, "GBK"); // 處理中文文件名亂碼的問題
			if ((!f.exists()) && (f.length() <= 0)) {
				throw new Exception("要解壓的文件不存在!");
			}
			String strPath, gbkPath, strtemp;
			File tempFile = new File(extPlace);
			strPath = tempFile.getAbsolutePath();
			Enumeration<?> e = zipFile.getEntries();
			while (e.hasMoreElements()) {
				org.apache.tools.zip.ZipEntry zipEnt = (org.apache.tools.zip.ZipEntry) e
						.nextElement();
				gbkPath = zipEnt.getName();
				if (zipEnt.isDirectory()) {
					strtemp = strPath + File.separator + gbkPath;
					File dir = new File(strtemp);
					dir.mkdirs();
					continue;
				} else { // 讀寫文件
					InputStream is = zipFile.getInputStream(zipEnt);
					BufferedInputStream bis = new BufferedInputStream(is);
					gbkPath = zipEnt.getName();
					strtemp = strPath + File.separator + gbkPath;// 建目錄
					String strsubdir = gbkPath;
					for (int i = 0; i < strsubdir.length(); i++) {
						if (strsubdir.substring(i, i + 1)
								.equalsIgnoreCase("/")) {
							String temp = strPath + File.separator
									+ strsubdir.substring(0, i);
							File subdir = new File(temp);
							if (!subdir.exists())
								subdir.mkdir();
						}
					}
					FileOutputStream fos = new FileOutputStream(strtemp);
					BufferedOutputStream bos = new BufferedOutputStream(fos);
					int c;
					while ((c = bis.read()) != -1) {
						bos.write((byte) c);
					}
					bos.close();
					fos.close();
				}
			}
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

}

 

發佈了91 篇原創文章 · 獲贊 92 · 訪問量 13萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章