phabricator: rework phabread to reduce memory usage and round-trips
authorJun Wu <quark@fb.com>
Tue, 04 Jul 2017 16:36:48 -0700
changeset 33267 dba9f88659a3
parent 33266 5b2391b46906
child 33268 85391b95961d
phabricator: rework phabread to reduce memory usage and round-trips This patch reworked phabread a bit so it fetches the lightweight "Differential Revision" metadata for a stack first. Then read other data. This allows the code to: a) send 1 request to get all `hg:meta` data instead of N requests b) patches are read in desired order, no need to buffer the output "b)" reduces the memory usage from O(N^2) to O(N) since we no longer keep old patch contents in memory. The above `N` means the number of patches in the stack.
contrib/phabricator.py
--- a/contrib/phabricator.py	Tue Jul 04 16:36:48 2017 -0700
+++ b/contrib/phabricator.py	Tue Jul 04 16:36:48 2017 -0700
@@ -314,50 +314,106 @@
 _metanamemap = util.sortdict([(r'user', 'User'), (r'date', 'Date'),
                               (r'node', 'Node ID'), (r'parent', 'Parent ')])
 
-def readpatch(repo, params, recursive=False):
+def querydrev(repo, params, stack=False):
+    """return a list of "Differential Revision" dicts
+
+    params is the input of "differential.query" API, and is expected to match
+    just a single Differential Revision.
+
+    A "Differential Revision dict" looks like:
+
+        {
+            "id": "2",
+            "phid": "PHID-DREV-672qvysjcczopag46qty",
+            "title": "example",
+            "uri": "https://phab.example.com/D2",
+            "dateCreated": "1499181406",
+            "dateModified": "1499182103",
+            "authorPHID": "PHID-USER-tv3ohwc4v4jeu34otlye",
+            "status": "0",
+            "statusName": "Needs Review",
+            "properties": [],
+            "branch": null,
+            "summary": "",
+            "testPlan": "",
+            "lineCount": "2",
+            "activeDiffPHID": "PHID-DIFF-xoqnjkobbm6k4dk6hi72",
+            "diffs": [
+              "3",
+              "4",
+            ],
+            "commits": [],
+            "reviewers": [],
+            "ccs": [],
+            "hashes": [],
+            "auxiliary": {
+              "phabricator:projects": [],
+              "phabricator:depends-on": [
+                "PHID-DREV-gbapp366kutjebt7agcd"
+              ]
+            },
+            "repositoryPHID": "PHID-REPO-hub2hx62ieuqeheznasv",
+            "sourcePath": null
+        }
+
+    If stack is True, return a list of "Differential Revision dict"s in an
+    order that the latter ones depend on the former ones. Otherwise, return a
+    list of a unique "Differential Revision dict".
+    """
+    result = []
+    queue = [params]
+    while queue:
+        params = queue.pop()
+        drevs = callconduit(repo, 'differential.query', params)
+        if len(drevs) != 1:
+            raise error.Abort(_('cannot get Differential Revision %r') % params)
+        drev = drevs[0]
+        result.append(drev)
+        if stack:
+            auxiliary = drev.get(r'auxiliary', {})
+            depends = auxiliary.get(r'phabricator:depends-on', [])
+            for phid in depends:
+                queue.append({'phids': [phid]})
+    result.reverse()
+    return result
+
+def readpatch(repo, params, write, stack=False):
     """generate plain-text patch readable by 'hg import'
 
-    params is passed to "differential.query". If recursive is True, also return
-    dependent patches.
+    write is usually ui.write. params is passed to "differential.query". If
+    stack is True, also write dependent patches.
     """
     # Differential Revisions
-    drevs = callconduit(repo, 'differential.query', params)
-    if len(drevs) == 1:
-        drev = drevs[0]
-    else:
-        raise error.Abort(_('cannot get Differential Revision %r') % params)
-
-    repo.ui.note(_('reading D%s\n') % drev[r'id'])
+    drevs = querydrev(repo, params, stack)
 
-    diffid = max(int(v) for v in drev[r'diffs'])
-    body = callconduit(repo, 'differential.getrawdiff', {'diffID': diffid})
-    desc = callconduit(repo, 'differential.getcommitmessage',
-                       {'revision_id': drev[r'id']})
-    header = '# HG changeset patch\n'
+    # Prefetch hg:meta property for all diffs
+    diffids = sorted(set(max(int(v) for v in drev[r'diffs']) for drev in drevs))
+    diffs = callconduit(repo, 'differential.querydiffs', {'ids': diffids})
 
-    # Remove potential empty "Summary:"
-    desc = _summaryre.sub('', desc)
+    # Generate patch for each drev
+    for drev in drevs:
+        repo.ui.note(_('reading D%s\n') % drev[r'id'])
 
-    # Try to preserve metadata from hg:meta property. Write hg patch headers
-    # that can be read by the "import" command. See patchheadermap and extract
-    # in mercurial/patch.py for supported headers.
-    diffs = callconduit(repo, 'differential.querydiffs', {'ids': [diffid]})
-    props = diffs[str(diffid)][r'properties'] # could be empty list or dict
-    if props and r'hg:meta' in props:
-        meta = props[r'hg:meta']
-        for k in _metanamemap.keys():
-            if k in meta:
-                header += '# %s %s\n' % (_metanamemap[k], meta[k])
+        diffid = max(int(v) for v in drev[r'diffs'])
+        body = callconduit(repo, 'differential.getrawdiff', {'diffID': diffid})
+        desc = callconduit(repo, 'differential.getcommitmessage',
+                           {'revision_id': drev[r'id']})
+        header = '# HG changeset patch\n'
+
+        # Remove potential empty "Summary:"
+        desc = _summaryre.sub('', desc)
 
-    patch = ('%s%s\n%s') % (header, desc, body)
+        # Try to preserve metadata from hg:meta property. Write hg patch
+        # headers that can be read by the "import" command. See patchheadermap
+        # and extract in mercurial/patch.py for supported headers.
+        props = diffs[str(diffid)][r'properties'] # could be empty list or dict
+        if props and r'hg:meta' in props:
+            meta = props[r'hg:meta']
+            for k in _metanamemap.keys():
+                if k in meta:
+                    header += '# %s %s\n' % (_metanamemap[k], meta[k])
 
-    # Check dependencies
-    if recursive:
-        auxiliary = drev.get(r'auxiliary', {})
-        depends = auxiliary.get(r'phabricator:depends-on', [])
-        for phid in depends:
-            patch = readpatch(repo, {'phids': [phid]}, recursive=True) + patch
-    return patch
+        write(('%s%s\n%s') % (header, desc, body))
 
 @command('phabread',
          [('', 'stack', False, _('read dependencies'))],
@@ -374,5 +430,4 @@
         revid = int(revid.split('/')[-1].replace('D', ''))
     except ValueError:
         raise error.Abort(_('invalid Revision ID: %s') % revid)
-    patch = readpatch(repo, {'ids': [revid]}, recursive=opts.get('stack'))
-    ui.write(patch)
+    readpatch(repo, {'ids': [revid]}, ui.write, opts.get('stack'))