1 # 2 # Copyright 2007 Google Inc. All Rights Reserved. 3 4 """Runs profilers on a machine when no autotest job is running. 5 6 This is used to profile a task when the task is running on a machine that is not 7 running through autotest. 8 """ 9 10 __author__ = 'cranger (at] google.com (Colby Ranger)' 11 12 import platform 13 import common 14 from autotest_lib.client.common_lib import barrier 15 16 # Client control file snippet used to synchronize profiler start & stop. 17 _RUNTEST_PATTERN = ("job.run_test('profiler_sync', timeout_sync=%r,\n" 18 " timeout_start=%r, timeout_stop=%r,\n" 19 " hostid='%s', masterid='%s', all_ids=%r)") 20 _PROF_MASTER = platform.node() 21 _PORT = 11920 22 23 24 def _encode_args(profiler, args, dargs): 25 parts = [repr(profiler)] 26 parts += [repr(arg) for arg in args] 27 parts += ["%s=%r" % darg for darg in dargs.iteritems()] 28 return ", ".join(parts) 29 30 31 def generate_test(machines, hostname, profilers, timeout_start, timeout_stop, 32 timeout_sync=180): 33 """ 34 Generate a control file that enables profilers and starts profiler_sync. 35 36 @param machines: sequence of all the hostnames involved in the barrier 37 synchronization 38 @param hostname: hostname of the machine running the generated control file 39 @param profilers: a sequence of 3 items tuples where the first item is a 40 string (the profiler name), second argument is a tuple with the 41 non keyword arguments to give to the profiler when being added 42 with "job.profilers.add()" in the control file, third item is 43 a dictionary of the keyword arguments to give it 44 @param timeout_start: how many seconds to wait in profiler_sync for the 45 profilers to start (None means no timeout) 46 @param timeout_stop: how many seconds to wait in profiler_sync for the 47 profilers to stop (None means no timeout) 48 @param timeout_sync: how many seconds to wait in profiler_sync for other 49 machines to reach the start of the profiler_sync (None means no 50 timeout) 51 """ 52 control_file = [] 53 for profiler in profilers: 54 control_file.append("job.profilers.add(%s)" 55 % _encode_args(*profiler)) 56 57 profiler_sync_call = (_RUNTEST_PATTERN % 58 (timeout_sync, timeout_start, timeout_stop, 59 hostname, _PROF_MASTER, machines)) 60 control_file.append(profiler_sync_call) 61 62 for profiler in reversed(profilers): 63 control_file.append("job.profilers.delete('%s')" % profiler[0]) 64 65 return "\n".join(control_file) 66 67 68 def wait_for_profilers(machines, timeout=300): 69 sb = barrier.barrier(_PROF_MASTER, "sync_profilers", 70 timeout, port=_PORT) 71 sb.rendezvous_servers(_PROF_MASTER, *machines) 72 73 74 def start_profilers(machines, timeout=120): 75 sb = barrier.barrier(_PROF_MASTER, "start_profilers", 76 timeout, port=_PORT) 77 sb.rendezvous_servers(_PROF_MASTER, *machines) 78 79 80 def stop_profilers(machines, timeout=120): 81 sb = barrier.barrier(_PROF_MASTER, "stop_profilers", 82 timeout, port=_PORT) 83 sb.rendezvous_servers(_PROF_MASTER, *machines) 84 85 86 def finish_profilers(machines, timeout=120): 87 sb = barrier.barrier(_PROF_MASTER, "finish_profilers", 88 timeout, port=_PORT) 89 sb.rendezvous_servers(_PROF_MASTER, *machines) 90
