Reformat MON to standardized format

[osm/MON.git] / osm_mon / cmd / mon_healthcheck.py

diff --git a/osm_mon/cmd/mon_healthcheck.py b/osm_mon/cmd/mon_healthcheck.py
index 412410b..7eae4c2 100644 (file)
--- 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 argparse
 import logging
 import subprocess
 import sys
-import uuid
 
 import requests
 
-from osm_mon.core.message_bus.consumer import Consumer
-
 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.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,18 +69,5 @@ def _is_prometheus_exporter_ok():
         return False
 
 
-def _is_kafka_ok():
-    try:
-        common_consumer = Consumer("mon-healthcheck-" + str(uuid.uuid4()))
-        topics = ['alarm_request', 'vim_account']
-        common_consumer.subscribe(topics)
-        common_consumer.poll()
-        common_consumer.close(autocommit=False)
-        return True
-    except Exception:
-        log.exception("MON can not connect to Kafka")
-        return False
-
-
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()