+x to scripts

[osm/MON.git] / core / message_bus / consumer.py

diff --git a/core/message_bus/consumer.py b/core/message_bus/consumer.py
index 8427076..c9021d2 100644 (file)
--- a/core/message_bus/consumer.py
+++ b/core/message_bus/consumer.py
@@ -27,7 +27,7 @@ alarms and metrics responses.
 '''
 
 __author__ = "Prithiv Mohan"
-__date__   = "06/Sep/2017"
+__date__ = "06/Sep/2017"
 
 
 from kafka import KafkaConsumer
@@ -37,6 +37,7 @@ import logging
 import logging.config
 import os
 
+
 def logging_handler(filename, mode='a+', encoding=None):
     if not os.path.exists(filename):
         open(filename, 'a').close()
@@ -50,9 +51,9 @@ log_config = {
         },
     },
     'handlers': {
-        'file':{
+        'file': {
             '()': logging_handler,
-            'level':'DEBUG',
+            'level': 'DEBUG',
             'formatter': 'default',
             'filename': '/var/log/osm_mon.log',
             'mode': 'a+',
@@ -73,10 +74,15 @@ log_config = {
 logging.config.dictConfig(log_config)
 logger = logging.getLogger('kafka')
 
+if "BROKER_URI" in os.environ:
+    broker = os.getenv("BROKER_URI")
+else:
+    broker = "localhost:9092"
 
-
-alarm_consumer = KafkaConsumer('alarm_response', 'osm_mon', bootstrap_servers = 'localhost:9092')
-metric_consumer = KafkaConsumer('metric_response', 'osm_mon', bootstrap_servers = 'localhost:9092')
+alarm_consumer = KafkaConsumer(
+    'alarm_response', 'osm_mon', bootstrap_servers=broker)
+metric_consumer = KafkaConsumer(
+    'metric_response', 'osm_mon', bootstrap_servers=broker)
 try:
     for message in alarm_consumer:
         logger.debug(message)
@@ -84,7 +90,6 @@ try:
         logger.debug(message)
 except KafkaError:
     log.exception()
-    pass
 
 alarm_consumer.subscribe('alarm_response')
 metric_consumer.subscribe('metric_response')