comparison mercurial/localrepo.py @ 37633:33a6eee08db2

wireproto: remove iterbatch() from peer interface (API)

Good riddance.

Some tests have been ported to the new API. This probably should
have been done in earlier commits. But duplicating the test coverage
would have been difficult. It was easier this way.

.. api::

   The wire protocol peer's ``iterbatch()`` for bulk executing commands
   has been remove.d Use ``peer.commandexecutor()`` instead.

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

37632:6c55ce51d6c3

    tags as tagsmod,
    transaction,
    txnutil,
    util,
    vfs as vfsmod,
    wireprotov1peer,
)
from .utils import (
    procutil,
    stringutil,
)

    return wrapper

moderncaps = {'lookup', 'branchmap', 'pushkey', 'known', 'getbundle',
              'unbundle'}
legacycaps = moderncaps.union({'changegroupsubset'})

class localiterbatcher(wireprotov1peer.iterbatcher):
    def __init__(self, local):
        super(localiterbatcher, self).__init__()
        self.local = local

    def submit(self):
        # submit for a local iter batcher is a noop
        pass

    def results(self):
        for name, args, opts, resref in self.calls:
            resref.set(getattr(self.local, name)(*args, **opts))
            yield resref.value

@zi.implementer(repository.ipeercommandexecutor)
class localcommandexecutor(object):
    def __init__(self, peer):
        self._peer = peer

    # Begin of peer interface.

    def commandexecutor(self):
        return localcommandexecutor(self)

    def iterbatch(self):
        return localiterbatcher(self)

    # End of peer interface.

class locallegacypeer(repository.legacypeer, localpeer):
    '''peer extension which implements legacy methods too; used for tests with

37633:33a6eee08db2

    tags as tagsmod,
    transaction,
    txnutil,
    util,
    vfs as vfsmod,
)
from .utils import (
    procutil,
    stringutil,
)

    return wrapper

moderncaps = {'lookup', 'branchmap', 'pushkey', 'known', 'getbundle',
              'unbundle'}
legacycaps = moderncaps.union({'changegroupsubset'})

@zi.implementer(repository.ipeercommandexecutor)
class localcommandexecutor(object):
    def __init__(self, peer):
        self._peer = peer

    # Begin of peer interface.

    def commandexecutor(self):
        return localcommandexecutor(self)

    # End of peer interface.

class locallegacypeer(repository.legacypeer, localpeer):
    '''peer extension which implements legacy methods too; used for tests with