| pulkit |
| hg-reviewers |
I'm about to add support for counters, and want to avoid duplicating this
logic.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | hgdemandimport/tracing.py (16 lines) |
| # Support code for event tracing in Mercurial. Lives in demandimport | # Support code for event tracing in Mercurial. Lives in demandimport | ||||
| # so it can also be used in demandimport. | # so it can also be used in demandimport. | ||||
| # | # | ||||
| # Copyright 2018 Google LLC. | # Copyright 2018 Google LLC. | ||||
| # | # | ||||
| # This software may be used and distributed according to the terms of the | # This software may be used and distributed according to the terms of the | ||||
| # GNU General Public License version 2 or any later version. | # GNU General Public License version 2 or any later version. | ||||
| from __future__ import absolute_import | from __future__ import absolute_import | ||||
| import contextlib | import contextlib | ||||
| import os | import os | ||||
| _pipe = None | _pipe = None | ||||
| _checked = False | _checked = False | ||||
| @contextlib.contextmanager | def _isactive(): | ||||
| def log(whencefmt, *whenceargs): | |||||
| global _pipe, _session, _checked | global _pipe, _session, _checked | ||||
| if _pipe is None: | if _pipe is None: | ||||
| if _checked: | if _checked: | ||||
| yield | return False | ||||
| return | |||||
| _checked = True | _checked = True | ||||
| if 'HGCATAPULTSERVERPIPE' not in os.environ: | if 'HGCATAPULTSERVERPIPE' not in os.environ: | ||||
| yield | return False | ||||
| return | |||||
| _pipe = open(os.environ['HGCATAPULTSERVERPIPE'], 'w', 1) | _pipe = open(os.environ['HGCATAPULTSERVERPIPE'], 'w', 1) | ||||
| _session = os.environ.get('HGCATAPULTSESSION', 'none') | _session = os.environ.get('HGCATAPULTSESSION', 'none') | ||||
| return True | |||||
| @contextlib.contextmanager | |||||
| def log(whencefmt, *whenceargs): | |||||
| if not _isactive(): | |||||
| yield | |||||
| return | |||||
| whence = whencefmt % whenceargs | whence = whencefmt % whenceargs | ||||
| try: | try: | ||||
| # Both writes to the pipe are wrapped in try/except to ignore | # Both writes to the pipe are wrapped in try/except to ignore | ||||
| # errors, as we can see mysterious errors in here if the pager | # errors, as we can see mysterious errors in here if the pager | ||||
| # is active. Presumably other conditions could trigger | # is active. Presumably other conditions could trigger | ||||
| # problems too. | # problems too. | ||||
| try: | try: | ||||
| _pipe.write('START %s %s\n' % (_session, whence)) | _pipe.write('START %s %s\n' % (_session, whence)) | ||||
| except IOError: | except IOError: | ||||
| pass | pass | ||||
| yield | yield | ||||
| finally: | finally: | ||||
| try: | try: | ||||
| _pipe.write('END %s %s\n' % (_session, whence)) | _pipe.write('END %s %s\n' % (_session, whence)) | ||||
| except IOError: | except IOError: | ||||
| pass | pass | ||||