Java8 Zip 压缩与解压缩

た入场券 2023-07-06 06:28 23阅读 0赞

网上找过几个例子都有点小问题，还是谷歌找出来的靠谱。主要是增加了指定文件的功能，通过 Java8 的 Lambda 判断是否加入 ZIP 压缩，比较方便。函数表达式的签名是 Function<File, Boolean>，参数是待加入的 File 对象，返回值 true 表示允许，反之不行。

完整代码在：[https://gitee.com/sp42\_admin/ajaxjs/blob/master/ajaxjs-base/src/main/java/com/ajaxjs/util/io/FileHelper.java][https_gitee.com_sp42_admin_ajaxjs_blob_master_ajaxjs-base_src_main_java_com_ajaxjs_util_io_FileHelper.java]

/** * Copyright sp42 frank@ajaxjs.com * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */
    package com.ajaxjs.util.io;
    
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.util.function.Function;
    import java.util.zip.ZipEntry;
    import java.util.zip.ZipInputStream;
    import java.util.zip.ZipOutputStream;
    
    import com.ajaxjs.util.logger.LogHelper;
    
    /** * ZIP 压缩/解压缩 * * @author sp42 * */
    public class ZipHelper { 
    	private static final LogHelper LOGGER = LogHelper.getLog(ZipHelper.class);
    
    	/** * 解压文件 * * @param save 解压文件的路径，必须为目录 * @param zipFile 输入的解压文件路径，例如C:/temp/foo.zip或 c:\\temp\\bar.zip */
    	public static void unzip(String save, String zipFile) { 
    		if (!new File(save).isDirectory())
    			throw new IllegalArgumentException("保存的路径必须为目录路径");
    
    		long start = System.currentTimeMillis();
    		File folder = new File(save);
    		if (!folder.exists())
    			folder.mkdirs();
    
    		try (ZipInputStream zis = new ZipInputStream(new FileInputStream(zipFile));) { 
    			ZipEntry ze;
    			while ((ze = zis.getNextEntry()) != null) { 
    				File newFile = new File(save + File.separator + ze.getName());
    				System.out.println("file unzip : " + newFile.getAbsoluteFile());
    
    				// 大部分网络上的源码，这里没有判断子目录
    				if (ze.isDirectory()) { 
    					newFile.mkdirs();
    				} else { 
    // new File(newFile.getParent()).mkdirs();
    					FileHelper.initFolder(newFile);
    					FileOutputStream fos = new FileOutputStream(newFile);
    					IoHelper.write(zis, fos, false);
    					fos.close();
    				}
    
    // ze = zis.getNextEntry();
    			}
    			zis.closeEntry();
    		} catch (IOException e) { 
    			LOGGER.warning(e);
    		}
    
    		LOGGER.info("解压缩完成，耗时：{0}ms，保存在{1}", System.currentTimeMillis() - start, save);
    	}
    
    	/** * 压缩文件 * * @param toZip 要压缩的目录或文件 * @param saveZip 压缩后保存的 zip 文件名 */
    	public static void zip(String toZip, String saveZip) { 
    		zip(toZip, saveZip, null);
    	}
    
    	/** * 压缩文件 * * @param toZip 要压缩的目录或文件 * @param saveZip 压缩后保存的 zip 文件名 * @param everyFile 输入 File，可在这 Lambda 里面判断是否加入 ZIP 压缩，返回 true 表示允许，反之不行 */
    	public static void zip(String toZip, String saveZip, Function<File, Boolean> everyFile) { 
    		long start = System.currentTimeMillis();
    		File fileToZip = new File(toZip);
    
    		FileHelper.initFolder(saveZip);
    
    		try (FileOutputStream fos = new FileOutputStream(saveZip); ZipOutputStream zipOut = new ZipOutputStream(fos);) { 
    			zip(fileToZip, fileToZip.getName(), zipOut, everyFile);
    		} catch (IOException e) { 
    			LOGGER.warning(e);
    		}
    
    		LOGGER.info("压缩完成，耗时：{0}ms，保存在{1}", System.currentTimeMillis() - start, saveZip);
    	}
    
    	/** * 内部的压缩方法 * * @param toZip 要压缩的目录或文件 * @param fileName ZIP 内的文件名 * @param zipOut ZIP 流 * @param everyFile 输入 File，可在这 Lambda 里面判断是否加入 ZIP 压缩，返回 true 表示允许，反之不行 */
    	private static void zip(File toZip, String fileName, ZipOutputStream zipOut, Function<File, Boolean> everyFile) { 
    		if (toZip.isHidden())
    			return;
    
    		if (everyFile != null && !everyFile.apply(toZip)) { 
    			return; // 跳过不要的
    		}
    
    		try { 
    			if (toZip.isDirectory()) { 
    				zipOut.putNextEntry(new ZipEntry(fileName.endsWith("/") ? fileName : fileName + "/"));
    				zipOut.closeEntry();
    
    				File[] children = toZip.listFiles();
    				for (File childFile : children) { 
    					zip(childFile, fileName + "/" + childFile.getName(), zipOut, everyFile);
    				}
    
    				return;
    			}
    
    			zipOut.putNextEntry(new ZipEntry(fileName));
    
    			try (FileInputStream in = new FileInputStream(toZip);) { 
    				IoHelper.write(in, zipOut, false);
    			}
    		} catch (IOException e) { 
    			LOGGER.warning(e);
    		}
    	}
    }

[https_gitee.com_sp42_admin_ajaxjs_blob_master_ajaxjs-base_src_main_java_com_ajaxjs_util_io_FileHelper.java]: https://gitee.com/sp42_admin/ajaxjs/blob/master/ajaxjs-base/src/main/java/com/ajaxjs/util/io/FileHelper.java