You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.5 KiB
48 lines
1.5 KiB
4 months ago
|
#!/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()
|