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"""