Mercurial > hg
comparison mercurial/dispatch.py @ 29783:5d44197c208b
profiling: make profiling functions context managers (API)
This makes profiling more flexible since we can now call multiple
functions when a profiler is active. But the real reason for this
is to enable a future consumer to profile a function that returns
a generator. We can't do this from the profiling function itself
because functions can either be generators or have return values:
they can't be both. So therefore it isn't possible to have a generic
profiling function that can both consume and re-emit a generator
and return a value.
author | Gregory Szorc <gregory.szorc@gmail.com> |
---|---|
date | Sun, 14 Aug 2016 18:25:22 -0700 |
parents | 97bfc2e5fba5 |
children | e3501546f7e4 |
comparison
equal
deleted
inserted
replaced
29782:97bfc2e5fba5 | 29783:5d44197c208b |
---|---|
907 return cmdfunc() | 907 return cmdfunc() |
908 except error.SignatureError: | 908 except error.SignatureError: |
909 raise error.CommandError(cmd, _("invalid arguments")) | 909 raise error.CommandError(cmd, _("invalid arguments")) |
910 | 910 |
911 if ui.configbool('profiling', 'enabled'): | 911 if ui.configbool('profiling', 'enabled'): |
912 return profiling.profile(ui, checkargs) | 912 with profiling.profile(ui): |
913 return checkargs() | |
913 else: | 914 else: |
914 return checkargs() | 915 return checkargs() |
915 | 916 |
916 def _exceptionwarning(ui): | 917 def _exceptionwarning(ui): |
917 """Produce a warning message for the current active exception""" | 918 """Produce a warning message for the current active exception""" |