Implement gathering overall task capacity

For use when running/planning jobs
This commit is contained in:
Matthew Jones
2016-11-07 13:44:17 -05:00
parent db5ffa7e03
commit 343966f744
6 changed files with 32 additions and 4 deletions

View File

@@ -37,7 +37,7 @@ logger = logging.getLogger('awx.main.scheduler')
class TaskManager():
def __init__(self):
self.graph = DependencyGraph()
self.capacity_total = 200
self.capacity_total = Instance.objects.total_capacity()
self.capacity_used = 0
def get_tasks(self):
@@ -152,7 +152,7 @@ class TaskManager():
map(lambda at: active_tasks.add(at['id']), active_task_queues[queue])
else:
if not hasattr(settings, 'CELERY_UNIT_TEST'):
return None
return (None, None)
return (active_task_queues, active_tasks)