comparison mercurial/wireprotov2server.py @ 37728:564a3eec6e63

wireprotov2: add support for more response types

This adds types to represent error and generator responses from
server commands.

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

37727:5cdde6158426

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

    if action == 'sendframes':

37728:564a3eec6e63

    if isinstance(rsp, wireprototypes.cborresponse):
        encoded = cbor.dumps(rsp.value, canonical=True)
        action, meta = reactor.oncommandresponseready(outstream,
                                                      command['requestid'],
                                                      encoded)
    elif isinstance(rsp, wireprototypes.v2streamingresponse):
        action, meta = reactor.oncommandresponsereadygen(outstream,
                                                         command['requestid'],
                                                         rsp.gen)
    elif isinstance(rsp, wireprototypes.v2errorresponse):
        action, meta = reactor.oncommanderror(outstream,
                                              command['requestid'],
                                              rsp.message,
                                              rsp.args)
    else:
        action, meta = reactor.onservererror(
            _('unhandled response type from wire proto command'))

    if action == 'sendframes':