49 lines
2 KiB
Python
49 lines
2 KiB
Python
#!/usr/bin/env python3
|
|
# Entrypoint/script to setup and start an LTTng tracing session
|
|
|
|
import sys
|
|
import time
|
|
import argparse
|
|
from tracetools_trace.tools.lttng import (
|
|
lttng_setup,
|
|
lttng_start,
|
|
lttng_stop,
|
|
lttng_destroy,
|
|
)
|
|
from tracetools_trace.tools.names import (
|
|
DEFAULT_EVENTS_ROS,
|
|
DEFAULT_EVENTS_KERNEL,
|
|
)
|
|
|
|
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=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=DEFAULT_EVENTS_KERNEL,
|
|
help='the kernel events to enable (default: all events) [to disable all UST events, provide this flag without any event name]')
|
|
args = parser.parse_args()
|
|
|
|
session_name = args.session_name
|
|
base_path = args.path
|
|
path = base_path + '/' + session_name
|
|
ros_events = args.events_ust
|
|
kernel_events = args.events_kernel
|
|
|
|
lttng_setup(session_name, path, ros_events=ros_events, kernel_events=kernel_events)
|
|
print(f'writting tracing session to: {path}')
|
|
|
|
input('press enter to start...')
|
|
lttng_start(session_name)
|
|
|
|
# TODO integrate this with launch + ROS shutdown
|
|
input('press enter to stop...')
|
|
|
|
print('stopping & destroying tracing session')
|
|
lttng_stop(session_name)
|
|
lttng_destroy(session_name)
|