mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 01:07:15 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
98 lines
2.8 KiB
Python
Executable File
98 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# hgperf - measure performance of Mercurial commands
|
|
#
|
|
# Copyright 2014 Matt Mackall <mpm@selenic.com>
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
'''measure performance of Mercurial commands
|
|
|
|
Using ``hgperf`` instead of ``hg`` measures performance of the target
|
|
Mercurial command. For example, the execution below measures
|
|
performance of :hg:`heads --topo`::
|
|
|
|
$ hgperf heads --topo
|
|
|
|
All command output via ``ui`` is suppressed, and just measurement
|
|
result is displayed: see also "perf" extension in "contrib".
|
|
|
|
Costs of processing before dispatching to the command function like
|
|
below are not measured::
|
|
|
|
- parsing command line (e.g. option validity check)
|
|
- reading configuration files in
|
|
|
|
But ``pre-`` and ``post-`` hook invocation for the target command is
|
|
measured, even though these are invoked before or after dispatching to
|
|
the command function, because these may be required to repeat
|
|
execution of the target command correctly.
|
|
'''
|
|
|
|
import os
|
|
import sys
|
|
|
|
libdir = '@LIBDIR@'
|
|
|
|
if libdir != '@' 'LIBDIR' '@':
|
|
if not os.path.isabs(libdir):
|
|
libdir = os.path.join(os.path.dirname(os.path.realpath(__file__)),
|
|
libdir)
|
|
libdir = os.path.abspath(libdir)
|
|
sys.path.insert(0, libdir)
|
|
|
|
# enable importing on demand to reduce startup time
|
|
try:
|
|
from edenscm.mercurial import demandimport; demandimport.enable()
|
|
except ImportError:
|
|
import sys
|
|
sys.stderr.write("abort: couldn't find mercurial libraries in [%s]\n" %
|
|
' '.join(sys.path))
|
|
sys.stderr.write("(check your install and PYTHONPATH)\n")
|
|
sys.exit(-1)
|
|
|
|
from edenscm.mercurial import (
|
|
dispatch,
|
|
util,
|
|
)
|
|
|
|
def timer(func, title=None):
|
|
results = []
|
|
begin = util.timer()
|
|
count = 0
|
|
while True:
|
|
ostart = os.times()
|
|
cstart = util.timer()
|
|
r = func()
|
|
cstop = util.timer()
|
|
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 title:
|
|
sys.stderr.write("! %s\n" % title)
|
|
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))
|
|
|
|
orgruncommand = dispatch.runcommand
|
|
|
|
def runcommand(lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions):
|
|
ui.pushbuffer()
|
|
lui.pushbuffer()
|
|
timer(lambda : orgruncommand(lui, repo, cmd, fullargs, ui,
|
|
options, d, cmdpats, cmdoptions))
|
|
ui.popbuffer()
|
|
lui.popbuffer()
|
|
|
|
dispatch.runcommand = runcommand
|
|
|
|
dispatch.run()
|