Java实现数据文件压缩和移动复制

Java实现数据文件压缩和移动复制

压缩：

package cn.com.sinosoft.tbf.common.util;

import java.io.*;
import java.util.Enumeration;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

import static org.springframework.util.StreamUtils.BUFFER_SIZE;

/**
 * 文件压缩工具
 *
 *
 */
public class ZipUtil {

	/**
	 * 压缩所有文件
	 * 
	 * @param files
	 * @param dest
	 * @throws Exception
	 */
	public static void zipAll(List<File> files, File dest) throws Exception {
		if(files == null || files.size() == 0 || dest == null) return;
		ZipOutputStream out = new ZipOutputStream(new FileOutputStream(dest));
		File zipParent = dest.getParentFile();
		if (!zipParent.exists()) {
			zipParent.mkdirs();
		}
		try {
			for (File file : files) {
				zip(out, file);
			}
		} finally {
			if (out != null) {
				out.close();
			}
		}
	}

	/**
	 * zip 压缩
	 * 
	 * @param source
	 *            源文件
	 * @param dest
	 *            目标文件
	 * @return
	 * @throws Exception
	 */
	public static void zip(File source, File dest) throws Exception {
		ZipOutputStream out = null;
		try {
			File zipParent = dest.getParentFile();
			if (!zipParent.exists()) {
				zipParent.mkdirs();
			}
			out = new ZipOutputStream(new FileOutputStream(dest));
			zip(out, source);
		} finally {
			if (out != null) {
				out.close();
				// 删除源文件
				// source.delete();
			}
		}
	}

	public static void zip(ZipOutputStream out, File f) throws Exception {
		if (f == null || !f.exists() || !f.isFile())
			return;
		out.putNextEntry(new ZipEntry(f.getName()));
		FileInputStream in = new FileInputStream(f);
		BufferedInputStream bi = new BufferedInputStream(in);
		int b;
		byte[] buffer = new byte[512];
		try {
			while ((b = bi.read(buffer)) != -1) {
				out.write(buffer, 0, b);
			}
		} finally {
			bi.close();
			in.close();
		}
	}


