按照文件自身层级来导出文件并压缩


import lombok.extern.slf4j.Slf4j;
import org.apache.logging.log4j.util.Strings;
import org.junit.runner.RunWith;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

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

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

@Slf4j
@RunWith(SpringJUnit4ClassRunner.class)
public class Test {

    @org.junit.Test
    public void main() throws Exception {
        //下载每条数据的所有附件.
        //然后分层级.zip/数据名称/N个附件
        List<File> list = new ArrayList<>();
        File a = new File("E:" + File.separator +"测试A.txt");
       
        File fileA = new File(File.separator +"tmp/数据A/测试12.txt");
        
        
        File b = new File("E:" + File.separator +"测试B.txt");
        File fileB = new File(File.separator +"tmp/数据A/测试112.txt");
        
        
        File c = new File("E:" + File.separator +"测试C.txt");
        File fileC = new File(File.separator +"tmp/数据C/测试阿萨德.txt");
        
        
        File d = new File("E:" + File.separator +"测试D.txt");
        File fileD = new File(File.separator +"tmp/数据D/测试12打是.txt");

        String parent = fileD.getParent();
        
        copyFile(a,fileA);
        copyFile(b,fileB);
        copyFile(c,fileC);
        copyFile(d,fileD);
        
        File file = new File(File.separator+"tmp");

        list.add(file);
        toZips(list, "/testzip.zip");
    }

    public static void toZips(List<File> srcFiles, String targetDir) throws RuntimeException {
        long start = System.currentTimeMillis();

        FileOutputStream out = null;
        String targetName = "";
        String sourcePath = srcFiles.get(0).getParent();

        ZipOutputStream zos = null;
        try {
            if (Strings.isEmpty(targetDir)) {
                if (srcFiles.size() > 0 && srcFiles != null) {
                    targetName = srcFiles.get(srcFiles.size() - 1).getName();//获得最后一个文件的名字
                    targetName = targetName.substring(0, targetName.lastIndexOf('.'));
                }
                out = new FileOutputStream(new File(sourcePath + "/" + targetName + ".zip"));

            } else {
                out = new FileOutputStream(new File(targetDir));
            }
            zos = new ZipOutputStream(out);
            for (File srcFile : srcFiles) {
                compress(srcFile, zos, srcFile.getName(), true);
            }
            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();
                }
            }
        }
    }


    /**
     * 递归压缩方法
     *
     * @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);
                    }
                }
            }
        }
    }
    
    public void copyFile(File start,File end) {
        if (end.exists()) {
            end.delete();
        }
        //目录不存在 则创建
        if (!end.getParentFile().exists()) {
            boolean mkdir = end.getParentFile().mkdirs();
            if (!mkdir) {
                throw new RuntimeException("创建目标文件所在目录失败!");
            }
        }
        try(BufferedInputStream bis=new BufferedInputStream(new FileInputStream(start));
            BufferedOutputStream bos=new BufferedOutputStream(new FileOutputStream(end))) {
            int len = 0;
            byte[] flush = new byte[1024];
            while((len=bis.read(flush)) != -1) {
                bos.write(flush, 0, len);
            }
            bos.flush();
        }catch(FileNotFoundException e) {
            e.printStackTrace();
        }catch(IOException e) {
            e.printStackTrace();
        }
    }
    
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值