comparison mercurial/bundle2.py @ 36963:9988fc10f49e

revbranchcache: add a bundle2 handler for a rbc part

We add the necessary bit to process a part containing rev-branch-cache
information. The local rev branch cache is then updated with the received
information.

Computing branch cache can become a significant part of the time spent pulling.

36962:95f4f1bfb650

from .i18n import _
from . import (
    bookmarks,
    changegroup,
    error,
    node as nodemod,
    obsolete,
    phases,
    pushkey,

        count += 1

    cache.write()
    op.ui.debug('applied %i hgtags fnodes cache entries\n' % count)

@parthandler('pushvars')
def bundle2getvars(op, part):
    '''unbundle a bundle2 containing shellvars on the server'''
    # An option to disable unbundling on server-side for security reasons
    if op.ui.configbool('push', 'pushvars.server'):

36963:9988fc10f49e

from .i18n import _
from . import (
    bookmarks,
    changegroup,
    encoding,
    error,
    node as nodemod,
    obsolete,
    phases,
    pushkey,

        count += 1

    cache.write()
    op.ui.debug('applied %i hgtags fnodes cache entries\n' % count)

rbcstruct = struct.Struct('>III')

@parthandler('cache:rev-branch-cache')
def handlerbc(op, inpart):
    """receive a rev-branch-cache payload and update the local cache

    The payload is a series of data related to each branch

    1) branch name length
    2) number of open heads
    3) number of closed heads
    4) open heads nodes
    5) closed heads nodes
    """
    total = 0
    rawheader = inpart.read(rbcstruct.size)
    cache = op.repo.revbranchcache()
    cl = op.repo.unfiltered().changelog
    while rawheader:
        header = rbcstruct.unpack(rawheader)
        total += header[1] + header[2]
        utf8branch = inpart.read(header[0])
        branch = encoding.tolocal(utf8branch)
        for x in xrange(header[1]):
            node = inpart.read(20)
            rev = cl.rev(node)
            cache.setdata(branch, rev, node, False)
        for x in xrange(header[2]):
            node = inpart.read(20)
            rev = cl.rev(node)
            cache.setdata(branch, rev, node, True)
        rawheader = inpart.read(rbcstruct.size)
    cache.write()

@parthandler('pushvars')
def bundle2getvars(op, part):
    '''unbundle a bundle2 containing shellvars on the server'''
    # An option to disable unbundling on server-side for security reasons
    if op.ui.configbool('push', 'pushvars.server'):