def _StartProfiling(self, configuration): """Starts profiling. Args: configuration (ProfilingConfiguration): profiling configuration. """ if not configuration: return if configuration.HaveProfileMemory(): self._memory_profiler = profilers.MemoryProfiler( self._name, configuration) self._memory_profiler.Start() if configuration.HaveProfileProcessing(): identifier = '{0:s}-processing'.format(self._name) self._processing_profiler = profilers.ProcessingProfiler( identifier, configuration) self._processing_profiler.Start() if configuration.HaveProfileSerializers(): identifier = '{0:s}-serializers'.format(self._name) self._serializers_profiler = profilers.SerializersProfiler( identifier, configuration) self._serializers_profiler.Start() if configuration.HaveProfileStorage(): self._storage_profiler = profilers.StorageProfiler( self._name, configuration) self._storage_profiler.Start() if configuration.HaveProfileTaskQueue(): self._task_queue_profiler = profilers.TaskQueueProfiler( self._name, configuration) self._task_queue_profiler.Start()
def testSample(self): """Tests the Sample function.""" task_status = processing_status.TasksStatus() profiling_configuration = configurations.ProfilingConfiguration() with shared_test_lib.TempDirectory() as temp_directory: profiling_configuration.directory = temp_directory test_profiler = profilers.TaskQueueProfiler( 'test', profiling_configuration) test_profiler.Start() for _ in range(5): test_profiler.Sample(task_status) time.sleep(0.01) test_profiler.Stop()