mirror of
https://github.com/facebook/sapling.git
synced 2024-10-09 08:18:15 +03:00
5e62f2ce10
-----BEGIN PGP SIGNED MESSAGE----- Hash: SHA1 [PATCH] rename under the other OS From: K Thananchayan <thananck@yahoo.com> Rename fails under windows if dest file exists. This patch add a rename method to util module that removes the dest file and retries if initial attempt fails. manifest hash: 2744d9fd1717e15133b411a269df909fa8ec0faf -----BEGIN PGP SIGNATURE----- Version: GnuPG v1.4.0 (GNU/Linux) iD8DBQFCuNzDywK+sNU5EO8RAsPBAJ9NQN3bTuJvTem5x+utGnoMkhYc0QCbBFSJ PoMP69H1UnVE6drFlnlFE0s= =pePw -----END PGP SIGNATURE-----
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
# transaction.py - simple journalling scheme for mercurial
|
|
#
|
|
# This transaction scheme is intended to gracefully handle program
|
|
# errors and interruptions. More serious failures like system crashes
|
|
# can be recovered with an fsck-like tool. As the whole repository is
|
|
# effectively log-structured, this should amount to simply truncating
|
|
# anything that isn't referenced in the changelog.
|
|
#
|
|
# Copyright 2005 Matt Mackall <mpm@selenic.com>
|
|
#
|
|
# This software may be used and distributed according to the terms
|
|
# of the GNU General Public License, incorporated herein by reference.
|
|
|
|
import os
|
|
import util
|
|
|
|
class transaction:
|
|
def __init__(self, opener, journal, after = None):
|
|
self.journal = None
|
|
|
|
# abort here if the journal already exists
|
|
if os.path.exists(journal):
|
|
raise "journal already exists - run hg recover"
|
|
|
|
self.opener = opener
|
|
self.after = after
|
|
self.entries = []
|
|
self.map = {}
|
|
self.journal = journal
|
|
|
|
self.file = open(self.journal, "w")
|
|
|
|
def __del__(self):
|
|
if self.entries: self.abort()
|
|
try: os.unlink(self.journal)
|
|
except: pass
|
|
|
|
def add(self, file, offset):
|
|
if file in self.map: return
|
|
self.entries.append((file, offset))
|
|
self.map[file] = 1
|
|
# add enough data to the journal to do the truncate
|
|
self.file.write("%s\0%d\n" % (file, offset))
|
|
self.file.flush()
|
|
|
|
def close(self):
|
|
self.file.close()
|
|
self.entries = []
|
|
if self.after:
|
|
util.rename(self.journal, self.after)
|
|
else:
|
|
os.unlink(self.journal)
|
|
|
|
def abort(self):
|
|
if not self.entries: return
|
|
|
|
print "transaction abort!"
|
|
|
|
for f, o in self.entries:
|
|
try:
|
|
self.opener(f, "a").truncate(o)
|
|
except:
|
|
print "failed to truncate", f
|
|
|
|
self.entries = []
|
|
|
|
print "rollback completed"
|
|
|
|
def rollback(opener, file):
|
|
for l in open(file).readlines():
|
|
f, o = l.split('\0')
|
|
opener(f, "a").truncate(int(o))
|
|
os.unlink(file)
|
|
|