| hg-reviewers |
As spotted by Gregory, we should use a monotonic timer to get better timings.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | contrib/catapipe.py (10 lines) |
| Status | Author | Revision | |
|---|---|---|---|
| Closed | lothiraldan | ||
| Closed | lothiraldan | ||
| Closed | lothiraldan | ||
| Closed | lothiraldan | ||
| Closed | lothiraldan | ||
| Closed | lothiraldan | ||
| Closed | lothiraldan | ||
| Closed | lothiraldan |
| Typically you'll want to place the path to the named pipe in the | Typically you'll want to place the path to the named pipe in the | ||||
| HGCATAPULTSERVERPIPE environment variable, which both run-tests and hg | HGCATAPULTSERVERPIPE environment variable, which both run-tests and hg | ||||
| understand. | understand. | ||||
| """ | """ | ||||
| from __future__ import absolute_import, print_function | from __future__ import absolute_import, print_function | ||||
| import argparse | import argparse | ||||
| import datetime | |||||
| import json | import json | ||||
| import os | import os | ||||
| import timeit | |||||
| _TYPEMAP = { | _TYPEMAP = { | ||||
| 'START': 'B', | 'START': 'B', | ||||
| 'END': 'E', | 'END': 'E', | ||||
| } | } | ||||
| _threadmap = {} | _threadmap = {} | ||||
| # Timeit already contains the whole logic about which timer to use based on | |||||
| # Python version and OS | |||||
| timer = timeit.default_timer | |||||
| def main(): | def main(): | ||||
| parser = argparse.ArgumentParser() | parser = argparse.ArgumentParser() | ||||
| parser.add_argument('pipe', type=str, nargs=1, | parser.add_argument('pipe', type=str, nargs=1, | ||||
| help='Path of named pipe to create and listen on.') | help='Path of named pipe to create and listen on.') | ||||
| parser.add_argument('output', default='trace.json', type=str, nargs='?', | parser.add_argument('output', default='trace.json', type=str, nargs='?', | ||||
| help='Path of json file to create where the traces ' | help='Path of json file to create where the traces ' | ||||
| 'will be stored.') | 'will be stored.') | ||||
| parser.add_argument('--debug', default=False, action='store_true', | parser.add_argument('--debug', default=False, action='store_true', | ||||
| help='Print useful debug messages') | help='Print useful debug messages') | ||||
| args = parser.parse_args() | args = parser.parse_args() | ||||
| fn = args.pipe[0] | fn = args.pipe[0] | ||||
| os.mkfifo(fn) | os.mkfifo(fn) | ||||
| try: | try: | ||||
| with open(fn) as f, open(args.output, 'w') as out: | with open(fn) as f, open(args.output, 'w') as out: | ||||
| out.write('[\n') | out.write('[\n') | ||||
| start = datetime.datetime.now() | start = timer() | ||||
| while True: | while True: | ||||
| ev = f.readline().strip() | ev = f.readline().strip() | ||||
| if not ev: | if not ev: | ||||
| continue | continue | ||||
| now = datetime.datetime.now() | now = timer() | ||||
| if args.debug: | if args.debug: | ||||
| print(ev) | print(ev) | ||||
| verb, session, label = ev.split(' ', 2) | verb, session, label = ev.split(' ', 2) | ||||
| if session not in _threadmap: | if session not in _threadmap: | ||||
| _threadmap[session] = len(_threadmap) | _threadmap[session] = len(_threadmap) | ||||
| pid = _threadmap[session] | pid = _threadmap[session] | ||||
| ts_micros = (now - start).total_seconds() * 1000000 | ts_micros = (now - start).total_seconds() * 1000000 | ||||
| out.write(json.dumps( | out.write(json.dumps( | ||||