# For those usages not covered by the Apache License, Version 2.0 please
# contact: bdiaz@whitestack.com or glavado@whitestack.com
##
-"""A common KafkaConsumer for all MON plugins."""
+"""
+MON component in charge of CRUD operations for vim_accounts and alarms. It uses the message bus to communicate.
+"""
import asyncio
import json
import logging
+import time
-from osm_mon.core.auth import AuthManager
-from osm_mon.core.common_db import CommonDbClient
from osm_mon.core.config import Config
-from osm_mon.core.database import DatabaseManager
from osm_mon.core.message_bus_client import MessageBusClient
from osm_mon.core.response import ResponseBuilder
+from osm_mon.server.service import ServerService
log = logging.getLogger(__name__)
class Server:
-
def __init__(self, config: Config, loop=None):
self.conf = config
if not loop:
loop = asyncio.get_event_loop()
self.loop = loop
- self.auth_manager = AuthManager(config)
- self.database_manager = DatabaseManager(config)
- self.database_manager.create_tables()
- self.common_db = CommonDbClient(config)
self.msg_bus = MessageBusClient(config)
+ self.service = ServerService(config)
+ self.service.populate_prometheus()
def run(self):
self.loop.run_until_complete(self.start())
- async def start(self):
- topics = [
- "vim_account",
- "alarm_request"
- ]
- await self.msg_bus.aioread(topics, self._process_msg)
+ async def start(self, wait_time=5):
+ topics = ["alarm_request"]
+ while True:
+ try:
+ await self.msg_bus.aioread(topics, self._process_msg)
+ log.info("Sucessfully subscribed to kafka topic(s) %s", str(topics))
+ break
+ except Exception as e:
+ # Failed to subscribe to kafka topic
+ log.error("Error when subscribing to topic(s) %s", str(topics))
+ log.exception("Exception %s", str(e))
+ # Wait for some time for kaka to stabilize and then reattempt to subscribe again
+ time.sleep(wait_time)
+ log.info("Retrying to subscribe the kafka topic(s) %s", str(topics))
async def _process_msg(self, topic, key, values):
log.info("Message arrived: %s", values)
try:
- if topic == "vim_account":
- if key == "create" or key == "edit":
- values['vim_password'] = self.common_db.decrypt_vim_password(values['vim_password'],
- values['schema_version'],
- values['_id'])
-
- vim_config_encrypted = ("admin_password", "nsx_password", "vcenter_password")
- if 'config' in values:
- for key in values['config']:
- if key in vim_config_encrypted:
- values['config'][key] = self.common_db.decrypt_vim_password(values['config'][key],
- values['schema_version'],
- values['_id'])
- self.auth_manager.store_auth_credentials(values)
-
- if key == "delete":
- self.auth_manager.delete_auth_credentials(values)
-
- elif topic == "alarm_request":
+
+ if topic == "alarm_request":
if key == "create_alarm_request":
- alarm_details = values['alarm_create_request']
- cor_id = alarm_details['correlation_id']
+ alarm_details = values["alarm_create_request"]
+ cor_id = alarm_details["correlation_id"]
response_builder = ResponseBuilder()
try:
- alarm = self.database_manager.save_alarm(
- alarm_details['alarm_name'],
- alarm_details['threshold_value'],
- alarm_details['operation'].lower(),
- alarm_details['severity'].lower(),
- alarm_details['statistic'].lower(),
- alarm_details['metric_name'],
- alarm_details['vdu_name'],
- alarm_details['vnf_member_index'],
- alarm_details['ns_id']
+ alarm = self.service.create_alarm(
+ alarm_details["alarm_name"],
+ alarm_details["threshold_value"],
+ alarm_details["operation"].lower(),
+ alarm_details["severity"].lower(),
+ alarm_details["statistic"].lower(),
+ alarm_details["metric_name"],
+ alarm_details["action"],
+ alarm_details["tags"],
+ )
+ response = response_builder.generate_response(
+ "create_alarm_response",
+ cor_id=cor_id,
+ status=True,
+ alarm_id=alarm.uuid,
)
- response = response_builder.generate_response('create_alarm_response',
- cor_id=cor_id,
- status=True,
- alarm_id=alarm.uuid)
except Exception:
log.exception("Error creating alarm: ")
- response = response_builder.generate_response('create_alarm_response',
- cor_id=cor_id,
- status=False,
- alarm_id=None)
- await self._publish_response('alarm_response_' + str(cor_id), 'create_alarm_response', response)
+ response = response_builder.generate_response(
+ "create_alarm_response",
+ cor_id=cor_id,
+ status=False,
+ alarm_id=None,
+ )
+ await self._publish_response(
+ "alarm_response_" + str(cor_id),
+ "create_alarm_response",
+ response,
+ )
if key == "delete_alarm_request":
- alarm_details = values['alarm_delete_request']
- alarm_uuid = alarm_details['alarm_uuid']
+ alarm_details = values["alarm_delete_request"]
+ alarm_uuid = alarm_details["alarm_uuid"]
response_builder = ResponseBuilder()
- cor_id = alarm_details['correlation_id']
+ cor_id = alarm_details["correlation_id"]
try:
- self.database_manager.delete_alarm(alarm_uuid)
- response = response_builder.generate_response('delete_alarm_response',
- cor_id=cor_id,
- status=True,
- alarm_id=alarm_uuid)
+ self.service.delete_alarm(alarm_uuid)
+ response = response_builder.generate_response(
+ "delete_alarm_response",
+ cor_id=cor_id,
+ status=True,
+ alarm_id=alarm_uuid,
+ )
except Exception:
log.exception("Error deleting alarm: ")
- response = response_builder.generate_response('delete_alarm_response',
- cor_id=cor_id,
- status=False,
- alarm_id=alarm_uuid)
- await self._publish_response('alarm_response_' + str(cor_id), 'delete_alarm_response', response)
+ response = response_builder.generate_response(
+ "delete_alarm_response",
+ cor_id=cor_id,
+ status=False,
+ alarm_id=alarm_uuid,
+ )
+ await self._publish_response(
+ "alarm_response_" + str(cor_id),
+ "delete_alarm_response",
+ response,
+ )
except Exception:
log.exception("Exception processing message: ")
async def _publish_response(self, topic: str, key: str, msg: dict):
- log.info("Sending response %s to topic %s with key %s", json.dumps(msg), topic, key)
+ log.info(
+ "Sending response %s to topic %s with key %s", json.dumps(msg), topic, key
+ )
await self.msg_bus.aiowrite(topic, key, msg)