comparison mercurial/wireprotoserver.py @ 37485:0b7475ea38cf

wireproto: port heads command to wire protocol v2

After much thought and consideration, wire protocol version 2's
commands will be defined in different functions from the existing
commands. This will make it easier to implement these commands
because it won't require shoehorning things like response formatting
and argument declaration into the same APIs.

For example, wire protocol version 1 requires that commands declare
a fixed and ordered list of argument names. It isn't really possible
to insert new arguments or have optional arguments without
breaking backwards compatibility. Wire protocol version 2, however,
uses CBOR maps for passing arguments. So arguments a) can be
optional b) can be added without BC c) can be strongly typed.

This commit starts our trek towards reimplementing the wire protocol
for version 2 with the heads command. It is pretty similar to the
existing heads command. One added feature is it can be told to
operate on only public phase changesets. This is useful for
making discovery faster when a repo has tens of thousands of
draft phase heads (such as Mozilla's "try" repository).

The HTTPv2 server-side protocol has had its `getargs()` implementation
updated to reflect that arguments are a map and not a list.

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

37484:c22fd3c4c23e

import struct
import sys
import threading

from .i18n import _
from .thirdparty.zope import (
    interface as zi,
)
from . import (
    encoding,

    if isinstance(rsp, wireprototypes.bytesresponse):
        action, meta = reactor.onbytesresponseready(outstream,
                                                    command['requestid'],
                                                    rsp.data)
    else:
        action, meta = reactor.onapplicationerror(
            _('unhandled response type from wire proto command'))

    if action == 'sendframes':

    def getargs(self, args):
        data = {}
        for k in args.split():
            if k == '*':
                raise NotImplementedError('do not support * args')
            else:
                data[k] = self._args[k]

        return [data[k] for k in args.split()]

    def getprotocaps(self):
        # Protocol capabilities are currently not implemented for HTTP V2.
        return set()

37485:0b7475ea38cf

import struct
import sys
import threading

from .i18n import _
from .thirdparty import (
    cbor,
)
from .thirdparty.zope import (
    interface as zi,
)
from . import (
    encoding,

    if isinstance(rsp, wireprototypes.bytesresponse):
        action, meta = reactor.onbytesresponseready(outstream,
                                                    command['requestid'],
                                                    rsp.data)
    elif isinstance(rsp, wireprototypes.cborresponse):
        encoded = cbor.dumps(rsp.value, canonical=True)
        action, meta = reactor.onbytesresponseready(outstream,
                                                    command['requestid'],
                                                    encoded,
                                                    iscbor=True)
    else:
        action, meta = reactor.onapplicationerror(
            _('unhandled response type from wire proto command'))

    if action == 'sendframes':

    def getargs(self, args):
        data = {}
        for k in args.split():
            if k == '*':
                raise NotImplementedError('do not support * args')
            elif k in self._args:
                data[k] = self._args[k]

        return data

    def getprotocaps(self):
        # Protocol capabilities are currently not implemented for HTTP V2.
        return set()