comparison mercurial/transaction.py @ 23204:10beda5bd2b7

transaction: allow registering a finalization callback

The new 'addfinalize' method allows people to register a callback to
be triggered when the transaction is closed. This aims to get rid of
explicit calls to 'changelog.finalize'. This also obsoletes the
'onclose' function but removing it is not in the scope of this series.

23203:3872d563e01a

        self._filegenerators = {}
        # hold callbalk to write pending data for hooks
        self._pendingcallback = {}
        # True is any pending data have been written ever
        self._anypending = False

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

            any = self._pendingcallback.pop(cat)()
            self._anypending = self._anypending or any
        return self._anypending

    @active
    def close(self):
        '''commit the transaction'''
        if self.count == 1 and self.onclose is not None:
            self._generatefiles()
            self.onclose()

        self.count -= 1
        if self.count != 0:
            return

23204:10beda5bd2b7

        self._filegenerators = {}
        # hold callbalk to write pending data for hooks
        self._pendingcallback = {}
        # True is any pending data have been written ever
        self._anypending = False
        # holds callback to call when writing the transaction
        self._finalizecallback = {}

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

            any = self._pendingcallback.pop(cat)()
            self._anypending = self._anypending or any
        return self._anypending

    @active
    def addfinalize(self, category, callback):
        """add a callback to be called when the transaction is closed

        Category is a unique identifier to allow overwriting old callbacks with
        newer callbacks.
        """
        self._finalizecallback[category] = callback

    @active
    def close(self):
        '''commit the transaction'''
        if self.count == 1 and self.onclose is not None:
            self._generatefiles()
            categories = sorted(self._finalizecallback)
            for cat in categories:
                self._finalizecallback[cat]()
            self.onclose()

        self.count -= 1
        if self.count != 0:
            return