62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
# Entrypoint/script to setup and start an LTTng tracing session
|
|
|
|
import argparse
|
|
import time
|
|
|
|
from tracetools_trace.tools import lttng
|
|
from tracetools_trace.tools import names
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Setup and launch an LTTng tracing session.')
|
|
parser.add_argument('--session-name', '-s', dest='session_name',
|
|
default=f'session-{time.strftime("%Y%m%d%H%M%S")}',
|
|
help='the name of the tracing session (default: session-YYYYMMDDHHMMSS)')
|
|
parser.add_argument('--path', '-p', dest='path',
|
|
default='/tmp',
|
|
help='path of the base directory for trace data (default: %(default)s)')
|
|
parser.add_argument('--ust', '-u', nargs='*', dest='events_ust', default=names.DEFAULT_EVENTS_ROS,
|
|
help='the ROS UST events to enable (default: all events) '
|
|
'[to disable all UST events, '
|
|
'provide this flag without any event name]')
|
|
parser.add_argument('--kernel', '-k', nargs='*', dest='events_kernel',
|
|
default=names.DEFAULT_EVENTS_KERNEL,
|
|
help='the kernel events to enable (default: all events) '
|
|
'[to disable all UST events, '
|
|
'provide this flag without any event name]')
|
|
parser.add_argument('--list', '-l', dest='list', action='store_true',
|
|
help='display lists of enabled events (default: %(default)s)')
|
|
args = parser.parse_args()
|
|
|
|
session_name = args.session_name
|
|
base_path = args.path
|
|
full_path = base_path + '/' + session_name
|
|
ros_events = args.events_ust
|
|
kernel_events = args.events_kernel
|
|
|
|
ust_enabled = len(ros_events) > 0
|
|
kernel_enabled = len(kernel_events) > 0
|
|
if ust_enabled:
|
|
print(f'UST tracing enabled ({len(ros_events)} events)')
|
|
if args.list:
|
|
print(f'\tevents: {ros_events}')
|
|
else:
|
|
print('UST tracing disabled')
|
|
if kernel_enabled:
|
|
print(f'kernel tracing enabled ({len(kernel_events)} events)')
|
|
if args.list:
|
|
print(f'\tevents: {kernel_events}')
|
|
else:
|
|
print('kernel tracing disabled')
|
|
|
|
lttng.lttng_setup(session_name, full_path, ros_events=ros_events, kernel_events=kernel_events)
|
|
print(f'writting tracing session to: {full_path}')
|
|
|
|
input('press enter to start...')
|
|
lttng.lttng_start(session_name)
|
|
input('press enter to stop...')
|
|
|
|
print('stopping & destroying tracing session')
|
|
lttng.lttng_stop(session_name)
|
|
lttng.lttng_destroy(session_name)
|