mercurial/transaction.py

39675:da9ce63bfa9b

        vfsmap[''] = opener  # set default value
        self._vfsmap = vfsmap
        self.after = after
        self.entries = []
        self.map = {}
        self.journal = journalname
        self._undoname = undoname
        self._queue = []
        # A callback to validate transaction content before closing it.
        # should raise exception is anything is wrong.
        # target user is repository hooks.

        # transaction.
        self.changes = {}

        # a dict of arguments to be passed to hooks
        self.hookargs = {}
        self.file = opener.open(self.journal, "w")

        # a list of ('location', 'path', 'backuppath', cache) entries.
        # - if 'backuppath' is empty, no file existed at backup time
        # - if 'path' is empty, this is a temporary transaction file
        # - if 'location' is not empty, the path is outside main opener reach.
        #   use 'location' value as a key in a vfsmap to find the right 'vfs'
        # (cache is currently unused)
        self._backupentries = []
        self._backupmap = {}
        self._backupjournal = "%s.backupfiles" % self.journal
        self._backupsfile = opener.open(self._backupjournal, 'w')
        self._backupsfile.write('%d\n' % version)

        if createmode is not None:
            opener.chmod(self.journal, createmode & 0o666)
            opener.chmod(self._backupjournal, createmode & 0o666)

        # hold file generations to be performed on commit
        self._filegenerators = {}
        # hold callback to write pending data for hooks

        name = r'/'.join(self.names)
        return (r'<transaction name=%s, count=%d, usages=%d>' %
                (name, self._count, self._usages))

    def __del__(self):
        if self.journal:
            self._abort()

    @active
    def startgroup(self):
        """delay registration of file entry

        if file in self.map or file in self._backupmap:
            return
        vfs = self._vfsmap[location]
        dirname, filename = vfs.split(file)
        backupfilename = "%s.backup.%s" % (self.journal, filename)
        backupfile = vfs.reljoin(dirname, backupfilename)
        if vfs.exists(file):
            filepath = vfs.join(file)
            backuppath = vfs.join(backupfile)
            util.copyfile(filepath, backuppath, hardlink=hardlink)

        if self.after:
            self.after()
            self.after = None # Help prevent cycles.
        if self.opener.isfile(self._backupjournal):
            self.opener.unlink(self._backupjournal)
        if self.opener.isfile(self.journal):
            self.opener.unlink(self.journal)
        for l, _f, b, c in self._backupentries:
            if l not in self._vfsmap and c:
                self.report("couldn't remove %s: unknown cache location"
                            "%s\n" % (b, l))
                continue

                        raise
                    # Abort may be raise by read only opener
                    self.report("couldn't remove %s: %s\n"
                                % (vfs.join(b), inst))
        self._backupentries = []
        self.journal = None

        self.releasefn(self, True) # notify success of closing transaction
        self.releasefn = None # Help prevent cycles.

        # run post close action

                    self.report("couldn't remove %s: unknown cache location"
                                "%s\n" % (b, l))
                    continue
                vfs = self._vfsmap[l]
                base, name = vfs.split(b)
                assert name.startswith(self.journal), name
                uname = name.replace(self.journal, self._undoname, 1)
                u = vfs.reljoin(base, uname)
                util.copyfile(vfs.join(b), vfs.join(u), hardlink=True)
            undobackupfile.write("%s\0%s\0%s\0%d\n" % (l, f, u, c))
        undobackupfile.close()

        try:
            if not self.entries and not self._backupentries:
                if self._backupjournal:
                    self.opener.unlink(self._backupjournal)
                if self.journal:
                    self.opener.unlink(self.journal)
                return

            self.report(_("transaction abort!\n"))

            try:
                for cat in sorted(self._abortcallback):
                    self._abortcallback[cat](self)
                # Prevent double usage and help clear cycles.
                self._abortcallback = None
                _playback(self.journal, self.report, self.opener, self._vfsmap,
                          self.entries, self._backupentries, False,
                          checkambigfiles=self.checkambigfiles)
                self.report(_("rollback completed\n"))
            except BaseException:
                self.report(_("rollback failed - please run hg recover\n"))
        finally:
            self.journal = None
            self.releasefn(self, False) # notify failure of transaction
            self.releasefn = None # Help prevent cycles.

def rollback(opener, vfsmap, file, report, checkambigfiles=None):
    """Rolls back the transaction contained in the given file

