mirror of https://github.com/dswd/zvault
58 lines
2.0 KiB
Rust
58 lines
2.0 KiB
Rust
use prelude::*;
|
|
|
|
use std::collections::HashMap;
|
|
|
|
|
|
impl Repository {
|
|
pub fn delete_bundle(&mut self, id: u32) -> Result<(), RepositoryError> {
|
|
if let Some(bundle) = self.bundle_map.remove(id) {
|
|
try!(self.bundles.delete_bundle(&bundle));
|
|
Ok(())
|
|
} else {
|
|
Err(IntegrityError::MissingBundleId(id).into())
|
|
}
|
|
}
|
|
|
|
pub fn rewrite_bundles(&mut self, rewrite_bundles: &[u32], usage: &HashMap<u32, BundleAnalysis>) -> Result<(), RepositoryError> {
|
|
for &id in ProgressIter::new(
|
|
tr!("rewriting bundles"),
|
|
rewrite_bundles.len(),
|
|
rewrite_bundles.iter()
|
|
)
|
|
{
|
|
let bundle = &usage[&id];
|
|
let bundle_id = self.bundle_map.get(id).unwrap();
|
|
let chunks = try!(self.bundles.get_chunk_list(&bundle_id));
|
|
let mode = usage[&id].info.mode;
|
|
for (chunk, &(hash, _len)) in chunks.into_iter().enumerate() {
|
|
if !bundle.chunk_usage.get(chunk) {
|
|
try!(self.index.delete(&hash));
|
|
continue;
|
|
}
|
|
let data = try!(self.bundles.get_chunk(&bundle_id, chunk));
|
|
try!(self.put_chunk_override(mode, hash, &data));
|
|
}
|
|
}
|
|
try!(self.flush());
|
|
tr_info!("Checking index");
|
|
for (hash, location) in self.index.iter() {
|
|
let loc_bundle = location.bundle;
|
|
let loc_chunk = location.chunk;
|
|
if rewrite_bundles.contains(&loc_bundle) {
|
|
tr_panic!(
|
|
"Removed bundle is still referenced in index: hash:{}, bundle:{}, chunk:{}",
|
|
hash,
|
|
loc_bundle,
|
|
loc_chunk
|
|
);
|
|
}
|
|
}
|
|
tr_info!("Deleting {} bundles", rewrite_bundles.len());
|
|
for &id in rewrite_bundles {
|
|
try!(self.delete_bundle(id));
|
|
}
|
|
try!(self.save_bundle_map());
|
|
Ok(())
|
|
}
|
|
}
|