Rust中如何将ZipWriter转换为Bytes字节数据

原有代码问题 #

• 采用固定长度64KB的字节数组作为写入缓冲区，当压缩后的Zip体积超过64KB时会直接触发写入错误，固定长度数组也无法动态适配Zip的实际大小。
• 逻辑顺序错误：zip.finish()必须在所有待打包文件全部写入Zip之后调用，调用完成后才能获取到完整的Zip数据，不能在写入文件前就尝试转换缓冲区。

实现代码 #

使用动态扩容的Vec<u8>作为内存缓冲区，所有操作均在内存中完成，不会写入本地磁盘：

use std::io::{Cursor, Write};
use zip::{write::FileOptions, CompressionMethod, ZipWriter};

pub fn to_zip() -> Vec<u8> {
    // 初始化动态扩容的内存缓冲区
    let buffer = Vec::new();
    let cursor = Cursor::new(buffer);
    let mut zip_writer = ZipWriter::new(cursor);

    // 配置Zip内文件的压缩参数：DEFLATE压缩，默认644权限
    let file_option = FileOptions::default()
        .compression_method(CompressionMethod::Deflated)
        .unix_permissions(0o644);

    // 示例：写入一个名为readme.txt的文件到压缩包
    zip_writer.start_file("readme.txt", file_option).unwrap();
    zip_writer.write_all(b"this is content in zip file").unwrap();
    // 需要添加多个文件时，重复上述start_file + write_all逻辑即可

    // 完成Zip写入，获取底层包裹的Cursor
    let finished_zip = zip_writer.finish().unwrap();

    // 从Cursor中取出存储完整Zip数据的Vec<u8>，即为目标字节数据
    let zip_bytes = finished_zip.into_inner();

    // 该字节数组可直接作为POST请求体发送
    zip_bytes
}

关键说明 #

• Vec<u8>会根据Zip的实际大小自动扩容，不存在固定缓冲区溢出的问题，所有读写操作均在堆内存完成，不会产生本地临时文件。
• Cursor的into_inner()方法会直接返回其包裹的底层Vec<u8>，不需要额外做内存拷贝，性能开销很低。
• 最终得到的zip_bytes为Vec<u8>类型，本身就是标准字节格式，主流Rust HTTP客户端（如reqwest）可直接将其作为请求体传入，不需要额外转换。

内容的提问来源于stack exchange，提问作者Rafael R. Camargo