标签 目录 下的文章

写工具时通常会生成ZIP文件,一般来说就传相对路径和内容就好。但是今天发现一个细节。还需要将文件所在的目录也要传进去。虽然现在的解压软件都支持不传,但是如果解压的代码不规范就会造成保存路径不存在,因为父级目录不存在:

let file: File = File::create(zip_path).unwrap();  
let mut writer: ZipWriter<File> = ZipWriter::new(file);
let options = zip::write::FileOptions::default()
    .compression_method(zip::CompressionMethod::Deflated);

// 因为这里已经将所有目录 遍历好。目录结构已经不存在的,只能通过文件路径反向还原,并使用map去重
let mut dirMap:HashMap<String, bool> = HashMap::new();

// .... 中间省略循环遍历
// 先将文件目录结构创建
let arr:Vec<&str> = path.split('/').collect();
let mut tmp_dir: String = String::new();
for i in 0..(arr.len() - 1) {
    let tmp = arr[i];
    tmp_dir.push('/');
    tmp_dir.push_str(tmp);
    if dirMap.contains_key(&tmp_dir){ // 缓存保存过的目录
        continue;
    }
    dirMap.insert(tmp_dir.clone(), true);
    writer.add_directory(&tmp_dir, options).unwrap();
}

// 将文件写入ZIP
let mut buffer = Vec::new();
let mut file = File::open(&tmp_path).unwrap();
file.read_to_end(&mut buffer).unwrap();
writer.start_file(path, options).unwrap();
writer.write_all(&buffer).unwrap();