comparison mercurial/branchmap.py @ 40375:76d4272bd57b

branchmap: avoid changelog and attribute lookups in replacecache()

This should make things faster. I'm not sure which operations would benefit
from it though. Maybe branchmap application on clone?

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

40374:47c03042cd1d

def replacecache(repo, bm):
    """Replace the branchmap cache for a repo with a branch mapping.

    This is likely only called during clone with a branch map from a remote.
    """
    rbheads = []
    closed = []
    for bheads in bm.itervalues():
        rbheads.extend(bheads)
        for h in bheads:
            r = repo.changelog.rev(h)
            b, c = repo.changelog.branchinfo(r)
            if c:
                closed.append(h)

    if rbheads:
        rtiprev = max((int(repo.changelog.rev(node))
                for node in rbheads))
        cache = branchcache(bm,
                            repo[rtiprev].node(),
                            rtiprev,
                            closednodes=closed)

40375:76d4272bd57b

def replacecache(repo, bm):
    """Replace the branchmap cache for a repo with a branch mapping.

    This is likely only called during clone with a branch map from a remote.
    """
    cl = repo.changelog
    clrev = cl.rev
    clbranchinfo = cl.branchinfo
    rbheads = []
    closed = []
    for bheads in bm.itervalues():
        rbheads.extend(bheads)
        for h in bheads:
            r = clrev(h)
            b, c = clbranchinfo(r)
            if c:
                closed.append(h)

    if rbheads:
        rtiprev = max((int(clrev(node))
                for node in rbheads))
        cache = branchcache(bm,
                            repo[rtiprev].node(),
                            rtiprev,
                            closednodes=closed)