changeset 51415:1df8d84e7c99

phases: gather the logic for phasesets update in a single method This logic is duplicated around for no good reason, we gather it in a single place. The conditional is the new function are a bit weird as we about going to extend it soon.
author Pierre-Yves David <pierre-yves.david@octobus.net>
date Wed, 21 Feb 2024 11:59:28 +0100
parents 630d5283ee82
children 0c04074f5414
files mercurial/phases.py
diffstat 1 files changed, 14 insertions(+), 13 deletions(-) [+]
line wrap: on
line diff
--- a/mercurial/phases.py	Thu Feb 22 10:58:54 2024 +0100
+++ b/mercurial/phases.py	Wed Feb 21 11:59:28 2024 +0100
@@ -383,10 +383,7 @@
     def hasnonpublicphases(self, repo: "localrepo.localrepository") -> bool:
         """detect if there are revisions with non-public phase"""
         repo = repo.unfiltered()
-        cl = repo.changelog
-        if len(cl) > self._loadedrevslen:
-            self.invalidate()
-            self.loadphaserevs(repo)
+        self._ensure_phase_sets(repo)
         return any(
             revs for phase, revs in self._phaseroots.items() if phase != public
         )
@@ -400,10 +397,7 @@
         descendants of draft revisions, their roots will still be present.
         """
         repo = repo.unfiltered()
-        cl = repo.changelog
-        if len(cl) > self._loadedrevslen:
-            self.invalidate()
-            self.loadphaserevs(repo)
+        self._ensure_phase_sets(repo)
         return set().union(
             *[
                 revs
@@ -420,7 +414,7 @@
     ) -> Any:
         # TODO: finish typing this
         """return a smartset for the given phases"""
-        self.loadphaserevs(repo)  # ensure phase's sets are loaded
+        self._ensure_phase_sets(repo)  # ensure phase's sets are loaded
         phases = set(phases)
         publicphase = public in phases
 
@@ -511,14 +505,20 @@
                 self._phasesets[phase] = ps
         self._loadedrevslen = len(cl)
 
-    def loadphaserevs(self, repo: "localrepo.localrepository") -> None:
-        """ensure phase information is loaded in the object"""
+    def _ensure_phase_sets(self, repo: "localrepo.localrepository") -> None:
+        """ensure phase information is loaded in the object and up to date"""
+        update = False
         if self._phasesets is None:
+            update = True
+        elif len(repo.changelog) > self._loadedrevslen:
+            update = True
+        if update:
             try:
                 res = self._getphaserevsnative(repo)
                 self._loadedrevslen, self._phasesets = res
             except AttributeError:
                 self._computephaserevspure(repo)
+            assert self._loadedrevslen == len(repo.changelog)
 
     def invalidate(self):
         self._loadedrevslen = 0
@@ -534,9 +534,10 @@
             return public
         if rev < nullrev:
             raise ValueError(_(b'cannot lookup negative revision'))
+        # double check self._loadedrevslen to avoid an extra method call as
+        # python is slow for that.
         if rev >= self._loadedrevslen:
-            self.invalidate()
-            self.loadphaserevs(repo)
+            self._ensure_phase_sets(repo)
         for phase in trackedphases:
             if rev in self._phasesets[phase]:
                 return phase