java zip 打包

Posted 2022-05-30 chengyangyang

package com.sxmd.utils;

import com.sxmd.exception.SxmdException;
import lombok.extern.slf4j.Slf4j;

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

/**
 * Description:  打包工具类
 *
 * @author cy
 * @date 2019年07月30日 21:15
 * version 1.0
 */
@Slf4j
public class ZipUtil 

    private ZipUtil() 
    

    /**
     * 创建ZIP文件
     * @param sourcePath 文件或文件夹路径
     * @param zipPath 生成的zip文件存在路径（包括文件名）
     * @param isDrop  是否删除原文件:true删除、false不删除
     */
    public static boolean createZip(String sourcePath, String zipPath,Boolean isDrop) 
        boolean flag = true;
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        try 
            fos = new FileOutputStream(zipPath);
            zos = new ZipOutputStream(fos);
            writeZip(new File(sourcePath), "", zos,isDrop);
         catch (Exception e) 
            flag = false;
            log.error("创建ZIP文件失败",e);
         finally 
            try 
                if (zos != null) 
                    zos.close();
                
             catch (IOException e) 
                log.error("流关闭失败",e);
            

        
        return flag;
    
    /**
     * 清空文件和文件目录
     *
     * @param f
     */
    public static void clean(File f)  
        String[] cs = f.list();
        if (cs == null || cs.length <= 0) 
            log.debug("delFile:[ " + f + " ]");
            boolean isDelete = f.delete();
            if (!isDelete) 
                log.error("delFile:[ " + f.getName() + "文件删除失败！" + " ]");
                throw new SxmdException(f.getName() + "文件删除失败！");
            
         else 
            for (int i = 0; i < cs.length; i++) 
                String cn = cs[i];
                String cp = f.getPath() + File.separator + cn;
                File f2 = new File(cp);
                if (f2.exists() && f2.isFile()) 
                    log.error("delFile:[ " + f2 + " ]");
                    boolean isDelete = f2.delete();
                    if (!isDelete) 
                        System.out.println("delFile:[ " + f2.getName() + "文件删除失败！" + " ]");
                        throw new SxmdException(f2.getName() + "文件删除失败！");
                    
                 else if (f2.exists() && f2.isDirectory()) 
                    clean(f2);
                
            
            System.out.println("delFile:[ " + f + " ]");
            boolean isDelete = f.delete();
            if (!isDelete) 
                System.out.println("delFile:[ " + f.getName() + "文件删除失败！" + " ]");
                throw new SxmdException(f.getName() + "文件删除失败！");
            
        
    
    private static void writeZip(File file, String parentPath, ZipOutputStream zos,Boolean isDrop) 
        if(file.exists())
            // 处理文件夹
            if(file.isDirectory())
                parentPath+=file.getName()+File.separator;
                File [] files=file.listFiles();
                if(files.length != 0)
                
                    for(File f:files)
                        writeZip(f, parentPath, zos,isDrop);
                    
                
                else
                       //空目录则创建当前目录
                    try 
                        zos.putNextEntry(new ZipEntry(parentPath));
                     catch (IOException e) 
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                    
                
            else
                FileInputStream fis=null;
                try 
                    fis=new FileInputStream(file);
                    ZipEntry ze = new ZipEntry(parentPath + file.getName());
                    zos.putNextEntry(ze);
                    byte [] content=new byte[1024];
                    int len;
                    while((len=fis.read(content))!=-1)
                        zos.write(content,0,len);
                        zos.flush();
                    

                 catch (FileNotFoundException e) 
                    log.error("创建ZIP文件失败",e);
                 catch (IOException e) 
                    log.error("创建ZIP文件失败",e);
                finally
                    try 
                        if(fis!=null)
                            fis.close();
                        
                        if(isDrop)
                            clean(file);
                        
                    catch(IOException e)
                        log.error("创建ZIP文件失败",e);
                     catch (Exception e) 
                        e.printStackTrace();
                    
                
            
        
    


    public static void main(String[] args) 
        System.out.println(createZip("E:\\sxmd_plug_log","E:\\a.zip",false));