diff options
Diffstat (limited to 'buildstream/_scheduler/scheduler.py')
-rw-r--r-- | buildstream/_scheduler/scheduler.py | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/buildstream/_scheduler/scheduler.py b/buildstream/_scheduler/scheduler.py index 176900b33..50ad7f07a 100644 --- a/buildstream/_scheduler/scheduler.py +++ b/buildstream/_scheduler/scheduler.py @@ -29,7 +29,7 @@ from contextlib import contextmanager # Local imports from .resources import Resources, ResourceType from .jobs import JobStatus, CacheSizeJob, CleanupJob -from .._profile import Topics, profile_start, profile_end +from .._profile import Topics, PROFILER # A decent return code for Scheduler.run() @@ -156,14 +156,11 @@ class Scheduler(): self._check_cache_management() # Start the profiler - profile_start(Topics.SCHEDULER, "_".join(queue.action_name for queue in self.queues)) - - # Run the queues - self._sched() - self.loop.run_forever() - self.loop.close() - - profile_end(Topics.SCHEDULER, "_".join(queue.action_name for queue in self.queues)) + with PROFILER.profile(Topics.SCHEDULER, "_".join(queue.action_name for queue in self.queues)): + # Run the queues + self._sched() + self.loop.run_forever() + self.loop.close() # Stop handling unix signals self._disconnect_signals() |