如何用Java函数式写法实现ZipInputStream解压无目录结构zip包

核心思路 #

由于ZIP条目数量无法提前获知，我们可以借助Java 9+引入的Stream.takeWhile()方法实现按需迭代，完全替换原有while循环逻辑，核心步骤如下：

• 使用Stream.generate()封装ZipInputStream::getNextEntry调用，持续生成下一个ZIP条目
• 使用takeWhile(Objects::nonNull)自动终止流，直到读取到null（代表无更多条目）
• 原有路径安全校验、文件复制逻辑直接作为流的终端操作执行
• 资源仍通过try-with-resources管理，和原有逻辑一致无泄漏风险

完整实现代码 #

import java.io.IOException;
import java.io.UncheckedIOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import java.util.Objects;
import java.util.stream.Stream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;

public static void unzip(Path source, Path target) throws IOException {
    try (ZipInputStream zis = new ZipInputStream(Files.newInputStream(source))) {
        // 生成ZIP条目流，直到读取到null停止
        Stream.generate(zis::getNextEntry)
              .takeWhile(Objects::nonNull)
              .forEach(zipEntry -> {
                  try {
                      Path normalizePath = target.resolve(zipEntry.getName()).normalize();
                      // 保留原有的zip slip漏洞防护逻辑
                      if (!normalizePath.startsWith(target)) {
                          throw new IOException("Bad zip entry: " + zipEntry.getName());
                      }
                      Files.copy(zis, normalizePath, StandardCopyOption.REPLACE_EXISTING);
                      zis.closeEntry();
                  } catch (IOException e) {
                      // 函数式接口不支持抛出受检异常，封装为运行时异常抛出
                      throw new UncheckedIOException(e);
                  }
              });
    } catch (UncheckedIOException e) {
        // 把封装的运行时异常转回原方法声明的受检异常
        throw e.getCause();
    }
}

注意事项 #

• 该实现和原有逻辑完全等价，保留了所有安全校验和功能特性
• 如果使用Java 8版本，没有内置的takeWhile方法，可以自定义流迭代器实现对应能力，核心思路不变
• 由于ZIP条目读取是强顺序依赖，流默认采用串行执行，不会出现并发问题

内容的提问来源于stack exchange，提问作者Thanos M