| #!/usr/bin/python2 |
| |
| # Copyright 2015 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Counts the number of jobs created in the last 24 hours.""" |
| |
| import argparse |
| from datetime import datetime, timedelta |
| import sys |
| |
| import common |
| from autotest_lib.frontend import setup_django_environment |
| from autotest_lib.frontend.afe import models |
| from autotest_lib.server import site_utils |
| |
| try: |
| from chromite.lib import metrics |
| except ImportError: |
| metrics = site_utils.metrics_mock |
| |
| |
| def number_of_jobs_since(delta): |
| """Returns the number of jobs kicked off in the last |duration| minutes. |
| |
| @param delta: A timedelta which indicates the maximum age of the jobs to count |
| """ |
| cutoff = datetime.now() - delta |
| return models.Job.objects.filter(created_on__gt=cutoff).count() |
| |
| |
| def main(): |
| """Counts the number of AFE jobs in the last day and report to monarch.""" |
| parser = argparse.ArgumentParser( |
| description=('A script which records the number of afe jobs run in a time interval.')) |
| parser.parse_args(sys.argv[1:]) |
| count = number_of_jobs_since(timedelta(days=1)) |
| |
| with site_utils.SetupTsMonGlobalState('count_jobs', short_lived=True): |
| # TODO: Reporting a stat for each job created from the afe directly could be better. |
| # More discussions are needed to decide whether to remove this file. |
| metrics.Gauge('chromeos/autotest/experimental/jobs_rate/afe_daily_count').set(count) |
| |
| |
| if __name__ == '__main__': |
| main() |