菜单 学习猿地 - LMONKEY

VIP

开通学习猿地VIP

尊享10项VIP特权 持续新增

知识通关挑战

打卡带练!告别无效练习

接私单赚外块

VIP优先接,累计金额超百万

学习猿地私房课免费学

大厂实战课仅对VIP开放

你的一对一导师

每月可免费咨询大牛30次

领取更多软件工程师实用特权

入驻
246
0

Java-ZipUtil

原创
05/13 14:22
阅读数 85968

Zip 压缩工具类,不支持压缩空文件夹。

 

简单版

import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.time.Instant;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipUtil {
    public static void main(String[] args) {
        zipCompression("D:\\123.zip", "D:\\123", "D:\\456", "D:\\er4.zip");
    }

    static void zipCompression(String zipPath, String... paths) {
        Path[] ps = new Path[paths.length];
        for (int i = 0; i < paths.length; i++) {
            ps[i] = Paths.get(paths[i]);
        }
        zipCompression(Paths.get(zipPath), ps);
    }

    static void zipCompression(Path zipPath, Path... paths) {
        long beginTime = Instant.now().toEpochMilli();
        try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipPath.toFile()))) {
            for (Path path : paths) {
                Files.walkFileTree(path, new SimpleFileVisitor<Path>() {
                    @Override // 访问一个文件
                    public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
                        zos.putNextEntry(new ZipEntry(file.toString().replace(path.getParent().toString(), "")));
                        Files.copy(file, zos);
                        zos.closeEntry();
                        return FileVisitResult.CONTINUE;
                    }
                });
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println("耗时:" + (Instant.now().toEpochMilli() - beginTime));
    }
}

 

内存映射+管道+异步线程版,效率似乎没有什改变。。。。。。

import java.io.*;
import java.nio.ByteBuffer;
import java.nio.MappedByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.FileChannel;
import java.nio.channels.Pipe;
import java.nio.channels.WritableByteChannel;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.time.Instant;
import java.util.concurrent.CompletableFuture;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipUtil {
    public static void main(String[] args) {
        zipCompression("D:\\123.zip", "D:\\123", "D:\\456", "D:\\er4.zip");
    }

    static void zipCompression(String zipPath, String... paths) {
        Path[] ps = new Path[paths.length];
        for (int i = 0; i < paths.length; i++) {
            ps[i] = Paths.get(paths[i]);
        }
        zipCompression(Paths.get(zipPath), ps);
    }

    static void zipCompression(Path zipPath, Path... paths) {
        long beginTime = Instant.now().toEpochMilli();
        try (FileOutputStream fileOutputStream = new FileOutputStream(zipPath.toFile());
             WritableByteChannel out = Channels.newChannel(fileOutputStream)) {
            Pipe pipe = Pipe.open();
            // 异步任务往通道中塞入数据
            CompletableFuture.runAsync(() -> runCompressionTask(pipe, paths));
            // 读取通道中数据
            Pipe.SourceChannel source = pipe.source();

            ByteBuffer buffer = ByteBuffer.allocate(2048);
            // ByteBuffer buffer = ByteBuffer.allocateDirect(2048);
            while (source.read(buffer) >= 0) {
                buffer.flip();
                out.write(buffer);
                buffer.clear();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println("耗时:" + (Instant.now().toEpochMilli() - beginTime));
    }

    // 异步任务
    public static void runCompressionTask(Pipe pipe, Path... paths) {
        try (Pipe.SinkChannel sink = pipe.sink();
             OutputStream os = Channels.newOutputStream(sink);
             ZipOutputStream zos = new ZipOutputStream(os);
             WritableByteChannel out = Channels.newChannel(zos)) {

            for (Path path : paths) {
                Files.walkFileTree(path, new SimpleFileVisitor<Path>() {
                    @Override // 访问一个目录
                    public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
                        if (dir.toFile().list().length == 0) {
                            // 无法打包空文件夹
                            // zos.putNextEntry(new ZipEntry(dir.toString().replace(path.getParent().toString(), "") + File.separator));
                            // System.out.println(dir.toString().replace(path.getParent().toString(), "") + File.separator);
                            // zos.closeEntry();
                        }
                        return FileVisitResult.CONTINUE;
                    }

                    @Override // 访问一个文件
                    public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
                        zos.putNextEntry(new ZipEntry(file.toString().replace(path.getParent().toString(), "")));

                        MappedByteBuffer mappedByteBuffer = new RandomAccessFile(file.toFile(), "r").getChannel().map(FileChannel.MapMode.READ_ONLY, 0, attrs.size());
                        out.write(mappedByteBuffer);

                        // FileChannel fileChannel = new FileInputStream(file.toFile()).getChannel();
                        // fileChannel.transferTo(0, fileChannel.size(), out);
                        // fileChannel.close();

                        zos.closeEntry();
                        return FileVisitResult.CONTINUE;
                    }
                });
            }
            zos.finish();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

 

用到了 NIO 相关特性


https://juejin.im/post/5d5626cdf265da03a65312be

https://www.cnblogs.com/jhxxb/p/11272727.html

https://www.cnblogs.com/jhxxb/p/11303947.html

发表评论

0/200
246 点赞
0 评论
收藏