changeset 39881:d63153611ed5

storageutil: extract revision number iteration This code is a bit quirky (and possibly buggy). It will likely be used by multiple storage backends. Let's extract it so it is reusable. Differential Revision: https://phab.mercurial-scm.org/D4757
author Gregory Szorc <gregory.szorc@gmail.com>
date Mon, 24 Sep 2018 15:19:52 -0700
parents 1b65fb4d43d6
children c841e8855cd3
files mercurial/revlog.py mercurial/utils/storageutil.py
diffstat 2 files changed, 19 insertions(+), 11 deletions(-) [+]
line wrap: on
line diff
--- a/mercurial/revlog.py	Mon Sep 24 14:54:28 2018 -0700
+++ b/mercurial/revlog.py	Mon Sep 24 15:19:52 2018 -0700
@@ -515,17 +515,7 @@
         return iter(pycompat.xrange(len(self)))
     def revs(self, start=0, stop=None):
         """iterate over all rev in this revlog (from start to stop)"""
-        step = 1
-        length = len(self)
-        if stop is not None:
-            if start > stop:
-                step = -1
-            stop += step
-            if stop > length:
-                stop = length
-        else:
-            stop = length
-        return pycompat.xrange(start, stop, step)
+        return storageutil.iterrevs(len(self), start=start, stop=stop)
 
     @util.propertycache
     def nodemap(self):
--- a/mercurial/utils/storageutil.py	Mon Sep 24 14:54:28 2018 -0700
+++ b/mercurial/utils/storageutil.py	Mon Sep 24 15:19:52 2018 -0700
@@ -13,6 +13,9 @@
 from ..node import (
     nullid,
 )
+from .. import (
+    pycompat,
+)
 
 _nullhash = hashlib.sha1(nullid)
 
@@ -81,3 +84,18 @@
 
     offset = text.index(b'\x01\n', 2)
     return text[offset + 2:]
+
+def iterrevs(storelen, start=0, stop=None):
+    """Iterate over revision numbers in a store."""
+    step = 1
+
+    if stop is not None:
+        if start > stop:
+            step = -1
+        stop += step
+        if stop > storelen:
+            stop = storelen
+    else:
+        stop = storelen
+
+    return pycompat.xrange(start, stop, step)