	/**
	 * zip解压
	 * @param srcFile        zip源文件
	 * @param destDirPath     解压后的目标文件夹
	 * @throws RuntimeException 解压失败会抛出运行时异常
	 */
	public static void unZip(File srcFile, String destDirPath) throws RuntimeException {
		long start = System.currentTimeMillis();
		// 判断源文件是否存在
		if (!srcFile.exists()) {
			throw new RuntimeException(srcFile.getPath() + "所指文件不存在");
		}
		// 开始解压
		ZipFile zipFile = null;
		try {
			zipFile = new ZipFile(srcFile);
			Enumeration<?> entries = zipFile.entries();
			while (entries.hasMoreElements()) {
				ZipEntry entry = (ZipEntry) entries.nextElement();
				System.out.println("解压文件：" + entry.getName());
				// 如果是文件夹，就创建个文件夹
				if (entry.isDirectory()) {
					String dirPath = destDirPath + "/" + entry.getName();
					File dir = new File(dirPath);
					dir.mkdirs();
				} else {
					// 如果是文件，就先创建一个文件，然后用io流把内容copy过去
					File file = new File(destDirPath + "/" + entry.getName());
					// 保证这个文件的父文件夹必须要存在
					if(!file.getParentFile().exists()){
						file.getParentFile().mkdirs();
					}
					file.createNewFile();
					// 将压缩文件内容写入到这个文件中
					InputStream is = zipFile.getInputStream(entry);
					FileOutputStream fos = new FileOutputStream(file);
					int len;
					byte[] buf = new byte[BUFFER_SIZE];
					while ((len = is.read(buf)) != -1) {
						fos.write(buf, 0, len);
					}
					// 关流顺序，先打开的后关闭
					fos.close();
					is.close();
				}
			}
			long end = System.currentTimeMillis();
			System.out.println("解压完成，耗时：" + (end - start) +" ms");
		} catch (Exception e) {
			throw new RuntimeException("unzip error from ZipUtils", e);
		} finally {
			if(zipFile != null){
				try {
					zipFile.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

}

复制和移动

package cn.com.sinosoft.tbf.common.util;

import org.apache.commons.io.FileUtils;

import java.io.*;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * 文件操作
 */
public class FileUtil {

    public static void main(String[] args) throws IOException {
        // copyFile("C:\\temp\\ehr_data_files\\src\\EHR_CHILD_DIS_SCREENDIAG\\418035665\\EHR_CHILD_DIS_SCREENDIAG-418035665-20190309.json.zip","C:\\temp\\ehr_data_files\\send\\haha.zip");
//        deleteFile("E:\\test\\upload\\");

        File fileFrom = new File("C:\\temp\\ehr_data_files\\src\\EHR_CHILD_DIS_SCREENDIAG\\418035665\\EHR_CHILD_DIS_SCREENDIAG-418035665-20190309.json.zip");
        File fileTo = new File("C:\\temp\\ehr_data_files\\send\\hehe.zip");
        copyFileUsingApacheCommonsIO(fileFrom,fileTo);
    }    /**
     * 移动 文件或者文件夹
     * @param oldPath
     * @param newPath
     * @throws IOException
     */
    public static void moveTo(String oldPath,String newPath) throws IOException {
        copyFile(oldPath,newPath);
        deleteFile(oldPath);
    }

    /**
     * 删除 文件或者文件夹
     * @param filePath
     */
    public static void deleteFile(String filePath){
        File file = new File(filePath);
        if (!file.exists()) {
            return;
        }
        if (file.isDirectory() ) {
            File[] list = file.listFiles();

            for (File f : list) {
                deleteFile(f.getAbsolutePath()) ;
            }
        }
        file.delete();
    }

    /**
     * 复制 文件或者文件夹
     * @param oldPath
     * @param newPath
     * @throws IOException
     */
    public static void  copyFile(String oldPath ,String newPath ) throws IOException {
        System.out.println("copy file from [" + oldPath + "] to [" + newPath +"]");

        File oldFile = new File(oldPath) ;
        if  (oldFile.exists())  {

            if(oldFile.isDirectory()){ // 如果是文件夹
                File newPathDir = new File(newPath);
                newPathDir.mkdirs();
                File[] lists = oldFile.listFiles() ;
                if(lists != null && lists.length > 0 ){
                    for (File file : lists) {
                        copyFile(file.getAbsolutePath(), newPath.endsWith(File.separator) ? newPath + file.getName() : newPath + File.separator + file.getName()) ;
                    }
                }
            }else {
                InputStream  inStream  =  new  FileInputStream(oldFile);  //读入原文件
                FileOutputStream  fs  =  new  FileOutputStream(newPath);
                write2Out(inStream ,fs) ;
                inStream.close();
            }
        }
    }

    /**
     * 重命名文件
     * @param file
     * @param name
     * @return
     */
    public static File renameFile(File file , String name ){
        String fileName = file.getParent()  + File.separator + name ;
        File dest = new File(fileName);
        file.renameTo(dest) ;
        return dest ;
    }

    /**
     * 压缩多个文件。
     * @param zipFileName 压缩输出文件名
     * @param files 需要压缩的文件
     * @return
     * @throws Exception
     */
    public static File createZip(String zipFileName, File... files) throws Exception {
        File outFile = new File(zipFileName) ;
        ZipOutputStream out = null;
        BufferedOutputStream bo = null;
        try {
            out = new ZipOutputStream(new FileOutputStream(outFile));
            bo = new BufferedOutputStream(out);

            for (File file : files) {
                zip(out, file, file.getName(), bo);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            try {
                bo.close();
            } finally {
                out.close(); // 输出流关闭
            }
        }
        return outFile;
    }

    /**
     *
     * @param zipFileName 压缩输出文件名
     * @param inputFile 需要压缩的文件
     * @return
     * @throws Exception
     */
    public static File createZip(String zipFileName, File inputFile) throws Exception {
        File outFile = new File(zipFileName) ;
        ZipOutputStream out = null;
        BufferedOutputStream bo = null;
        try {
            out = new ZipOutputStream(new FileOutputStream(outFile));
            bo = new BufferedOutputStream(out);
            zip(out, inputFile, inputFile.getName(), bo);
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            try {
                bo.close();
            } finally {
                out.close(); // 输出流关闭
            }
        }
        return outFile;
    }

    private static void zip(ZipOutputStream out, File f, String base,BufferedOutputStream bo) throws Exception { // 方法重载
        if (f.isDirectory()) {
            File[] fl = f.listFiles();
            if ( fl == null ||  fl.length == 0) {
                out.putNextEntry(new ZipEntry(base + "/")); // 创建创建一个空的文件夹
            }else{
                for (int i = 0; i < fl.length; i++) {
                    zip(out, fl[i], base + "/" + fl[i].getName(), bo); // 递归遍历子文件夹
                }
            }

        } else {
            out.putNextEntry(new ZipEntry(base)); // 创建zip压缩进入 base 文件
            System.out.println(base);
            BufferedInputStream bi = new BufferedInputStream(new FileInputStream(f));

            try {
                write2Out(bi,out) ;
            } catch (IOException e) {
                //Ignore
            }finally {
                bi.close();// 输入流关闭
            }
        }
    }

    private static void write2Out(InputStream input , OutputStream out) throws IOException {
        byte[] b = new byte[1024];
        int c = 0 ;
        while ( (c = input.read(b)) != -1 ) {
            out.write(b,0,c);
            out.flush();
        }
        out.flush();
    }

    /**
     * 使用FileUtils拷贝文件
     * @param source
     * @param dest
     * @throws IOException
     */
    public static void copyFileUsingApacheCommonsIO(File source, File dest)
            throws IOException {
        FileUtils.copyFile(source, dest);
    }
}