# HG changeset patch # User Gregory Szorc # Date 2018-04-13 18:10:59 # Node ID 0ed11f9368fd1e3dcdbd247fbb1abea5d1661a5b # Parent 33a6eee08db25019d1ea4e0dd00645413a4cd693 treediscovery: switch to command executor interface We now have a new interface for requesting that commands run. Switch to it. Differential Revision: https://phab.mercurial-scm.org/D3272 diff --git a/mercurial/treediscovery.py b/mercurial/treediscovery.py --- a/mercurial/treediscovery.py +++ b/mercurial/treediscovery.py @@ -36,7 +36,8 @@ def findcommonincoming(repo, remote, hea base = set() if not heads: - heads = remote.heads() + with remote.commandexecutor() as e: + heads = e.callcommand('heads', {}).result() if repo.changelog.tip() == nullid: base.add(nullid) @@ -65,7 +66,10 @@ def findcommonincoming(repo, remote, hea # a 'branch' here is a linear segment of history, with four parts: # head, root, first parent, second parent # (a branch always has two parents (or none) by definition) - unknown = collections.deque(remote.branches(unknown)) + with remote.commandexecutor() as e: + branches = e.callcommand('branches', {'nodes': unknown}).result() + + unknown = collections.deque(branches) while unknown: r = [] while unknown: @@ -107,7 +111,12 @@ def findcommonincoming(repo, remote, hea repo.ui.debug("request %d: %s\n" % (reqcnt, " ".join(map(short, r)))) for p in xrange(0, len(r), 10): - for b in remote.branches(r[p:p + 10]): + with remote.commandexecutor() as e: + branches = e.callcommand('branches', { + 'nodes': r[p:p + 10], + }).result() + + for b in branches: repo.ui.debug("received %s:%s\n" % (short(b[0]), short(b[1]))) unknown.append(b) @@ -117,7 +126,11 @@ def findcommonincoming(repo, remote, hea newsearch = [] reqcnt += 1 repo.ui.progress(_('searching'), reqcnt, unit=_('queries')) - for n, l in zip(search, remote.between(search)): + + with remote.commandexecutor() as e: + between = e.callcommand('between', {'pairs': search}).result() + + for n, l in zip(search, between): l.append(n[1]) p = n[0] f = 1