osm_mon/plugins/OpenStack/Aodh/notifier.py

   1 # Copyright 2017 Intel Research and Development Ireland Limited
   2 # *************************************************************
   3
   4 # This file is part of OSM Monitoring module
   5 # All Rights Reserved to Intel Corporation
   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: helena.mcgough@intel.com or adrian.hoban@intel.com
  21 ##
  22 # __author__ = Helena McGough
  23 #
  24 """A Webserver to send alarm notifications from Aodh to the SO."""
  25 import json
  26 import logging
  27 import os
  28 import sys
  29 import time
  30
  31 from six.moves.BaseHTTPServer import BaseHTTPRequestHandler
  32 from six.moves.BaseHTTPServer import HTTPServer
  33
  34 # Initialise a logger for alarm notifier
  35 from osm_mon.core.settings import Config
  36
  37 cfg = Config.instance()
  38
  39 logging.basicConfig(stream=sys.stdout,
  40                     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
  41                     datefmt='%m/%d/%Y %I:%M:%S %p',
  42                     level=logging.getLevelName(cfg.OSMMON_LOG_LEVEL))
  43 log = logging.getLogger(__name__)
  44
  45 kafka_logger = logging.getLogger('kafka')
  46 kafka_logger.setLevel(logging.getLevelName(cfg.OSMMON_KAFKA_LOG_LEVEL))
  47 kafka_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
  48 kafka_handler = logging.StreamHandler(sys.stdout)
  49 kafka_handler.setFormatter(kafka_formatter)
  50 kafka_logger.addHandler(kafka_handler)
  51
  52 sys.path.append(os.path.abspath(os.path.join(os.path.realpath(__file__), '..', '..', '..', '..', '..')))
  53
  54 from osm_mon.core.database import DatabaseManager
  55 from osm_mon.core.message_bus.producer import KafkaProducer
  56
  57 from osm_mon.plugins.OpenStack.response import OpenStack_Response
  58
  59
  60 class NotifierHandler(BaseHTTPRequestHandler):
  61     """Handler class for alarm_actions triggered by OSM alarms."""
  62
  63     def _set_headers(self):
  64         """Set the headers for a request."""
  65         self.send_response(200)
  66         self.send_header('Content-type', 'text/html')
  67         self.end_headers()
  68
  69     def do_GET(self):
  70         """Get request functionality."""
  71         self._set_headers()
  72
  73     def do_POST(self):
  74         """POST request function."""
  75         # Gets header and data from the post request and records info
  76         self._set_headers()
  77         # Gets the size of data
  78         content_length = int(self.headers['Content-Length'])
  79         post_data = self.rfile.read(content_length)
  80         # Python 2/3 string compatibility
  81         try:
  82             post_data = post_data.decode()
  83         except AttributeError:
  84             pass
  85         log.info("This alarm was triggered: %s", post_data)
  86
  87         # Send alarm notification to message bus
  88         try:
  89             self.notify_alarm(json.loads(post_data))
  90         except Exception:
  91             log.exception("Error notifying alarm")
  92
  93     def notify_alarm(self, values):
  94         """Sends alarm notification message to bus."""
  95
  96         # Initialise configuration and authentication for response message
  97         response = OpenStack_Response()
  98         producer = KafkaProducer('alarm_response')
  99
 100         database_manager = DatabaseManager()
 101
 102         alarm_id = values['alarm_id']
 103         alarm = database_manager.get_alarm(alarm_id, 'openstack')
 104         # Process an alarm notification if resource_id is valid
 105         # Get date and time for response message
 106         a_date = time.strftime("%d-%m-%Y") + " " + time.strftime("%X")
 107         # Generate and send response
 108         resp_message = response.generate_response(
 109             'notify_alarm',
 110             alarm_id=alarm_id,
 111             vdu_name=alarm.vdu_name,
 112             vnf_member_index=alarm.vnf_member_index,
 113             ns_id=alarm.ns_id,
 114             metric_name=alarm.metric_name,
 115             operation=alarm.operation,
 116             threshold_value=alarm.threshold,
 117             sev=values['severity'],
 118             date=a_date,
 119             state=values['current'])
 120         producer.publish_alarm_response(
 121             'notify_alarm', resp_message)
 122         log.info("Sent alarm notification: %s", resp_message)
 123
 124
 125 def run(server_class=HTTPServer, handler_class=NotifierHandler, port=8662):
 126     """Run the webserver application to retrieve alarm notifications."""
 127     try:
 128         server_address = ('', port)
 129         httpd = server_class(server_address, handler_class)
 130         print('Starting alarm notifier...')
 131         log.info("Starting alarm notifier server on port: %s", port)
 132         httpd.serve_forever()
 133     except Exception as exc:
 134         log.warning("Failed to start webserver, %s", exc)
 135
 136
 137 if __name__ == "__main__":
 138     from sys import argv
 139
 140     # Runs the webserver
 141     if len(argv) == 2:
 142         run(port=int(argv[1]))
 143     else:
 144         run()