perf.py
130 lines
| 3.7 KiB
| text/x-python
|
PythonLexer
/ contrib / perf.py
|
r7366 | # perf.py - performance test routines | ||
|
r7872 | from mercurial import cmdutil, match, commands | ||
|
r7366 | import time, os, sys | ||
def timer(func): | ||||
results = [] | ||||
begin = time.time() | ||||
count = 0 | ||||
while 1: | ||||
ostart = os.times() | ||||
cstart = time.time() | ||||
r = func() | ||||
cstop = time.time() | ||||
ostop = os.times() | ||||
count += 1 | ||||
a, b = ostart, ostop | ||||
results.append((cstop - cstart, b[0] - a[0], b[1]-a[1])) | ||||
if cstop - begin > 3 and count >= 100: | ||||
break | ||||
if cstop - begin > 10 and count >= 3: | ||||
break | ||||
if r: | ||||
sys.stderr.write("! result: %s\n" % r) | ||||
m = min(results) | ||||
sys.stderr.write("! wall %f comb %f user %f sys %f (best of %d)\n" | ||||
% (m[0], m[1] + m[2], m[1], m[2], count)) | ||||
def perfwalk(ui, repo, *pats): | ||||
try: | ||||
m = cmdutil.match(repo, pats, {}) | ||||
timer(lambda: len(list(repo.dirstate.walk(m, True, False)))) | ||||
except: | ||||
try: | ||||
m = cmdutil.match(repo, pats, {}) | ||||
timer(lambda: len([b for a,b,c in repo.dirstate.statwalk([], m)])) | ||||
except: | ||||
timer(lambda: len(list(cmdutil.walk(repo, pats, {})))) | ||||
def perfstatus(ui, repo, *pats): | ||||
#m = match.always(repo.root, repo.getcwd()) | ||||
#timer(lambda: sum(map(len, repo.dirstate.status(m, False, False, False)))) | ||||
timer(lambda: sum(map(len, repo.status()))) | ||||
def perfheads(ui, repo): | ||||
timer(lambda: len(repo.changelog.heads())) | ||||
def perftags(ui, repo): | ||||
import mercurial.changelog, mercurial.manifest | ||||
def t(): | ||||
repo.changelog = mercurial.changelog.changelog(repo.sopener) | ||||
repo.manifest = mercurial.manifest.manifest(repo.sopener) | ||||
repo.tagscache = None | ||||
return len(repo.tags()) | ||||
timer(t) | ||||
def perfdirstate(ui, repo): | ||||
"a" in repo.dirstate | ||||
def d(): | ||||
repo.dirstate.invalidate() | ||||
"a" in repo.dirstate | ||||
timer(d) | ||||
def perfdirstatedirs(ui, repo): | ||||
"a" in repo.dirstate | ||||
def d(): | ||||
"a" in repo.dirstate._dirs | ||||
del repo.dirstate._dirs | ||||
timer(d) | ||||
def perfmanifest(ui, repo): | ||||
def d(): | ||||
t = repo.manifest.tip() | ||||
m = repo.manifest.read(t) | ||||
repo.manifest.mapcache = None | ||||
repo.manifest._cache = None | ||||
timer(d) | ||||
def perfindex(ui, repo): | ||||
import mercurial.changelog | ||||
def d(): | ||||
t = repo.changelog.tip() | ||||
repo.changelog = mercurial.changelog.changelog(repo.sopener) | ||||
repo.changelog._loadindexmap() | ||||
timer(d) | ||||
def perfstartup(ui, repo): | ||||
cmd = sys.argv[0] | ||||
def d(): | ||||
os.system("HGRCPATH= %s version -q > /dev/null" % cmd) | ||||
timer(d) | ||||
def perfparents(ui, repo): | ||||
nl = [repo.changelog.node(i) for i in xrange(1000)] | ||||
def d(): | ||||
for n in nl: | ||||
repo.changelog.parents(n) | ||||
timer(d) | ||||
def perflookup(ui, repo, rev): | ||||
timer(lambda: len(repo.lookup(rev))) | ||||
|
r7872 | def perflog(ui, repo): | ||
ui.pushbuffer() | ||||
timer(lambda: commands.log(ui, repo, rev=[], date='', user='')) | ||||
ui.popbuffer() | ||||
def perftemplating(ui, repo): | ||||
ui.pushbuffer() | ||||
timer(lambda: commands.log(ui, repo, rev=[], date='', user='', | ||||
template='{date|shortdate} [{rev}:{node|short}]' | ||||
' {author|person}: {desc|firstline}\n')) | ||||
ui.popbuffer() | ||||
|
r7366 | cmdtable = { | ||
'perflookup': (perflookup, []), | ||||
'perfparents': (perfparents, []), | ||||
'perfstartup': (perfstartup, []), | ||||
'perfstatus': (perfstatus, []), | ||||
'perfwalk': (perfwalk, []), | ||||
'perfmanifest': (perfmanifest, []), | ||||
'perfindex': (perfindex, []), | ||||
'perfheads': (perfheads, []), | ||||
'perftags': (perftags, []), | ||||
'perfdirstate': (perfdirstate, []), | ||||
'perfdirstatedirs': (perfdirstate, []), | ||||
|
r7872 | 'perflog': (perflog, []), | ||
'perftemplating': (perftemplating, []), | ||||
|
r7366 | } | ||