mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
214a35d6cb
Summary: Upstream mercurial has changed, breaking statprof Test Plan: HGPROF=stat ./hg-dev --config extensions.statprofext=~/facebook-hg-rpms/fb-hgext/hgext3rd/statprofext.py --config profiling.enabled=True --config profiling.frequency=10000 --profile log -l 1000 --pager=off Reviewers: durham, quark, #mercurial, ttung Reviewed By: ttung Subscribers: mjpieters Differential Revision: https://phabricator.intern.facebook.com/D3786387 Signature: t1:3786387:1472497930:488d7d584395ca686de4dc458c87eb05fc944282
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
# statprofext.py - improved statprof integration
|
|
#
|
|
# Copyright 2013 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
"""integrates the statprof profiler more tightly with Mercurial
|
|
|
|
This extension configures the statprof based on Mercurial config knobs.
|
|
|
|
statprof.mechanism - configures which profiling mechanism is used
|
|
(thread, signal)
|
|
statprof.format - configures the output format
|
|
(hotpath, json)
|
|
"""
|
|
|
|
import contextlib
|
|
from mercurial import profiling, extensions
|
|
|
|
def extsetup(ui):
|
|
extensions.wrapfunction(profiling, 'statprofile', statprofile)
|
|
|
|
@contextlib.contextmanager
|
|
def statprofile(orig, ui, fp):
|
|
try:
|
|
import statprof
|
|
except ImportError:
|
|
raise error.Abort(_(
|
|
'statprof not available - install using "easy_install statprof"'))
|
|
|
|
freq = ui.configint('profiling', 'freq', default=1000)
|
|
if freq > 0:
|
|
statprof.reset(freq)
|
|
else:
|
|
ui.warn(_("invalid sampling frequency '%s' - ignoring\n") % freq)
|
|
|
|
mechanism = ui.config('statprof', 'mechanism', 'thread')
|
|
statprof.start(mechanism=mechanism)
|
|
try:
|
|
yield
|
|
finally:
|
|
statprof.stop()
|
|
|
|
profformat = ui.config('statprof', 'format', 'hotpath')
|
|
|
|
if profformat == 'hotpath':
|
|
displayformat = statprof.DisplayFormats.Hotpath
|
|
elif profformat == 'json':
|
|
displayformat = statprof.DisplayFormats.Json
|
|
else:
|
|
ui.warn(_("unknown profiler output format: %s") % profformat)
|
|
displayformat = statprof.DisplayFormats.Hotpath
|
|
|
|
statprof.display(fp, format=displayformat)
|