view mercurial/ @ 42:91f1fa847158

Fix multiple changes to file per transaction
date Tue, 10 May 2005 00:31:00 -0800
parents eb87b7dc4236
children 42177b56b949
line wrap: on
line source

# - 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 <>
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.

import os

class transaction:
    def __init__(self, opener, journal):
        self.opener = opener
        self.entries = [] = {}
        self.journal = journal

        # abort here if the journal already exists
        if os.path.exists(self.journal):
            raise "Journal already exists!"
        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 return
        self.entries.append((file, offset))[file] = 1
        # add enough data to the journal to do the truncate
        self.file.write("%s\0%d\n" % (file, offset))

    def close(self):
        self.entries = []

    def abort(self):
        if not self.entries: return

        print "transaction abort!"

        for f, o in self.entries:
            self.opener(f, "a").truncate(o)

        self.entries = []

        print "rollback completed"
    def recover(self):
        for l in open(self.journal).readlines():
            f, o = l.split('\0')
            self.opener(f, "a").truncate(int(o))