Patchwork D3672: retractboundary: add dryrun parameter

login
register
mail settings
Submitter phabricator
Date June 1, 2018, 6:59 a.m.
Message ID <bdd9a8011518ec7aea83f41e84a4d1f9@localhost.localdomain>
Download mbox | patch
Permalink /patch/31928/
State Not Applicable
Headers show

Comments

phabricator - June 1, 2018, 6:59 a.m.
khanchi97 updated this revision to Diff 8948.
khanchi97 edited the summary of this revision.

REPOSITORY
  rHG Mercurial

CHANGES SINCE LAST UPDATE
  https://phab.mercurial-scm.org/D3672?vs=8929&id=8948

REVISION DETAIL
  https://phab.mercurial-scm.org/D3672

AFFECTED FILES
  mercurial/phases.py

CHANGE DETAILS




To: khanchi97, #hg-reviewers
Cc: mercurial-devel

Patch

diff --git a/mercurial/phases.py b/mercurial/phases.py
--- a/mercurial/phases.py
+++ b/mercurial/phases.py
@@ -392,32 +392,42 @@ 
             self._retractboundary(repo, tr, targetphase, delroots)
         repo.invalidatevolatilesets()
 
-    def retractboundary(self, repo, tr, targetphase, nodes):
+    def retractboundary(self, repo, tr, targetphase, nodes, dryrun=None):
         oldroots = self.phaseroots[:targetphase + 1]
         if tr is None:
             phasetracking = None
         else:
             phasetracking = tr.changes.get('phases')
         repo = repo.unfiltered()
-        if (self._retractboundary(repo, tr, targetphase, nodes)
-            and phasetracking is not None):
-
-            # find the affected revisions
-            new = self.phaseroots[targetphase]
-            old = oldroots[targetphase]
-            affected = set(repo.revs('(%ln::) - (%ln::)', new, old))
+        changes = set() # set or revisions whose phase can be changed.
+        if dryrun:
+            getphase = repo._phasecache.phase
+            nds = [n for n in nodes
+                   if getphase(repo, repo[n].rev()) < targetphase]
+            targetphroots = oldroots[-1]
+            affected = set(repo.revs('(%ln::) - (%ln::)', nds, targetphroots))
+            changes.update(affected)
+        else:
+            if (self._retractboundary(repo, tr, targetphase, nodes)
+                and phasetracking is not None):
 
-            # find the phase of the affected revision
-            for phase in xrange(targetphase, -1, -1):
-                if phase:
-                    roots = oldroots[phase]
-                    revs = set(repo.revs('%ln::%ld', roots, affected))
-                    affected -= revs
-                else: # public phase
-                    revs = affected
-                for r in revs:
-                    _trackphasechange(phasetracking, r, phase, targetphase)
-        repo.invalidatevolatilesets()
+                # find the affected revisions
+                new = self.phaseroots[targetphase]
+                old = oldroots[targetphase]
+                affected = set(repo.revs('(%ln::) - (%ln::)', new, old))
+
+                # find the phase of the affected revision
+                for phase in xrange(targetphase, -1, -1):
+                    if phase:
+                        roots = oldroots[phase]
+                        revs = set(repo.revs('%ln::%ld', roots, affected))
+                        affected -= revs
+                    else: # public phase
+                        revs = affected
+                    for r in revs:
+                        _trackphasechange(phasetracking, r, phase, targetphase)
+            repo.invalidatevolatilesets()
+        return changes
 
     def _retractboundary(self, repo, tr, targetphase, nodes):
         # Be careful to preserve shallow-copied values: do not update
@@ -489,17 +499,24 @@ 
     phcache.advanceboundary(repo, tr, targetphase, nodes)
     repo._phasecache.replace(phcache)
 
-def retractboundary(repo, tr, targetphase, nodes):
+def retractboundary(repo, tr, targetphase, nodes, dryrun=None):
     """Set nodes back to a phase changing other nodes phases if
     necessary.
 
     This function move boundary *backward* this means that all nodes
     are set in the target phase or kept in a *higher* phase.
 
-    Simplify boundary to contains phase roots only."""
+    Simplify boundary to contains phase roots only.
+
+    If dryrun is true then it will not perform any action and only returns
+    set of revision whose phase can be changed.
+    """
     phcache = repo._phasecache.copy()
-    phcache.retractboundary(repo, tr, targetphase, nodes)
-    repo._phasecache.replace(phcache)
+    changes = phcache.retractboundary(repo, tr, targetphase, nodes,
+                                      dryrun=dryrun)
+    if not dryrun:
+        repo._phasecache.replace(phcache)
+    return changes
 
 def registernew(repo, tr, targetphase, nodes):
     """register a new revision and its phase