import threading
import asyncio
from http import HTTPStatus
+
from osm_common import dbmongo, dbmemory, msglocal, msgkafka
from osm_common.dbbase import DbException
from osm_common.msgbase import MsgException
from osm_nbi.engine import EngineException
+from osm_nbi.notifications import NsLcmNotification
__author__ = "Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>"
"public": None,
"method": "delete",
}
+ self.nslcm = None
async def start_kafka(self):
# timeout_wait_for_kafka = 3*60
if not kafka_working:
self.logger.critical("kafka is working again")
kafka_working = True
- if not self.aiomain_task_admin or self.aiomain_task_admin._state == "FINISHED":
+ if not self.aiomain_task_admin:
await asyncio.sleep(10, loop=self.loop)
self.logger.debug("Starting admin subscription task")
self.aiomain_task_admin = asyncio.ensure_future(self.msg.aioread(("admin",), loop=self.loop,
group_id=False,
aiocallback=self._msg_callback),
loop=self.loop)
- if not self.aiomain_task or self.aiomain_task._state == "FINISHED":
+ if not self.aiomain_task:
await asyncio.sleep(10, loop=self.loop)
self.logger.debug("Starting non-admin subscription task")
self.aiomain_task = asyncio.ensure_future(self.msg.aioread(("ns", "nsi"), loop=self.loop,
aiocallback=self._msg_callback),
loop=self.loop)
- await asyncio.wait([self.aiomain_task, self.aiomain_task_admin],
- timeout=None, loop=self.loop, return_when=asyncio.FIRST_COMPLETED)
+ done, _ = await asyncio.wait([self.aiomain_task, self.aiomain_task_admin],
+ timeout=None, loop=self.loop, return_when=asyncio.FIRST_COMPLETED)
+ try:
+ if self.aiomain_task_admin in done:
+ exc = self.aiomain_task_admin.exception()
+ self.logger.error("admin subscription task exception: {}".format(exc))
+ self.aiomain_task_admin = None
+ if self.aiomain_task in done:
+ exc = self.aiomain_task.exception()
+ self.logger.error("non-admin subscription task exception: {}".format(exc))
+ self.aiomain_task = None
+ except asyncio.CancelledError:
+ pass
except Exception as e:
if self.to_terminate:
return
else:
raise SubscriptionException("Invalid configuration param '{}' at '[message]':'driver'".format(
config_msg["driver"]))
-
+ self.nslcm = NsLcmNotification(self.db)
except (DbException, MsgException) as e:
raise SubscriptionException(str(e), http_code=e.http_code)
self.engine.del_item(self.internal_session, "nsrs", _id=params["nsr_id"],
not_send_msg=msg_to_send)
self.logger.debug("ns={} deleted from database".format(params["nsr_id"]))
+ # Check for nslcm notification
+ if isinstance(params, dict):
+ # Check availability of operationState and command
+ if (not params.get("operationState")) or (not command) or (not params.get("operationParams")):
+ self.logger.debug("Message can not be used for notification of nslcm")
+ else:
+ nsd_id = params["operationParams"].get("nsdId")
+ ns_instance_id = params["operationParams"].get("nsInstanceId")
+ # Any one among nsd_id, ns_instance_id should be present.
+ if not (nsd_id or ns_instance_id):
+ self.logger.debug("Message can not be used for notification of nslcm")
+ else:
+ op_state = params["operationState"]
+ event_details = {"topic": topic, "command": command.upper(), "params": params}
+ subscribers = self.nslcm.get_subscribers(nsd_id, ns_instance_id, command.upper(), op_state,
+ event_details)
+ # self.logger.debug("subscribers list: ")
+ # self.logger.debug(subscribers)
+ asyncio.ensure_future(self.nslcm.send_notifications(subscribers, loop=self.loop))
+ else:
+ self.logger.debug("Message can not be used for notification of nslcm")
elif topic == "nsi":
if command == "terminated" and params["operationState"] in ("COMPLETED", "PARTIALLY_COMPLETED"):
self.logger.debug("received nsi terminated {}".format(params))
self.to_terminate = True
if self.aiomain_task:
self.loop.call_soon_threadsafe(self.aiomain_task.cancel)
+ if self.aiomain_task_admin:
+ self.loop.call_soon_threadsafe(self.aiomain_task_admin.cancel)