util.fspath: use a dict rather than a linear scan for lookups
Previously, we'd scan through the entire directory listing looking for a
normalized match. This is O(N) in the number of files in the directory. If we
decide to call util.fspath on each file in it, the overall complexity works out
to O(N^2). This becomes a problem with directories a few thousand files or
larger.
Switch to using a dictionary instead. There is a slightly higher upfront cost
to pay, but for cases like the above this is amortized O(1). Plus there is a
lower constant factor because generator comprehensions are faster than for
loops, so overall it works out to be a very small loss in performance for 1
file, and a huge gain when there's more.
For a large repo with around 200k files in it on a case-insensitive file
system, for a large directory with over 30,000 files in it, the following
command was tested:
ls | shuf -n $COUNT | xargs hg status
This command leads to util.fspath being called on $COUNT files in the
directory.
COUNT before after
1 0.77s 0.78s
100 1.42s 0.80s
1000 6.3s 0.96s
I also tested with COUNT=10000, but before took too long so I gave up.
--- a/mercurial/util.py Mon Oct 27 16:53:01 2014 -0500
+++ b/mercurial/util.py Fri Oct 24 11:39:39 2014 -0700
@@ -899,11 +899,8 @@
The root should be normcase-ed, too.
'''
- def find(p, contents):
- for n in contents:
- if normcase(n) == p:
- return n
- return None
+ def _makefspathcacheentry(dir):
+ return dict((normcase(n), n) for n in os.listdir(dir))
seps = os.sep
if os.altsep:
@@ -919,16 +916,15 @@
continue
if dir not in _fspathcache:
- _fspathcache[dir] = os.listdir(dir)
+ _fspathcache[dir] = _makefspathcacheentry(dir)
contents = _fspathcache[dir]
- found = find(part, contents)
+ found = contents.get(part)
if not found:
# retry "once per directory" per "dirstate.walk" which
# may take place for each patches of "hg qpush", for example
- contents = os.listdir(dir)
- _fspathcache[dir] = contents
- found = find(part, contents)
+ _fspathcache[dir] = contents = _makefspathcacheentry(dir)
+ found = contents.get(part)
result.append(found or part)
dir = os.path.join(dir, part)