comparison mercurial/wireprotov2server.py @ 39815:d3d333ab167a

wireprotov2: teach changesetdata to fetch ancestors until depth

For shallow clone, it is useful to specify a starting node and tell
the server to send up to N ancestors from that starting point. This
enables the server to perform the DAG walk without the client having
to discover the base/stop node(s) first.

This commit implements support for said queries on the changesetdata
command.

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

39814:d059cb669632

        },
        'nodes': {
            'type': 'list',
            'example': [b'0123456...'],
        },
        'fields': {
            'type': 'set',
            'default': set,
            'example': {b'parents', b'revision'},
            'validvalues': {b'bookmarks', b'parents', b'phase', b'revision'},
        },
    },
    permission='pull')
def changesetdata(repo, proto, noderange, nodes, fields):
    # TODO look for unknown fields and abort when they can't be serviced.
    # This could probably be validated by dispatcher using validvalues.

    if noderange is None and nodes is None:
        raise error.WireprotoCommandError(
            'noderange or nodes must be defined')

    if noderange is not None:
        if len(noderange) != 2:
            raise error.WireprotoCommandError(
                'noderange must consist of 2 elements')

    seen = set()
    outgoing = []

    if nodes is not None:
        outgoing.extend(n for n in nodes if hasnode(n))
        seen |= set(outgoing)

    if noderange is not None:
        if noderange[0]:
            common = [n for n in noderange[0] if hasnode(n)]

39815:d3d333ab167a

        },
        'nodes': {
            'type': 'list',
            'example': [b'0123456...'],
        },
        'nodesdepth': {
            'type': 'int',
            'example': 10,
        },
        'fields': {
            'type': 'set',
            'default': set,
            'example': {b'parents', b'revision'},
            'validvalues': {b'bookmarks', b'parents', b'phase', b'revision'},
        },
    },
    permission='pull')
def changesetdata(repo, proto, noderange, nodes, nodesdepth, fields):
    # TODO look for unknown fields and abort when they can't be serviced.
    # This could probably be validated by dispatcher using validvalues.

    if noderange is None and nodes is None:
        raise error.WireprotoCommandError(
            'noderange or nodes must be defined')

    if nodesdepth is not None and nodes is None:
        raise error.WireprotoCommandError(
            'nodesdepth requires the nodes argument')

    if noderange is not None:
        if len(noderange) != 2:
            raise error.WireprotoCommandError(
                'noderange must consist of 2 elements')

    seen = set()
    outgoing = []

    if nodes is not None:
        outgoing = [n for n in nodes if hasnode(n)]

        if nodesdepth:
            outgoing = [cl.node(r) for r in
                        repo.revs(b'ancestors(%ln, %d)', outgoing,
                                  nodesdepth - 1)]

        seen |= set(outgoing)

    if noderange is not None:
        if noderange[0]:
            common = [n for n in noderange[0] if hasnode(n)]