osm_mon/server/server.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright 2018 Whitestack, LLC
   4 # *************************************************************
   5
   6 # This file is part of OSM Monitoring module
   7 # All Rights Reserved to Whitestack, LLC
   8
   9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
  10 # not use this file except in compliance with the License. You may obtain
  11 # a copy of the License at
  12
  13 #         http://www.apache.org/licenses/LICENSE-2.0
  14
  15 # Unless required by applicable law or agreed to in writing, software
  16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  18 # License for the specific language governing permissions and limitations
  19 # under the License.
  20 # For those usages not covered by the Apache License, Version 2.0 please
  21 # contact: bdiaz@whitestack.com or glavado@whitestack.com
  22 ##
  23 """
  24 MON component in charge of CRUD operations for vim_accounts and alarms. It uses the message bus to communicate.
  25 """
  26 import asyncio
  27 import json
  28 import logging
  29 import time
  30
  31 from osm_mon.core.config import Config
  32 from osm_mon.core.message_bus_client import MessageBusClient
  33 from osm_mon.core.response import ResponseBuilder
  34 from osm_mon.server.service import ServerService
  35
  36 log = logging.getLogger(__name__)
  37
  38
  39 class Server:
  40     def __init__(self, config: Config, loop=None):
  41         self.conf = config
  42         if not loop:
  43             loop = asyncio.get_event_loop()
  44         self.loop = loop
  45         self.msg_bus = MessageBusClient(config)
  46         self.service = ServerService(config)
  47         self.service.populate_prometheus()
  48
  49     def run(self):
  50         self.loop.run_until_complete(self.start())
  51
  52     async def start(self, wait_time=5):
  53         topics = ["alarm_request"]
  54         while True:
  55             try:
  56                 await self.msg_bus.aioread(topics, self._process_msg)
  57                 log.info("Sucessfully subscribed to kafka topic(s) %s", str(topics))
  58                 break
  59             except Exception as e:
  60                 # Failed to subscribe to kafka topic
  61                 log.error("Error when subscribing to topic(s) %s", str(topics))
  62                 log.exception("Exception %s", str(e))
  63                 # Wait for some time for kaka to stabilize and then reattempt to subscribe again
  64                 time.sleep(wait_time)
  65                 log.info("Retrying to subscribe the kafka topic(s) %s", str(topics))
  66
  67     async def _process_msg(self, topic, key, values):
  68         log.info("Message arrived: %s", values)
  69         try:
  70
  71             if topic == "alarm_request":
  72                 if key == "create_alarm_request":
  73                     alarm_details = values["alarm_create_request"]
  74                     cor_id = alarm_details["correlation_id"]
  75                     response_builder = ResponseBuilder()
  76                     try:
  77                         alarm = self.service.create_alarm(
  78                             alarm_details["alarm_name"],
  79                             alarm_details["threshold_value"],
  80                             alarm_details["operation"].lower(),
  81                             alarm_details["severity"].lower(),
  82                             alarm_details["statistic"].lower(),
  83                             alarm_details["metric_name"],
  84                             alarm_details["action"],
  85                             alarm_details["tags"],
  86                         )
  87                         response = response_builder.generate_response(
  88                             "create_alarm_response",
  89                             cor_id=cor_id,
  90                             status=True,
  91                             alarm_id=alarm.uuid,
  92                         )
  93                     except Exception:
  94                         log.exception("Error creating alarm: ")
  95                         response = response_builder.generate_response(
  96                             "create_alarm_response",
  97                             cor_id=cor_id,
  98                             status=False,
  99                             alarm_id=None,
 100                         )
 101                     await self._publish_response(
 102                         "alarm_response_" + str(cor_id),
 103                         "create_alarm_response",
 104                         response,
 105                     )
 106
 107                 if key == "delete_alarm_request":
 108                     alarm_details = values["alarm_delete_request"]
 109                     alarm_uuid = alarm_details["alarm_uuid"]
 110                     response_builder = ResponseBuilder()
 111                     cor_id = alarm_details["correlation_id"]
 112                     try:
 113                         self.service.delete_alarm(alarm_uuid)
 114                         response = response_builder.generate_response(
 115                             "delete_alarm_response",
 116                             cor_id=cor_id,
 117                             status=True,
 118                             alarm_id=alarm_uuid,
 119                         )
 120                     except Exception:
 121                         log.exception("Error deleting alarm: ")
 122                         response = response_builder.generate_response(
 123                             "delete_alarm_response",
 124                             cor_id=cor_id,
 125                             status=False,
 126                             alarm_id=alarm_uuid,
 127                         )
 128                     await self._publish_response(
 129                         "alarm_response_" + str(cor_id),
 130                         "delete_alarm_response",
 131                         response,
 132                     )
 133
 134         except Exception:
 135             log.exception("Exception processing message: ")
 136
 137     async def _publish_response(self, topic: str, key: str, msg: dict):
 138         log.info(
 139             "Sending response %s to topic %s with key %s", json.dumps(msg), topic, key
 140         )
 141         await self.msg_bus.aiowrite(topic, key, msg)