Adds OSMMON_LOG_LEVEL env var to config log level
[osm/MON.git] / osm_mon / plugins / OpenStack / Aodh / notifier.py
index 5330197..c43f238 100644 (file)
 #
 """A Webserver to send alarm notifications from Aodh to the SO."""
 import json
-
 import logging
-
+import os
 import sys
-
 import time
 
-from BaseHTTPServer import BaseHTTPRequestHandler
-from BaseHTTPServer import HTTPServer
+from six.moves.BaseHTTPServer import BaseHTTPRequestHandler
+from six.moves.BaseHTTPServer import HTTPServer
 
 # Initialise a logger for alarm notifier
-logging.basicConfig(filename='aodh_notify.log',
-                    format='%(asctime)s %(message)s',
-                    datefmt='%m/%d/%Y %I:%M:%S %p', filemode='a',
-                    level=logging.INFO)
+from osm_mon.core.settings import Config
+
+cfg = Config.instance()
+
+logging.basicConfig(stream=sys.stdout,
+                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+                    datefmt='%m/%d/%Y %I:%M:%S %p',
+                    level=logging.getLevelName(cfg.OSMMON_LOG_LEVEL))
 log = logging.getLogger(__name__)
 
-sys.path.append("/root/MON")
+sys.path.append(os.path.abspath(os.path.join(os.path.realpath(__file__), '..', '..', '..', '..', '..')))
 
-from core.message_bus.producer import KafkaProducer
+from osm_mon.core.database import DatabaseManager
+from osm_mon.core.message_bus.producer import KafkaProducer
 
-from plugins.OpenStack.Aodh.alarming import Alarming
-from plugins.OpenStack.common import Common
-from plugins.OpenStack.response import OpenStack_Response
-from plugins.OpenStack.settings import Config
+from osm_mon.plugins.OpenStack.response import OpenStack_Response
 
 
 class NotifierHandler(BaseHTTPRequestHandler):
@@ -62,7 +62,6 @@ class NotifierHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         """Get request functionality."""
         self._set_headers()
-        self.wfile.write("<html><body><h1>hi!</h1></body></html>")
 
     def do_POST(self):
         """POST request function."""
@@ -71,61 +70,53 @@ class NotifierHandler(BaseHTTPRequestHandler):
         # Gets the size of data
         content_length = int(self.headers['Content-Length'])
         post_data = self.rfile.read(content_length)
-        self.wfile.write("<html><body><h1>POST!</h1></body></tml>")
-        log.info("This alarm was triggered: %s", json.loads(post_data))
-
-        # Generate a notify_alarm response for the SO
-        self.notify_alarm(json.loads(post_data))
+        # Python 2/3 string compatibility
+        try:
+            post_data = post_data.decode()
+        except AttributeError:
+            pass
+        log.info("This alarm was triggered: %s", post_data)
+
+        # Send alarm notification to message bus
+        try:
+            self.notify_alarm(json.loads(post_data))
+        except Exception:
+            log.exception("Error notifying alarm")
 
     def notify_alarm(self, values):
-        """Send a notifcation repsonse message to the SO."""
-        # Initialiase configuration and authentication for response message
-        config = Config.instance()
-        config.read_environ("aodh")
-        self._alarming = Alarming()
-        self._common = Common()
-        self._response = OpenStack_Response()
-        self._producer = KafkaProducer('alarm_response')
+        """Sends alarm notification message to bus."""
+
+        # Initialise configuration and authentication for response message
+        response = OpenStack_Response()
+        producer = KafkaProducer('alarm_response')
+
+        database_manager = DatabaseManager()
 
         alarm_id = values['alarm_id']
-        auth_token = self._common._authenticate()
-        endpoint = self._common.get_endpoint("alarming")
-
-        # If authenticated generate and send response message
-        if (auth_token is not None and endpoint is not None):
-            url = "{}/v2/alarms/%s".format(endpoint) % alarm_id
-
-            # Get the resource_id of the triggered alarm
-            result = self._common._perform_request(
-                url, auth_token, req_type="get")
-            alarm_details = json.loads(result.text)
-            gnocchi_rule = alarm_details['gnocchi_resources_threshold_rule']
-            resource_id = gnocchi_rule['resource_id']
-
-            # Process an alarm notification if resource_id is valid
-            if resource_id is not None:
-                # Get date and time for response message
-                a_date = time.strftime("%d-%m-%Y") + " " + time.strftime("%X")
-                # Try generate and send response
-                try:
-                    resp_message = self._response.generate_response(
-                        'notify_alarm', a_id=alarm_id,
-                        r_id=resource_id,
-                        sev=values['severity'], date=a_date,
-                        state=values['current'], vim_type="OpenStack")
-                    self._producer.notify_alarm(
-                        'notify_alarm', resp_message, 'alarm_response')
-                    log.info("Sent an alarm response to SO: %s", resp_message)
-                except Exception as exc:
-                    log.warn("Couldn't notify SO of the alarm: %s", exc)
-            else:
-                log.warn("No resource_id for alarm; no SO response sent.")
-        else:
-            log.warn("Authentication failure; SO notification not sent.")
+        alarm = database_manager.get_alarm(alarm_id, 'openstack')
+        # Process an alarm notification if resource_id is valid
+        # Get date and time for response message
+        a_date = time.strftime("%d-%m-%Y") + " " + time.strftime("%X")
+        # Generate and send response
+        resp_message = response.generate_response(
+            'notify_alarm',
+            a_id=alarm_id,
+            vdu_name=alarm.vdu_name,
+            vnf_member_index=alarm.vnf_member_index,
+            ns_id=alarm.ns_id,
+            metric_name=alarm.metric_name,
+            operation=alarm.operation,
+            threshold_value=alarm.threshold,
+            sev=values['severity'],
+            date=a_date,
+            state=values['current'])
+        producer.publish_alarm_response(
+            'notify_alarm', resp_message)
+        log.info("Sent alarm notification: %s", resp_message)
 
 
 def run(server_class=HTTPServer, handler_class=NotifierHandler, port=8662):
-    """Run the webserver application to retreive alarm notifications."""
+    """Run the webserver application to retrieve alarm notifications."""
     try:
         server_address = ('', port)
         httpd = server_class(server_address, handler_class)
@@ -133,7 +124,8 @@ def run(server_class=HTTPServer, handler_class=NotifierHandler, port=8662):
         log.info("Starting alarm notifier server on port: %s", port)
         httpd.serve_forever()
     except Exception as exc:
-        log.warn("Failed to start webserver, %s", exc)
+        log.warning("Failed to start webserver, %s", exc)
+
 
 if __name__ == "__main__":
     from sys import argv