comparison mercurial/changegroup.py @ 45789:09735cde6275

phases: allow registration and boundary advancement with revision sets

The core internals either use revision sets already or can trivially use
them. Use the new interface in cg1unpacker.apply to avoid materializing
the list of all new nodes as it is normally just a revision range. This
avoids about 67 Bytes / changeset on AMD64 in peak RSS.

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

45788:a5206e71c536

            self.callback = progress.increment

            efilesset = set()
            cgnodes = []

            def onchangelog(cl, node):
                efilesset.update(cl.readfiles(node))
                cgnodes.append(node)

            def ondupchangelog(cl, node):
                cgnodes.append(node)

            self.changelogheader()
            deltas = self.deltaiter()
            if not cl.addgroup(
                deltas,

                ml = repo.manifestlog
                # validate incoming csets have their manifests
                for cset in pycompat.xrange(clstart, clend):
                    mfnode = cl.changelogrevision(cset).manifest
                    mfest = ml[mfnode].readdelta()
                    # store file cgnodes we must see
                    for f, n in pycompat.iteritems(mfest):
                        needfiles.setdefault(f, set()).add(n)

            # process the files
            repo.ui.status(_(b"adding file changes\n"))

                    b'pretxnchangegroup',
                    throw=True,
                    **pycompat.strkwargs(hookargs)
                )

            added = [cl.node(r) for r in pycompat.xrange(clstart, clend)]
            phaseall = None
            if srctype in (b'push', b'serve'):
                # Old servers can not push the boundary themselves.
                # New servers won't push the boundary if changeset already
                # exists locally as secret

                    # outside, their phases are going to be pushed
                    # alongside. Therefor `targetphase` is
                    # ignored.
                    targetphase = phaseall = phases.draft
            if added:
                phases.registernew(repo, tr, targetphase, added)
            if phaseall is not None:
                phases.advanceboundary(repo, tr, phaseall, cgnodes)

            if changesets > 0:

                def runhooks(unused_success):
                    # These hooks run when the lock releases, not when the

                    if clstart >= len(repo):
                        return

                    repo.hook(b"changegroup", **pycompat.strkwargs(hookargs))

                    for n in added:
                        args = hookargs.copy()
                        args[b'node'] = hex(n)
                        del args[b'node_last']
                        repo.hook(b"incoming", **pycompat.strkwargs(args))

                    newheads = [h for h in repo.heads() if h not in oldheads]
                    repo.ui.log(

45789:09735cde6275

            self.callback = progress.increment

            efilesset = set()
            cgnodes = []

            def ondupchangelog(cl, node):
                if cl.rev(node) < clstart:
                    cgnodes.append(node)

            def onchangelog(cl, node):
                efilesset.update(cl.readfiles(node))

            self.changelogheader()
            deltas = self.deltaiter()
            if not cl.addgroup(
                deltas,

                ml = repo.manifestlog
                # validate incoming csets have their manifests
                for cset in pycompat.xrange(clstart, clend):
                    mfnode = cl.changelogrevision(cset).manifest
                    mfest = ml[mfnode].readdelta()
                    # store file nodes we must see
                    for f, n in pycompat.iteritems(mfest):
                        needfiles.setdefault(f, set()).add(n)

            # process the files
            repo.ui.status(_(b"adding file changes\n"))

                    b'pretxnchangegroup',
                    throw=True,
                    **pycompat.strkwargs(hookargs)
                )

            added = pycompat.xrange(clstart, clend)
            phaseall = None
            if srctype in (b'push', b'serve'):
                # Old servers can not push the boundary themselves.
                # New servers won't push the boundary if changeset already
                # exists locally as secret

                    # outside, their phases are going to be pushed
                    # alongside. Therefor `targetphase` is
                    # ignored.
                    targetphase = phaseall = phases.draft
            if added:
                phases.registernew(repo, tr, targetphase, [], revs=added)
            if phaseall is not None:
                phases.advanceboundary(repo, tr, phaseall, cgnodes, revs=added)
                cgnodes = []

            if changesets > 0:

                def runhooks(unused_success):
                    # These hooks run when the lock releases, not when the

                    if clstart >= len(repo):
                        return

                    repo.hook(b"changegroup", **pycompat.strkwargs(hookargs))

                    for rev in added:
                        args = hookargs.copy()
                        args[b'node'] = hex(cl.node(rev))
                        del args[b'node_last']
                        repo.hook(b"incoming", **pycompat.strkwargs(args))

                    newheads = [h for h in repo.heads() if h not in oldheads]
                    repo.ui.log(