rust web开发中直接在内存中生成zip文件返回

在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开发中写到响应中即可
}

 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值