X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_mon%2Fcmd%2Fmon_healthcheck.py;h=7eae4c295a9054659dea16c84e4b122cbbdcb227;hb=HEAD;hp=1fa2c2b8e371d4085abf02307cc97ba3abdbdfda;hpb=274a6e9fa4268db2a87525a27bc574c46eaa80dc;p=osm%2FMON.git diff --git a/osm_mon/cmd/mon_healthcheck.py b/osm_mon/cmd/mon_healthcheck.py index 1fa2c2b..7eae4c2 100644 --- a/osm_mon/cmd/mon_healthcheck.py +++ b/osm_mon/cmd/mon_healthcheck.py @@ -19,25 +19,24 @@ # For those usages not covered by the Apache License, Version 2.0 please # contact: bdiaz@whitestack.com or glavado@whitestack.com ## -import asyncio +import argparse import logging import subprocess import sys import requests -from aiokafka import AIOKafkaConsumer - -from osm_mon.core.settings import Config log = logging.getLogger(__name__) def main(): - # Check Kafka + parser = argparse.ArgumentParser(prog="osm-mon-healthcheck") + parser.add_argument("--config-file", nargs="?", help="MON configuration file") + # args = parser.parse_args() + # cfg = Config(args.config_file) + if not _processes_running(): sys.exit(1) - if not _is_kafka_ok(): - sys.exit(1) if not _is_prometheus_exporter_ok(): sys.exit(1) sys.exit(0) @@ -49,18 +48,20 @@ def _processes_running(): if process_name in row: return True return False - processes_to_check = ['osm-mon-collector', 'osm-mon-evaluator', 'osm-mon-server'] - ps = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE).communicate()[0] - processes_running = ps.decode().split('\n') + + processes_to_check = ["osm-mon-collector", "osm-mon-evaluator", "osm-mon-server"] + ps = subprocess.Popen(["ps", "aux"], stdout=subprocess.PIPE).communicate()[0] + processes_running = ps.decode().split("\n") for p in processes_to_check: if not _contains_process(processes_running, p): + log.error("Process %s not running!" % p) return False return True def _is_prometheus_exporter_ok(): try: - r = requests.get('http://localhost:8000') + r = requests.get("http://localhost:8000") r.raise_for_status() return True except Exception: @@ -68,23 +69,5 @@ def _is_prometheus_exporter_ok(): return False -def _is_kafka_ok(): - async def _test_kafka(loop): - cfg = Config.instance() - consumer = AIOKafkaConsumer( - 'healthcheck', - loop=loop, bootstrap_servers=cfg.BROKER_URI) - await consumer.start() - await consumer.stop() - - try: - loop = asyncio.get_event_loop() - loop.run_until_complete(_test_kafka(loop)) - return True - except Exception: - log.exception("MON can not connect to Kafka") - return False - - -if __name__ == '__main__': +if __name__ == "__main__": main()