scripts/sysmon/mainlib.py

# Copyright 2016 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Send system monitoring data to the timeseries monitoring API."""

from __future__ import absolute_import

import logging
import time

from chromite.third_party.infra_libs.ts_mon.common import interface

from chromite.lib import commandline
from chromite.lib import metrics
from chromite.lib import ts_mon_config
from chromite.scripts.sysmon import git_metrics
from chromite.scripts.sysmon import loop
from chromite.scripts.sysmon import net_metrics
from chromite.scripts.sysmon import osinfo_metrics
from chromite.scripts.sysmon import proc_metrics
from chromite.scripts.sysmon import puppet_metrics
from chromite.scripts.sysmon import system_metrics


logger = logging.getLogger(__name__)


class _MetricCollector:
    """Metric collector class."""

    def __init__(self):
        self._collect_osinfo = _TimedCallback(
            callback=osinfo_metrics.collect_os_info, interval=60 * 60
        )

    def __call__(self):
        """Collect metrics."""
        system_metrics.collect_uptime()
        system_metrics.collect_cpu_info()
        system_metrics.collect_disk_info()
        system_metrics.collect_mem_info()
        net_metrics.collect_net_info()
        proc_metrics.collect_proc_info()
        system_metrics.collect_load_avg()
        puppet_metrics.collect_puppet_summary()
        git_metrics.collect_git_metrics()
        self._collect_osinfo()
        system_metrics.collect_unix_time()  # must be just before flush
        metrics.Flush()


class _TimedCallback:
    """Limits callback to one call in a given interval."""

    def __init__(self, callback, interval):
        """Initialize instance.

        Args:
            callback: function to call
            interval: Number of seconds between allowed calls
        """
        self._callback = callback
        self._interval = interval
        self._last_called = float("-inf")

    def __call__(self):
        if time.time() >= self._next_call:
            self._callback()
            self._last_called = time.time()

    @property
    def _next_call(self):
        return self._last_called + self._interval


def main():
    parser = commandline.ArgumentParser(
        description=__doc__, default_log_level="DEBUG"
    )
    parser.add_argument(
        "--interval",
        default=60,
        type=int,
        help="time (in seconds) between sampling system metrics",
    )
    opts = parser.parse_args()
    opts.Freeze()

    # This call returns a context manager that doesn't do anything, so we
    # ignore the return value.
    ts_mon_config.SetupTsMonGlobalState("sysmon", auto_flush=False)
    # The default prefix is '/chrome/infra/'.
    interface.state.metric_name_prefix = (
        interface.state.metric_name_prefix + "chromeos/sysmon/"
    )

    collector = _MetricCollector()
    loop.SleepLoop(callback=collector, interval=opts.interval).loop_forever()