osm_mon/cmd/mon_healthcheck.py

   1 # Copyright 2018 Whitestack, LLC
   2 # *************************************************************
   3
   4 # This file is part of OSM Monitoring module
   5 # All Rights Reserved to Whitestack, LLC
   6
   7 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   8 # not use this file except in compliance with the License. You may obtain
   9 # a copy of the License at
  10
  11 #         http://www.apache.org/licenses/LICENSE-2.0
  12
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  15 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  16 # License for the specific language governing permissions and limitations
  17 # under the License.
  18
  19 # For those usages not covered by the Apache License, Version 2.0 please
  20 # contact: bdiaz@whitestack.com or glavado@whitestack.com
  21 ##
  22 import argparse
  23 import asyncio
  24 import logging
  25 import subprocess
  26 import sys
  27
  28 import requests
  29 from aiokafka import AIOKafkaConsumer
  30
  31 from osm_mon.core.config import Config
  32
  33 log = logging.getLogger(__name__)
  34
  35
  36 def main():
  37     parser = argparse.ArgumentParser(prog='osm-mon-healthcheck')
  38     parser.add_argument('--config-file', nargs='?', help='MON configuration file')
  39     args = parser.parse_args()
  40     cfg = Config(args.config_file)
  41
  42     if not _processes_running():
  43         sys.exit(1)
  44     if not _is_kafka_ok(cfg.get('message', 'host'), cfg.get('message', 'port')):
  45         sys.exit(1)
  46     if not _is_prometheus_exporter_ok():
  47         sys.exit(1)
  48     sys.exit(0)
  49
  50
  51 def _processes_running():
  52     def _contains_process(processes, process_name):
  53         for row in processes:
  54             if process_name in row:
  55                 return True
  56         return False
  57
  58     processes_to_check = ['osm-mon-collector', 'osm-mon-evaluator', 'osm-mon-server']
  59     ps = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE).communicate()[0]
  60     processes_running = ps.decode().split('\n')
  61     for p in processes_to_check:
  62         if not _contains_process(processes_running, p):
  63             return False
  64     return True
  65
  66
  67 def _is_prometheus_exporter_ok():
  68     try:
  69         r = requests.get('http://localhost:8000')
  70         r.raise_for_status()
  71         return True
  72     except Exception:
  73         log.exception("MON Prometheus exporter is not running")
  74         return False
  75
  76
  77 def _is_kafka_ok(host, port):
  78     async def _test_kafka(loop):
  79         consumer = AIOKafkaConsumer(
  80             'healthcheck',
  81             loop=loop, bootstrap_servers='{}:{}'.format(host, port))
  82         await consumer.start()
  83         await consumer.stop()
  84
  85     try:
  86         loop = asyncio.get_event_loop()
  87         loop.run_until_complete(_test_kafka(loop))
  88         return True
  89     except Exception:
  90         log.exception("MON can not connect to Kafka")
  91         return False
  92
  93
  94 if __name__ == '__main__':
  95     main()