comparison mercurial/bundle2.py @ 34325:4fbbdd9b04f1

phases: use a Struct object for binary encoding and decoding

We will move the binary encoding and decoding code to 'phases.py' in order to
make it easier to reuse. First, let's cleanup it a bit.

34324:e45ec589f962

_fparttypesize = '>B'
_fpartid = '>I'
_fpayloadsize = '>i'
_fpartparamcount = '>BB'

_fphasesentry = '>i20s'

preferedchunksize = 4096

_parttypeforbidden = re.compile('[^a-zA-Z0-9_:-]')

    if opts.get('phases', False):
        headsbyphase = phases.subsetphaseheads(repo, outgoing.missing)
        phasedata = []
        for phase in phases.allphases:
            for head in headsbyphase[phase]:
                phasedata.append(_pack(_fphasesentry, phase, head))
        bundler.newpart('phase-heads', data=''.join(phasedata))

def addparttagsfnodescache(repo, bundler, outgoing):
    # we include the tags fnode cache for the bundle changeset
    # (as an optional parts)

                kwargs[key] = inpart.params[key]
        raise error.PushkeyFailed(partid=str(inpart.id), **kwargs)

def _readphaseheads(inpart):
    headsbyphase = [[] for i in phases.allphases]
    entrysize = struct.calcsize(_fphasesentry)
    while True:
        entry = inpart.read(entrysize)
        if len(entry) < entrysize:
            if entry:
                raise error.Abort(_('bad phase-heads bundle part'))
            break
        phase, node = struct.unpack(_fphasesentry, entry)
        headsbyphase[phase].append(node)
    return headsbyphase

@parthandler('phase-heads')
def handlephases(op, inpart):

34325:4fbbdd9b04f1

_fparttypesize = '>B'
_fpartid = '>I'
_fpayloadsize = '>i'
_fpartparamcount = '>BB'

_fphasesentry = struct.Struct('>i20s')

preferedchunksize = 4096

_parttypeforbidden = re.compile('[^a-zA-Z0-9_:-]')

    if opts.get('phases', False):
        headsbyphase = phases.subsetphaseheads(repo, outgoing.missing)
        phasedata = []
        for phase in phases.allphases:
            for head in headsbyphase[phase]:
                phasedata.append(_fphasesentry.pack(phase, head))
        bundler.newpart('phase-heads', data=''.join(phasedata))

def addparttagsfnodescache(repo, bundler, outgoing):
    # we include the tags fnode cache for the bundle changeset
    # (as an optional parts)

                kwargs[key] = inpart.params[key]
        raise error.PushkeyFailed(partid=str(inpart.id), **kwargs)

def _readphaseheads(inpart):
    headsbyphase = [[] for i in phases.allphases]
    entrysize = _fphasesentry.size
    while True:
        entry = inpart.read(entrysize)
        if len(entry) < entrysize:
            if entry:
                raise error.Abort(_('bad phase-heads bundle part'))
            break
        phase, node = _fphasesentry.unpack(entry)
        headsbyphase[phase].append(node)
    return headsbyphase

@parthandler('phase-heads')
def handlephases(op, inpart):