mirror/borg
1
0
Fork 0
mirror of https://github.com/borgbackup/borg.git synced 2025-02-20 21:27:32 +00:00
Code Issues Releases Wiki Activity

cache sync: convert existing archive chunks idx to compact

Browse source
This commit is contained in:
Marian Beermann 2017-06-09 11:07:17 +02:00
parent 295ac86d71
commit 09a9d892cf
1 changed files with 26 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
src/borg/cache.py
View file

@ -563,12 +563,14 @@ def cleanup_outdated(ids):
for id in ids: for id in ids:
cleanup_cached_archive(id) cleanup_cached_archive(id)
def cleanup_cached_archive(id): def cleanup_cached_archive(id, cleanup_compact=True):
try: try:
os.unlink(mkpath(id)) os.unlink(mkpath(id))
os.unlink(mkpath(id) + '.integrity') os.unlink(mkpath(id) + '.integrity')
except FileNotFoundError: except FileNotFoundError:
pass pass
if not cleanup_compact:
return
try: try:
os.unlink(mkpath(id, suffix='.compact')) os.unlink(mkpath(id, suffix='.compact'))
os.unlink(mkpath(id, suffix='.compact') + '.integrity') os.unlink(mkpath(id, suffix='.compact') + '.integrity')
@ -578,7 +580,6 @@ def cleanup_cached_archive(id):
def fetch_and_build_idx(archive_id, decrypted_repository, chunk_idx): def fetch_and_build_idx(archive_id, decrypted_repository, chunk_idx):
nonlocal processed_item_metadata_bytes nonlocal processed_item_metadata_bytes
nonlocal processed_item_metadata_chunks nonlocal processed_item_metadata_chunks
nonlocal compact_chunks_archive_saved_space
csize, data = decrypted_repository.get(archive_id) csize, data = decrypted_repository.get(archive_id)
chunk_idx.add(archive_id, 1, len(data), csize) chunk_idx.add(archive_id, 1, len(data), csize)
archive = ArchiveItem(internal_dict=msgpack.unpackb(data)) archive = ArchiveItem(internal_dict=msgpack.unpackb(data))
@ -591,17 +592,21 @@ def fetch_and_build_idx(archive_id, decrypted_repository, chunk_idx):
processed_item_metadata_chunks += 1 processed_item_metadata_chunks += 1
sync.feed(data) sync.feed(data)
if self.do_cache: if self.do_cache:
compact_chunks_archive_saved_space += chunk_idx.compact() write_archive_index(archive_id, chunk_idx)
fn = mkpath(archive_id, suffix='.compact')
fn_tmp = mkpath(archive_id, suffix='.tmp') def write_archive_index(archive_id, chunk_idx):
try: nonlocal compact_chunks_archive_saved_space
with DetachedIntegrityCheckedFile(path=fn_tmp, write=True, compact_chunks_archive_saved_space += chunk_idx.compact()
filename=bin_to_hex(archive_id)) as fd: fn = mkpath(archive_id, suffix='.compact')
chunk_idx.write(fd) fn_tmp = mkpath(archive_id, suffix='.tmp')
except Exception: try:
os.unlink(fn_tmp) with DetachedIntegrityCheckedFile(path=fn_tmp, write=True,
else: filename=bin_to_hex(archive_id)) as fd:
os.rename(fn_tmp, fn) chunk_idx.write(fd)
except Exception:
os.unlink(fn_tmp)
else:
os.rename(fn_tmp, fn)
def get_archive_ids_to_names(archive_ids): def get_archive_ids_to_names(archive_ids):
# Pass once over all archives and build a mapping from ids to names. # Pass once over all archives and build a mapping from ids to names.
@ -642,11 +647,19 @@ def create_master_idx(chunk_idx):
logger.info("Reading cached archive chunk index for %s ...", archive_name) logger.info("Reading cached archive chunk index for %s ...", archive_name)
try: try:
try: try:
# Attempt to load compact index first
with DetachedIntegrityCheckedFile(path=archive_chunk_idx_path + '.compact', write=False) as fd: with DetachedIntegrityCheckedFile(path=archive_chunk_idx_path + '.compact', write=False) as fd:
archive_chunk_idx = ChunkIndex.read(fd, permit_compact=True) archive_chunk_idx = ChunkIndex.read(fd, permit_compact=True)
# In case a non-compact index exists, delete it.
cleanup_cached_archive(archive_id, cleanup_compact=False)
except FileNotFoundError: except FileNotFoundError:
# No compact index found, load non-compact index
with DetachedIntegrityCheckedFile(path=archive_chunk_idx_path, write=False) as fd: with DetachedIntegrityCheckedFile(path=archive_chunk_idx_path, write=False) as fd:
archive_chunk_idx = ChunkIndex.read(fd) archive_chunk_idx = ChunkIndex.read(fd)
# Automatically convert to compact index. Delete the existing index first.
logger.debug('Found non-compact index for %s, converting to compact.', archive_name)
cleanup_cached_archive(archive_id)
write_archive_index(archive_id, archive_chunk_idx)
except FileIntegrityError as fie: except FileIntegrityError as fie:
logger.error('Cached archive chunk index of %s is corrupted: %s', archive_name, fie) logger.error('Cached archive chunk index of %s is corrupted: %s', archive_name, fie)
# Delete it and fetch a new index # Delete it and fetch a new index