comparison mercurial/localrepo.py @ 47078:223b47235d1c

sidedata: enable sidedata computers to optionally rewrite flags

Sidedata computers may want to influence the flags of the revision they touch.
For example, the computer for changelog-based copytracing can add a flag to
signify that this revision might affect copytracing, inversely removing said
flag if the information is no longer applicable.

See inline documentation in `storageutil` for more details.

Differential Revision: https://phab.mercurial-scm.org/D10344

47077:119790e1c67c

        if requirementsmod.REVLOGV2_REQUIREMENT not in self.requirements:
            # Only revlogv2 repos can want sidedata.
            return
        self._wanted_sidedata.add(pycompat.bytestr(category))

    def register_sidedata_computer(self, kind, category, keys, computer):
        if kind not in revlogconst.ALL_KINDS:
            msg = _(b"unexpected revlog kind %r.")
            raise error.ProgrammingError(msg % kind)
        category = pycompat.bytestr(category)
        if category in self._sidedata_computers.get(kind, []):
            msg = _(
                b"cannot register a sidedata computer twice for category '%s'."
            )
            raise error.ProgrammingError(msg % category)
        self._sidedata_computers.setdefault(kind, {})
        self._sidedata_computers[kind][category] = (keys, computer)


# used to avoid circular references so destructors work
def aftertrans(files):
    renamefiles = [tuple(t) for t in files]

47078:223b47235d1c

        if requirementsmod.REVLOGV2_REQUIREMENT not in self.requirements:
            # Only revlogv2 repos can want sidedata.
            return
        self._wanted_sidedata.add(pycompat.bytestr(category))

    def register_sidedata_computer(self, kind, category, keys, computer, flags):
        if kind not in revlogconst.ALL_KINDS:
            msg = _(b"unexpected revlog kind '%s'.")
            raise error.ProgrammingError(msg % kind)
        category = pycompat.bytestr(category)
        if category in self._sidedata_computers.get(kind, []):
            msg = _(
                b"cannot register a sidedata computer twice for category '%s'."
            )
            raise error.ProgrammingError(msg % category)
        self._sidedata_computers.setdefault(kind, {})
        self._sidedata_computers[kind][category] = (keys, computer, flags)


# used to avoid circular references so destructors work
def aftertrans(files):
    renamefiles = [tuple(t) for t in files]