深入解析zip.js中的流式压缩与解压缩实现

在JavaScript生态中，zip.js是一个功能强大的ZIP文件处理库。本文将深入探讨如何在该库中实现流式压缩和解压缩功能，这对于处理大型ZIP文件或需要内存高效处理的场景尤为重要。

流式解压缩的实现

zip.js原有的ZipReader类提供了基本的解压缩功能，但缺乏直接的流式接口。通过创建一个ZipDecompressStream转换流，我们可以实现更优雅的流式处理：

class ZipDecompressStream {
  constructor(options) {
    // 创建转换流
    const {readable, writable} = new TransformStream()
    
    // 使用ZipReader处理输入流
    const gen = new ZipReader(readable, options).getEntriesGenerator()
    
    // 输出流将包含解压后的条目信息
    this.readable = new ReadableStream({
      async pull(controller) {
        const {done, value} = await gen.next()
        if (done) return controller.close()
        
        // 为每个条目提供可读流接口
        const chunk = {
          ...value,
          readable: value.getData ? createReadableStream(value) : undefined
        }
        delete chunk.getData
        controller.enqueue(chunk)
      }
    })
    this.writable = writable
  }
}

这种实现方式允许开发者直接将网络响应流通过pipeThrough传递给解压缩流，无需等待整个文件下载完成即可开始处理。

流式压缩的演进

最初的压缩流实现仅支持单个文件压缩，但经过迭代后发展出了更强大的MultiZipCompressStream：

class MultiZipCompressStream {
  constructor(options) {
    const {readable, writable} = new TransformStream()
    this.#readable = readable
    this.#zipWriter = new ZipWriter(writable, options)
  }

  // 为每个文件创建独立的可写流
  writable(path) {
    const {readable, writable} = new TransformStream()
    this.#zipWriter.add(path, readable)
    return writable
  }

  // 必须显式调用close方法完成压缩
  close(comment, options) {
    return this.#zipWriter.close(comment, options)
  }
}

最终版本ZipCompressStream进一步整合了功能，既支持单文件转换流模式，也支持多文件并行压缩：

class ZipCompressStream {
  // 单文件转换模式
  transform(path) {
    const {readable, writable} = new TransformStream({
      flush: () => this.#zipWriter.close()
    })
    this.#zipWriter.add(path, readable)
    return {readable: this.#readable, writable}
  }

  // 多文件并行模式
  writable(path) {
    const {readable, writable} = new TransformStream()
    this.#zipWriter.add(path, readable)
    return writable
  }
}

实际应用场景

这些流式接口特别适合以下场景：

1. 网络文件处理：直接从网络下载ZIP文件并实时解压处理
2. 大文件处理：避免将整个文件加载到内存中
3. 管道式处理：与其他流处理器(如CSV解析器)无缝衔接

示例应用展示了如何下载、解压、处理并重新压缩文件：

// 下载并解压
for await (const entry of fetch(url).body.pipeThrough(new ZipDecompressStream())) {
  if (entry.readable) {
    // 处理CSV内容
    const csvStream = entry.readable
      .pipeThrough(new TextDecoderStream())
      .pipeThrough(new CsvParseStream())
    
    // 重新压缩
    entry.readable
      .pipeThrough(new ZipCompressStream().transform(entry.filename))
      .pipeTo(createWriteStream(entry.filename + '.zip'))
  }
}

注意事项

1. 资源清理：必须完全消费解压缩流或显式调用cancel()，否则会导致内存泄漏
2. 错误处理：压缩过程中取消操作可能导致ZIP文件损坏
3. 并行控制：多文件压缩时需要协调各流的完成状态

通过zip.js的这些流式扩展，开发者可以构建更高效、更灵活的ZIP文件处理应用，特别是在需要处理大型文件或实现复杂处理管道的场景下。