Java 实现文件排序与压缩

标题Java 实现文件排序与压缩

       记录一下，大学毕业后写的第一个项目。需要实现将文件按照"项目名/调查组编号-调查点编号/test.jpg"的形式来存放，并且打成压缩包。压缩包的方法网上有很多，找了一个摘抄了一下。这里增加了按照规则生成临时文件夹。

public class ZipUtils {
	private static final int  BUFFER_SIZE = 2 * 1024;
	private static final String SOURCE_PATH = ServletActionContext.getServletContext().getRealPath("/EDEOISMultiMedia/UploadPointMultiMedia");//测试
	/**
	 * 压缩成ZIP 方法1
	 * 
	 * @param srcDir
	 *            压缩文件夹路径 
	 * @param out
	 *            压缩文件输出流 
	 * @param KeepDirStructure
	 *            是否保留原来的目录结构,true:保留目录结构; 
	 *            false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败) 
	 * @throws RuntimeException
	 *             压缩失败会抛出运行时异常 
	 */
	public void toZip(String srcDir, OutputStream out, boolean KeepDirStructure)throws RuntimeException {
		ZipOutputStream zos = null;
		try {
			zos = new ZipOutputStream(out);
			File sourceFile = new File(srcDir);
			compress(sourceFile, zos, sourceFile.getName(), KeepDirStructure);
		} catch (Exception e) {
			throw new RuntimeException("zip error from ZipUtils", e);
		} finally {
			if (zos != null) {
				try {
					zos.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	/**
	 * 压缩成ZIP 方法2 
	 * 
	 * @param srcFiles
	 *            需要压缩的文件列表 
	 * @param out
	 *            压缩文件输出流 
	 * @throws RuntimeException
	 *             压缩失败会抛出运行时异常 
	 */
	public void toZip(List<File> srcFiles, OutputStream out) throws RuntimeException {
		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();
			}
		} catch (Exception e) {
			throw new RuntimeException("zip error from ZipUtils", e);
		} 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 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);
					}
				}
			}
		}
	}
	
	/**
	 * 根据文件列表，创建临时文件夹
	 * @param fileList 调查点列表
	 * @return String 文件夹地址
	 */
	public String createTempFile(List<UploadPointEntity> fileList){
		//创建临时文件夹
		String dirPath = SOURCE_PATH + "/temp_"+UUIDUtils.generateShortUuid();
		File soureDir = new File(dirPath.toString());
		if (!soureDir.exists()) {
			soureDir.mkdirs();
		}
		for(UploadPointEntity up : fileList){
			//按照规则在临时文件夹中，创建不同项目不同点的文件夹
			File dir = new File(dirPath+"/"+up.getProject_name()+"/"+up.getGroup_num()+"-"+up.getSurvey_spot_num());
			if (!dir.exists()) {
				dir.mkdirs();
			}
			//将文件复制到临时文件夹中
			//1.图片文件复制
			copyToTemp(up.getPic_path(),dir);
			//2.视频文件复制
			copyToTemp(up.getVideo_path(),dir);
			//3.音频文件复制
			copyToTemp(up.getAudio_path(),dir);
		}
		return dirPath;
	} 
	
	/**
	 * 将文件复制到目标文件夹
	 * @param filePaths	文件地址，以“,”分隔开
	 * @param targetDir 目标文件夹
	 */
	public void copyToTemp(String filePaths,File targetDir){
		if(filePaths == null || "".equals(filePaths))
			return;
		String[] files = filePaths.split(",");
		for(int i = 0; i < files.length; i++){
			File sourceFile = new File(SOURCE_PATH+"/"+files[i]);
			try {
				FileUtils.copyFileToDirectory(sourceFile, targetDir);
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
}