X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Fcore%2Fagent.py;h=95cc83073cb8d63ee582f92cc9d1b3251c847fc4;hb=8537aea539c90c284f2d9a303533950e3853d1da;hp=350f383af86df8c4b22d4438c651a128032fbcf1;hpb=f7451f8879d188c9fef9c976d82aa39141601938;p=osm%2FPOL.git diff --git a/osm_policy_module/core/agent.py b/osm_policy_module/core/agent.py index 350f383..95cc830 100644 --- a/osm_policy_module/core/agent.py +++ b/osm_policy_module/core/agent.py @@ -23,17 +23,20 @@ ## import asyncio import logging +from pathlib import Path +import os import peewee from osm_policy_module.alarming.service import AlarmingService from osm_policy_module.autoscaling.service import AutoscalingService +from osm_policy_module.common.common_db_client import CommonDbClient from osm_policy_module.common.message_bus_client import MessageBusClient from osm_policy_module.core.config import Config log = logging.getLogger(__name__) -ALLOWED_KAFKA_KEYS = ['instantiated', 'scaled', 'terminated', 'notify_alarm'] +ALLOWED_KAFKA_KEYS = ["instantiated", "scaled", "terminated", "notify_alarm"] class PolicyModuleAgent: @@ -43,6 +46,7 @@ class PolicyModuleAgent: loop = asyncio.get_event_loop() self.loop = loop self.msg_bus = MessageBusClient(config) + self.db_client = CommonDbClient(config) self.autoscaling_service = AutoscalingService(config, loop) self.alarming_service = AlarmingService(config, loop) @@ -50,28 +54,29 @@ class PolicyModuleAgent: self.loop.run_until_complete(self.start()) async def start(self): - topics = [ - "ns", - "alarm_response" - ] + Path("/tmp/osm_pol_agent_health_flag").touch() + topics = ["ns", "alarm_response"] await self.msg_bus.aioread(topics, self._process_msg) log.critical("Exiting...") + if os.path.exists("/tmp/osm_pol_agent_health_flag"): + os.remove("/tmp/osm_pol_agent_health_flag") async def _process_msg(self, topic, key, msg): + Path("/tmp/osm_pol_agent_health_flag").touch() log.debug("_process_msg topic=%s key=%s msg=%s", topic, key, msg) try: if key in ALLOWED_KAFKA_KEYS: - if key == 'instantiated': + if key == "instantiated": await self._handle_instantiated(msg) - if key == 'scaled': + if key == "scaled": await self._handle_scaled(msg) - if key == 'terminated': + if key == "terminated": await self._handle_terminated(msg) - if key == 'notify_alarm': + if key == "notify_alarm": await self._handle_alarm_notification(msg) else: log.debug("Key %s is not in ALLOWED_KAFKA_KEYS", key) @@ -83,17 +88,20 @@ class PolicyModuleAgent: async def _handle_alarm_notification(self, content): log.debug("_handle_alarm_notification: %s", content) - alarm_uuid = content['notify_details']['alarm_uuid'] - status = content['notify_details']['status'] + alarm_uuid = content["notify_details"]["alarm_uuid"] + status = content["notify_details"]["status"] await self.autoscaling_service.handle_alarm(alarm_uuid, status) await self.alarming_service.handle_alarm(alarm_uuid, status, content) async def _handle_instantiated(self, content): log.debug("_handle_instantiated: %s", content) - nslcmop_id = content['nslcmop_id'] - nslcmop = self.autoscaling_service.get_nslcmop(nslcmop_id) - if nslcmop['operationState'] == 'COMPLETED' or nslcmop['operationState'] == 'PARTIALLY_COMPLETED': - nsr_id = nslcmop['nsInstanceId'] + nslcmop_id = content["nslcmop_id"] + nslcmop = self.db_client.get_nslcmop(nslcmop_id) + if ( + nslcmop["operationState"] == "COMPLETED" + or nslcmop["operationState"] == "PARTIALLY_COMPLETED" + ): + nsr_id = nslcmop["nsInstanceId"] log.info("Configuring nsr_id: %s", nsr_id) await self.autoscaling_service.configure_scaling_groups(nsr_id) await self.alarming_service.configure_vnf_alarms(nsr_id) @@ -101,14 +109,18 @@ class PolicyModuleAgent: log.info( "Network_service is not in COMPLETED or PARTIALLY_COMPLETED state. " "Current state is %s. Skipping...", - nslcmop['operationState']) + nslcmop["operationState"], + ) async def _handle_scaled(self, content): log.debug("_handle_scaled: %s", content) - nslcmop_id = content['nslcmop_id'] - nslcmop = self.autoscaling_service.get_nslcmop(nslcmop_id) - if nslcmop['operationState'] == 'COMPLETED' or nslcmop['operationState'] == 'PARTIALLY_COMPLETED': - nsr_id = nslcmop['nsInstanceId'] + nslcmop_id = content["nslcmop_id"] + nslcmop = self.db_client.get_nslcmop(nslcmop_id) + if ( + nslcmop["operationState"] == "COMPLETED" + or nslcmop["operationState"] == "PARTIALLY_COMPLETED" + ): + nsr_id = nslcmop["nsInstanceId"] log.info("Configuring scaled service with nsr_id: %s", nsr_id) await self.autoscaling_service.configure_scaling_groups(nsr_id) await self.autoscaling_service.delete_orphaned_alarms(nsr_id) @@ -117,17 +129,25 @@ class PolicyModuleAgent: log.debug( "Network service is not in COMPLETED or PARTIALLY_COMPLETED state. " "Current state is %s. Skipping...", - nslcmop['operationState']) + nslcmop["operationState"], + ) async def _handle_terminated(self, content): log.debug("_handle_deleted: %s", content) - nsr_id = content['nsr_id'] - if content['operationState'] == 'COMPLETED' or content['operationState'] == 'PARTIALLY_COMPLETED': - log.info("Deleting scaling groups and alarms for network autoscaling_service with nsr_id: %s", nsr_id) + nsr_id = content["nsr_id"] + if ( + content["operationState"] == "COMPLETED" + or content["operationState"] == "PARTIALLY_COMPLETED" + ): + log.info( + "Deleting scaling groups and alarms for network autoscaling_service with nsr_id: %s", + nsr_id, + ) await self.autoscaling_service.delete_scaling_groups(nsr_id) await self.alarming_service.delete_vnf_alarms(nsr_id) else: log.info( "Network service is not in COMPLETED or PARTIALLY_COMPLETED state. " "Current state is %s. Skipping...", - content['operationState']) + content["operationState"], + )