39676:77c4e2ae9f07

        vfsmap[''] = opener  # set default value
        self._vfsmap = vfsmap
        self.after = after
        self.entries = []
        self.map = {}
        self._journal = journalname
        self._undoname = undoname
        self._queue = []
        # A callback to validate transaction content before closing it.
        # should raise exception is anything is wrong.
        # target user is repository hooks.

        # transaction.
        self.changes = {}

        # a dict of arguments to be passed to hooks
        self.hookargs = {}
        self.file = opener.open(self._journal, "w")

        # a list of ('location', 'path', 'backuppath', cache) entries.
        # - if 'backuppath' is empty, no file existed at backup time
        # - if 'path' is empty, this is a temporary transaction file
        # - if 'location' is not empty, the path is outside main opener reach.
        #   use 'location' value as a key in a vfsmap to find the right 'vfs'
        # (cache is currently unused)
        self._backupentries = []
        self._backupmap = {}
        self._backupjournal = "%s.backupfiles" % self._journal
        self._backupsfile = opener.open(self._backupjournal, 'w')
        self._backupsfile.write('%d\n' % version)

        if createmode is not None:
            opener.chmod(self._journal, createmode & 0o666)
            opener.chmod(self._backupjournal, createmode & 0o666)

        # hold file generations to be performed on commit
        self._filegenerators = {}
        # hold callback to write pending data for hooks

        name = r'/'.join(self.names)
        return (r'<transaction name=%s, count=%d, usages=%d>' %
                (name, self._count, self._usages))

    def __del__(self):
        if self._journal:
            self._abort()

    @active
    def startgroup(self):
        """delay registration of file entry

        if file in self.map or file in self._backupmap:
            return
        vfs = self._vfsmap[location]
        dirname, filename = vfs.split(file)
        backupfilename = "%s.backup.%s" % (self._journal, filename)
        backupfile = vfs.reljoin(dirname, backupfilename)
        if vfs.exists(file):
            filepath = vfs.join(file)
            backuppath = vfs.join(backupfile)
            util.copyfile(filepath, backuppath, hardlink=hardlink)

        if self.after:
            self.after()
            self.after = None # Help prevent cycles.
        if self.opener.isfile(self._backupjournal):
            self.opener.unlink(self._backupjournal)
        if self.opener.isfile(self._journal):
            self.opener.unlink(self._journal)
        for l, _f, b, c in self._backupentries:
            if l not in self._vfsmap and c:
                self.report("couldn't remove %s: unknown cache location"
                            "%s\n" % (b, l))
                continue

                        raise
                    # Abort may be raise by read only opener
                    self.report("couldn't remove %s: %s\n"
                                % (vfs.join(b), inst))
        self._backupentries = []
        self._journal = None

        self.releasefn(self, True) # notify success of closing transaction
        self.releasefn = None # Help prevent cycles.

        # run post close action

                    self.report("couldn't remove %s: unknown cache location"
                                "%s\n" % (b, l))
                    continue
                vfs = self._vfsmap[l]
                base, name = vfs.split(b)
                assert name.startswith(self._journal), name
                uname = name.replace(self._journal, self._undoname, 1)
                u = vfs.reljoin(base, uname)
                util.copyfile(vfs.join(b), vfs.join(u), hardlink=True)
            undobackupfile.write("%s\0%s\0%s\0%d\n" % (l, f, u, c))
        undobackupfile.close()

        try:
            if not self.entries and not self._backupentries:
                if self._backupjournal:
                    self.opener.unlink(self._backupjournal)
                if self._journal:
                    self.opener.unlink(self._journal)
                return

            self.report(_("transaction abort!\n"))

            try:
                for cat in sorted(self._abortcallback):
                    self._abortcallback[cat](self)
                # Prevent double usage and help clear cycles.
                self._abortcallback = None
                _playback(self._journal, self.report, self.opener, self._vfsmap,
                          self.entries, self._backupentries, False,
                          checkambigfiles=self.checkambigfiles)
                self.report(_("rollback completed\n"))
            except BaseException:
                self.report(_("rollback failed - please run hg recover\n"))
        finally:
            self._journal = None
            self.releasefn(self, False) # notify failure of transaction
            self.releasefn = None # Help prevent cycles.

def rollback(opener, vfsmap, file, report, checkambigfiles=None):
    """Rolls back the transaction contained in the given file