mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 07:17:55 +03:00
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
|
# fbamend.py - improved amend functionality
|
||
|
|
#
|
||
|
|
# 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)
|
||
|
|
"""
|
||
|
|
|
||
|
|
from mercurial import dispatch, extensions
|
||
|
|
|
||
|
|
def extsetup(ui):
|
||
|
|
extensions.wrapfunction(dispatch, 'statprofile', statprofile)
|
||
|
|
|
||
|
|
def statprofile(orig, ui, func, 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:
|
||
|
|
return func()
|
||
|
|
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)
|