# HG changeset patch
# User Gregory Szorc <gregory.szorc@gmail.com>
# Date 1523643259 25200
# Node ID 65b86ee69383a8bb7fd16d2e9f83b0cba0e87dca
# Parent  1964d2d1f421d821e8d8f36ea5f98f8c46add78e
streamclone: use command executor for wire protocol commands

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

diff -r 1964d2d1f421 -r 65b86ee69383 mercurial/streamclone.py
--- a/mercurial/streamclone.py	Fri Apr 13 11:13:05 2018 -0700
+++ b/mercurial/streamclone.py	Fri Apr 13 11:14:19 2018 -0700
@@ -126,11 +126,18 @@
     # creation.
     rbranchmap = None
     if remote.capable('branchmap'):
-        rbranchmap = remote.branchmap()
+        with remote.commandexecutor() as e:
+            rbranchmap = e.callcommand('branchmap', {}).result()
 
     repo.ui.status(_('streaming all changes\n'))
 
-    fp = remote.stream_out()
+    with remote.commandexecutor() as e:
+        fp = e.callcommand('stream_out', {}).result()
+
+    # TODO strictly speaking, this code should all be inside the context
+    # manager because the context manager is supposed to ensure all wire state
+    # is flushed when exiting. But the legacy peers don't do this, so it
+    # doesn't matter.
     l = fp.readline()
     try:
         resp = int(l)