comparison mercurial/upgrade.py @ 43134:75ad8af9c95e

upgrade: allow upgrade to repository using sidedata

Repository can now be migrated to support sidedata. More requirements and
migration will be needed to actual side-data usage. This is a step in that
direction.

To test the feature, we leverage the test extension. It make sure the `update`
part of the side-data migration actually works.

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

43133:3de4d13f22be

# list of requirements that request a clone of all revlog if added/removed
RECLONES_REQUIREMENTS = {
    b'generaldelta',
    localrepo.SPARSEREVLOG_REQUIREMENT,
}


def requiredsourcerequirements(repo):
    """Obtain requirements required to be present to upgrade a repo.

        b'fncache',
        b'generaldelta',
        b'revlogv1',
        b'store',
        localrepo.SPARSEREVLOG_REQUIREMENT,
    }
    for name in compression.compengines:
        engine = compression.compengines[name]
        if engine.available() and engine.revlogheader():
            supported.add(b'exp-compression-%s' % name)

    supported = {
        b'dotencode',
        b'fncache',
        b'generaldelta',
        localrepo.SPARSEREVLOG_REQUIREMENT,
    }
    for name in compression.compengines:
        engine = compression.compengines[name]
        if engine.available() and engine.revlogheader():
            supported.add(b'exp-compression-%s' % name)

UPGRADE_FILELOG = object()

UPGRADE_ALL_REVLOGS = frozenset(
    [UPGRADE_CHANGELOG, UPGRADE_MANIFEST, UPGRADE_FILELOG]
)


def matchrevlog(revlogfilter, entry):
    """check is a revlog is selected for cloning

    progress = None

    def oncopiedrevision(rl, rev, node):
        progress.increment()

    # Do the actual copying.
    # FUTURE this operation can be farmed off to worker processes.
    seen = set()
    for unencoded, encoded, size in alldatafiles:
        if unencoded.endswith(b'.d'):

                tr,
                newrl,
                addrevisioncb=oncopiedrevision,
                deltareuse=deltareuse,
                forcedeltabothparents=forcedeltabothparents,
            )
        else:
            msg = _(b'blindly copying %s containing %i revisions\n')
            ui.note(msg % (unencoded, len(oldrl)))
            _copyrevlog(tr, dstrepo, oldrl, unencoded)

43134:75ad8af9c95e

# list of requirements that request a clone of all revlog if added/removed
RECLONES_REQUIREMENTS = {
    b'generaldelta',
    localrepo.SPARSEREVLOG_REQUIREMENT,
    localrepo.SIDEDATA_REQUIREMENT,
}


def requiredsourcerequirements(repo):
    """Obtain requirements required to be present to upgrade a repo.

        b'fncache',
        b'generaldelta',
        b'revlogv1',
        b'store',
        localrepo.SPARSEREVLOG_REQUIREMENT,
        localrepo.SIDEDATA_REQUIREMENT,
    }
    for name in compression.compengines:
        engine = compression.compengines[name]
        if engine.available() and engine.revlogheader():
            supported.add(b'exp-compression-%s' % name)

    supported = {
        b'dotencode',
        b'fncache',
        b'generaldelta',
        localrepo.SPARSEREVLOG_REQUIREMENT,
        localrepo.SIDEDATA_REQUIREMENT,
    }
    for name in compression.compengines:
        engine = compression.compengines[name]
        if engine.available() and engine.revlogheader():
            supported.add(b'exp-compression-%s' % name)

UPGRADE_FILELOG = object()

UPGRADE_ALL_REVLOGS = frozenset(
    [UPGRADE_CHANGELOG, UPGRADE_MANIFEST, UPGRADE_FILELOG]
)


def getsidedatacompanion(srcrepo, destrepo):
    return None


def matchrevlog(revlogfilter, entry):
    """check is a revlog is selected for cloning

    progress = None

    def oncopiedrevision(rl, rev, node):
        progress.increment()

    sidedatacompanion = getsidedatacompanion(srcrepo, dstrepo)

    # Do the actual copying.
    # FUTURE this operation can be farmed off to worker processes.
    seen = set()
    for unencoded, encoded, size in alldatafiles:
        if unencoded.endswith(b'.d'):

                tr,
                newrl,
                addrevisioncb=oncopiedrevision,
                deltareuse=deltareuse,
                forcedeltabothparents=forcedeltabothparents,
                sidedatacompanion=sidedatacompanion,
            )
        else:
            msg = _(b'blindly copying %s containing %i revisions\n')
            ui.note(msg % (unencoded, len(oldrl)))
            _copyrevlog(tr, dstrepo, oldrl, unencoded)