server/standalone_profiler.py

#
# Copyright 2007 Google Inc. All Rights Reserved.

"""Runs profilers on a machine when no autotest job is running.

This is used to profile a task when the task is running on a machine that is not
running through autotest.
"""

__author__ = 'cranger@google.com (Colby Ranger)'

import common
from autotest_lib.client.common_lib import barrier

RUNTEST_PATTERN="job.run_test('barriertest',timeout_sync=%r,timeout_start=%r,\
timeout_stop=%r,hostid='%s',masterid='%s',all_ids=%r)"

def _encode_args(profiler, args, dargs):
    parts = [repr(profiler)]
    parts += [repr(arg) for arg in args]
    parts += ["%s=%r" % darg for darg in dargs.iteritems()]
    return ", ".join(parts)


def generate_test(machines, hostname, profilers, timeout_start, timeout_stop,
                        timeout_sync=180):
    """
    Generate control file that enables given profilers and starts barriertest.

    @param machines: sequence of all the hostnames involved in the barrier
            synchronization
    @param hostname: hostname of the machine running the generated control file
    @param profilers: a sequence of 3 items tuples where the first item is a
            string (the profiler name), second argument is a tuple with the
            non keyword arguments to give to the profiler when being added
            with "job.profilers.add()" in the control file, third item is
            a dictionary of the keyword arguments to give it
    @param timeout_start: how many seconds to wait in barriertest for the
            profilers to start (None means no timeout)
    @param timeout_stop: how many seconds to wait in barriertest for the
            profilers to stop (None means no timeout)
    @param timeout_sync: how many seconds to wait in barriertest for other
            machines to reach the start of the barriertest (None means no
            timeout)
    """
    control_file = []
    for profiler in profilers:
        control_file.append("job.profilers.add(%s)"
                            % _encode_args(*profiler))

    control_file.append(RUNTEST_PATTERN % (timeout_sync, timeout_start,
        timeout_stop, hostname, "PROF_MASTER", machines))

    for profiler in profilers:
        control_file.append("job.profilers.delete('%s')" % profiler[0])

    return "\n".join(control_file)


def wait_for_profilers(machines, timeout = 300):
    sb = barrier.barrier("PROF_MASTER", "sync_profilers",
            timeout, port=11920)
    sb.rendezvous_servers("PROF_MASTER", *machines)


def start_profilers(machines, timeout = 120):
    sb = barrier.barrier("PROF_MASTER", "start_profilers",
            timeout, port=11920)
    sb.rendezvous_servers("PROF_MASTER", *machines)


def stop_profilers(machines, timeout = 120):
    sb = barrier.barrier("PROF_MASTER", "stop_profilers",
            timeout, port=11920)
    sb.rendezvous_servers("PROF_MASTER", *machines)


def finish_profilers(machines, timeout = 120):
    sb = barrier.barrier("PROF_MASTER", "finish_profilers",
            timeout, port=11920)
    sb.rendezvous_servers("PROF_MASTER", *machines)