在rust中使用zip进行压缩时,尤其是在web开发中,生成多个报表打包成一个zip文件,不想生成临时文件,一般喜欢直接在内存中生成好zip文件,然后序列化成流响应返回给web端。
cargo.toml
zip = "0.5.5"
walkdir = "2"
代码:
use std::fs::File;
use std::io::{copy, Read, Seek, Write, sink, BufWriter};
use std::path::Path;
use std::str;
use std::fs;
use walkdir::{DirEntry, WalkDir};
use zip::write::FileOptions;
use zip::ZipWriter;
/// 在内存中生成zip文件
/// 一般在做web开发时,都喜欢在内存中动态生成报表多,然后直接打包成一个
pub fn compress<T>(zip: &mut ZipWriter<T>, file_name: &str, b: &[u8]) -> zip::result::ZipResult<()>
where T: Write + Seek {
let options = FileOptions::default()
.compression_method(zip::CompressionMethod::Bzip2)//直接用了bzip2压缩方式,其它参看枚举
.unix_permissions(0o755);//unix系统权限
println!("adding file {:?} ...", file_name);
zip.start_file(file_name, options)?;
zip.write_all(b)?;
Result::Ok(())
}
#[test]
fn zip() {
let buf=vec![];
// Cursor A Cursor wraps an in-memory buffer and provides it with a Seek implementation.
let mut writer = std::io::Cursor::new(buf);
let mut zip = zip::ZipWriter::new(writer);//参数要求必须实现io:Seek,Cursor满足要求
let spit_result: Vec<&[u8]> = vec![b"hello", b"rust", b"golang", b"java"];
for (i, subPersons) in spit_result.iter().enumerate() {//循环增加三个文件
compress(&mut zip, &format!("person/person{}.json", i + 1), subPersons);
}
let writer= zip.finish().unwrap();//到这一步就转成生成的字节了
let mut f = fs::File::create("d:/person.zip").unwrap();
f.write_all(writer.get_ref()).unwrap()//写入到文件中,在web开发中写到响应中即可
}