mirror
/
borg
mirror of https://github.com/borgbackup/borg.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Implemented Repository.commit() and Repository.apply_txn()

This commit is contained in:
Jonas Borgström 2010-02-15 21:38:17 +01:00
parent 7d6aa3763d
commit 02c2490959
1 changed files with 30 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
dedupstore/repository.py
View File

@ -49,7 +49,9 @@ class Repository(object):
if os.path.exists(os.path.join(self.path, 'txn-active')):
self.rollback()
if os.path.exists(os.path.join(self.path, 'txn-commit')):
self.commit()
self.apply_txn()
if os.path.exists(os.path.join(self.path, 'txn-applied')):
shutil.rmtree(os.path.join(self.path, 'txn-applied'))
self.state = Repository.OPEN
def close(self):
@ -66,13 +68,33 @@ class Repository(object):
remove_fd = open(os.path.join(self.path, 'txn-active', 'remove'), 'wb')
remove_fd.write('\n'.join(self.txn_removed))
remove_fd.close()
tid_fd = open(os.path.join(self.path, 'txn-active', 'tid'), 'wb')
tid_fd.write(str(self.tid + 1))
tid_fd.close()
os.rename(os.path.join(self.path, 'txn-active'),
os.path.join(self.path, 'txn-commit'))
self.commit_txn()
self.apply_txn()
def commit_txn(self):
def apply_txn(self):
assert os.path.isdir(os.path.join(self.path, 'txn-commit'))
raise NotImplementedError
tid = int(open(os.path.join(self.path, 'txn-commit', 'tid'), 'rb').read())
assert tid >= self.tid
remove_list = [line.strip() for line in
open(os.path.join(self.path, 'txn-commit', 'remove'), 'rb').readlines()]
for name in remove_list:
path = os.path.join(self.path, 'data', name)
os.unlink(path)
add_path = os.path.join(self.path, 'txn-commit', 'add')
for name in os.listdir(add_path):
shutil.move(os.path.join(add_path, name),
os.path.join(self.path, 'data'))
tid_fd = open(os.path.join(self.path, 'tid'), 'wb')
tid_fd.write(str(tid))
tid_fd.close()
os.rename(os.path.join(self.path, 'txn-commit'),
os.path.join(self.path, 'txn-applied'))
shutil.rmtree(os.path.join(self.path, 'txn-applied'))
self.state = Repository.IDLE
def rollback(self):
"""
@ -88,7 +110,6 @@ class Repository(object):
elif self.state == Repository.OPEN:
os.mkdir(os.path.join(self.path, 'txn-active'))
os.mkdir(os.path.join(self.path, 'txn-active', 'add'))
open(os.path.join(self.path, 'txn-active', 'tid'), 'wb').write(str(self.tid + 1))
self.txn_removed = []
self.state = Repository.ACTIVE
@ -176,7 +197,11 @@ class RepositoryTestCase(unittest.TestCase):
self.assertRaises(Exception, lambda: self.repo.get_file('foo'))
self.assertEqual(self.repo.listdir(''), ['bar'])
self.assertEqual(self.repo.state, Repository.ACTIVE)
self.assertEqual(os.path.exists(os.path.join(self.tmppath, 'repo', 'data', 'bar')), False)
self.repo.commit()
self.assertEqual(os.path.exists(os.path.join(self.tmppath, 'repo', 'data', 'bar')), True)
self.assertEqual(self.repo.listdir(''), ['bar'])
self.assertEqual(self.repo.state, Repository.IDLE)
if __name__ == '__main__':
unittest.main()