graft: use revset for intersecting with ancestor set
authorMartin von Zweigbergk <martinvonz@google.com>
Fri, 27 Dec 2019 21:11:36 -0800
changeset 44046 4322de8f7016
parent 44045 df82c06e1a05
child 44047 de5d34ca01bd
graft: use revset for intersecting with ancestor set This addresses a TODO added in a1381eea7c7d (graft: do not use `.remove` on a smart set (regression), 2014-04-28). Differential Revision: https://phab.mercurial-scm.org/D7806
mercurial/commands.py
--- a/mercurial/commands.py	Fri Dec 27 21:11:33 2019 -0800
+++ b/mercurial/commands.py	Fri Dec 27 21:11:36 2019 -0800
@@ -3079,14 +3079,10 @@
     # already, they'd have been in the graftstate.
     if not (cont or opts.get(b'force')) and basectx is None:
         # check for ancestors of dest branch
-        crev = repo[b'.'].rev()
-        ancestors = repo.changelog.ancestors([crev], inclusive=True)
-        # XXX make this lazy in the future
-        for rev in revs:
-            if rev in ancestors:
-                ui.warn(
-                    _(b'skipping ancestor revision %d:%s\n') % (rev, repo[rev])
-                )
+        ancestors = repo.revs(b'%ld & (::.)', revs)
+        for rev in ancestors:
+            ui.warn(_(b'skipping ancestor revision %d:%s\n') % (rev, repo[rev]))
+
         revs = [r for r in revs if r not in ancestors]
 
         if not revs:
@@ -3105,7 +3101,7 @@
 
         # The only changesets we can be sure doesn't contain grafts of any
         # revs, are the ones that are common ancestors of *all* revs:
-        for rev in repo.revs(b'only(%d,ancestor(%ld))', crev, revs):
+        for rev in repo.revs(b'only(%d,ancestor(%ld))', repo[b'.'].rev(), revs):
             ctx = repo[rev]
             n = ctx.extra().get(b'source')
             if n in ids: