| 1 | # Copyright (C) 2015 - Philippe Proulx <pproulx@efficios.com> |
| 2 | # Copyright (C) 2014 - David Goulet <dgoulet@efficios.com> |
| 3 | # |
| 4 | # This library is free software; you can redistribute it and/or modify it under |
| 5 | # the terms of the GNU Lesser General Public License as published by the Free |
| 6 | # Software Foundation; version 2.1 of the License. |
| 7 | # |
| 8 | # This library is distributed in the hope that it will be useful, but WITHOUT |
| 9 | # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
| 10 | # FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
| 11 | # details. |
| 12 | # |
| 13 | # You should have received a copy of the GNU Lesser General Public License |
| 14 | # along with this library; if not, write to the Free Software Foundation, Inc., |
| 15 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 16 | |
| 17 | from __future__ import unicode_literals, print_function |
| 18 | import logging |
| 19 | import time |
| 20 | import sys |
| 21 | import argparse |
| 22 | import os |
| 23 | |
| 24 | |
| 25 | def _perror(msg): |
| 26 | print(msg, file=sys.stderr) |
| 27 | sys.exit(1) |
| 28 | |
| 29 | |
| 30 | try: |
| 31 | import lttngust |
| 32 | except (ImportError) as e: |
| 33 | _perror('lttngust package not found: {}'.format(e)) |
| 34 | |
| 35 | |
| 36 | def _main(): |
| 37 | ev1 = logging.getLogger('python-ev-test1'); |
| 38 | ev2 = logging.getLogger('python-ev-test2'); |
| 39 | |
| 40 | logging.basicConfig() |
| 41 | |
| 42 | parser = argparse.ArgumentParser() |
| 43 | parser.add_argument('-n', '--nr-iter', required=True) |
| 44 | parser.add_argument('-s', '--wait', required=True) |
| 45 | parser.add_argument('-d', '--fire-debug-event', action="store_true") |
| 46 | parser.add_argument('-e', '--fire-second-event', action="store_true") |
| 47 | parser.add_argument('-r', '--ready-file') |
| 48 | parser.add_argument('-g', '--go-file') |
| 49 | args = parser.parse_args() |
| 50 | |
| 51 | nr_iter = int(args.nr_iter) |
| 52 | wait_time = float(args.wait) |
| 53 | fire_debug_ev = args.fire_debug_event |
| 54 | fire_second_ev = args.fire_second_event |
| 55 | |
| 56 | ready_file = args.ready_file |
| 57 | go_file = args.go_file |
| 58 | |
| 59 | if ready_file is not None and os.path.exists(ready_file): |
| 60 | raise ValueError('Ready file already exist') |
| 61 | |
| 62 | if go_file is not None and os.path.exists(go_file): |
| 63 | raise ValueError('Go file already exist. Review synchronization') |
| 64 | |
| 65 | if (ready_file is None) != (go_file is None): |
| 66 | raise ValueError('--go-file and --ready-file need each others, review' |
| 67 | 'synchronization') |
| 68 | |
| 69 | |
| 70 | # Inform that we are ready, if necessary |
| 71 | if ready_file is not None: |
| 72 | open(ready_file, 'a').close() |
| 73 | |
| 74 | # Wait for go, if necessary |
| 75 | while go_file is not None and not os.path.exists(go_file): |
| 76 | time.sleep(0.5) |
| 77 | |
| 78 | for i in range(nr_iter): |
| 79 | ev1.info('{} fired [INFO]'.format(ev1.name)) |
| 80 | |
| 81 | if fire_debug_ev: |
| 82 | ev1.debug('{} fired [DEBUG]'.format(ev1.name)) |
| 83 | |
| 84 | time.sleep(wait_time) |
| 85 | |
| 86 | if fire_second_ev: |
| 87 | ev2.info('{} fired [INFO]'.format(ev2.name)) |
| 88 | |
| 89 | if ready_file is not None: |
| 90 | try: |
| 91 | os.unlink(ready_file) |
| 92 | except: |
| 93 | print("Unexpected error on ready file unlink:", sys.exc_info()[0]) |
| 94 | raise |
| 95 | |
| 96 | |
| 97 | if __name__ == '__main__': |
| 98 | _main() |