class MonClient:
def __init__(self, config: Config, loop=None):
- self.kafka_server = '{}:{}'.format(config.get('message', 'host'),
- config.get('message', 'port'))
+ self.kafka_server = "{}:{}".format(
+ config.get("message", "host"), config.get("message", "port")
+ )
if not loop:
loop = asyncio.get_event_loop()
self.loop = loop
- async def create_alarm(self, metric_name: str, ns_id: str, vdu_name: str, vnf_member_index: int, threshold: int,
- statistic: str, operation: str):
+ async def create_alarm(
+ self,
+ metric_name: str,
+ ns_id: str,
+ vdu_name: str,
+ vnf_member_index: str,
+ threshold: int,
+ operation: str,
+ statistic: str = "AVERAGE",
+ action: str = "",
+ ):
cor_id = random.randint(1, 10e7)
- msg = self._build_create_alarm_payload(cor_id,
- metric_name,
- ns_id,
- vdu_name,
- vnf_member_index,
- threshold,
- statistic,
- operation)
+ msg = self._build_create_alarm_payload(
+ cor_id,
+ metric_name,
+ ns_id,
+ vdu_name,
+ vnf_member_index,
+ threshold,
+ statistic,
+ operation,
+ action,
+ )
log.debug("Sending create_alarm_request %s", msg)
- producer = AIOKafkaProducer(loop=self.loop,
- bootstrap_servers=self.kafka_server,
- key_serializer=str.encode,
- value_serializer=str.encode)
+ producer = AIOKafkaProducer(
+ loop=self.loop,
+ bootstrap_servers=self.kafka_server,
+ key_serializer=str.encode,
+ value_serializer=str.encode,
+ )
await producer.start()
try:
- await producer.send_and_wait("alarm_request", key="create_alarm_request", value=json.dumps(msg))
+ await producer.send_and_wait(
+ "alarm_request", key="create_alarm_request", value=json.dumps(msg)
+ )
finally:
await producer.stop()
log.debug("Waiting for create_alarm_response...")
bootstrap_servers=self.kafka_server,
key_deserializer=bytes.decode,
value_deserializer=bytes.decode,
- auto_offset_reset='earliest')
+ auto_offset_reset="earliest",
+ )
await consumer.start()
alarm_uuid = None
try:
except JSONDecodeError:
content = yaml.safe_load(message.value)
log.debug("Received create_alarm_response %s", content)
- if content['alarm_create_response']['correlation_id'] == cor_id:
- if not content['alarm_create_response']['status']:
+ if content["alarm_create_response"]["correlation_id"] == cor_id:
+ if not content["alarm_create_response"]["status"]:
raise ValueError("Error creating alarm in MON")
- alarm_uuid = content['alarm_create_response']['alarm_uuid']
+ alarm_uuid = content["alarm_create_response"]["alarm_uuid"]
break
finally:
await consumer.stop()
if not alarm_uuid:
- raise ValueError('No alarm deletion response from MON. Is MON up?')
+ raise ValueError("No alarm deletion response from MON. Is MON up?")
return alarm_uuid
- async def delete_alarm(self, ns_id: str, vnf_member_index: int, vdu_name: str, alarm_uuid: str):
+ async def delete_alarm(
+ self, ns_id: str, vnf_member_index: str, vdu_name: str, alarm_uuid: str
+ ):
cor_id = random.randint(1, 10e7)
- msg = self._build_delete_alarm_payload(cor_id, ns_id, vdu_name, vnf_member_index, alarm_uuid)
+ msg = self._build_delete_alarm_payload(
+ cor_id, ns_id, vdu_name, vnf_member_index, alarm_uuid
+ )
log.debug("Sending delete_alarm_request %s", msg)
- producer = AIOKafkaProducer(loop=self.loop,
- bootstrap_servers=self.kafka_server,
- key_serializer=str.encode,
- value_serializer=str.encode)
+ producer = AIOKafkaProducer(
+ loop=self.loop,
+ bootstrap_servers=self.kafka_server,
+ key_serializer=str.encode,
+ value_serializer=str.encode,
+ )
await producer.start()
try:
- await producer.send_and_wait("alarm_request", key="delete_alarm_request", value=json.dumps(msg))
+ await producer.send_and_wait(
+ "alarm_request", key="delete_alarm_request", value=json.dumps(msg)
+ )
finally:
await producer.stop()
log.debug("Waiting for delete_alarm_response...")
bootstrap_servers=self.kafka_server,
key_deserializer=bytes.decode,
value_deserializer=bytes.decode,
- auto_offset_reset='earliest')
+ auto_offset_reset="earliest",
+ )
await consumer.start()
alarm_uuid = None
try:
content = json.loads(message.value)
except JSONDecodeError:
content = yaml.safe_load(message.value)
- if content['alarm_delete_response']['correlation_id'] == cor_id:
+ if content["alarm_delete_response"]["correlation_id"] == cor_id:
log.debug("Received delete_alarm_response %s", content)
- if not content['alarm_delete_response']['status']:
- raise ValueError("Error deleting alarm in MON. Response status is False.")
- alarm_uuid = content['alarm_delete_response']['alarm_uuid']
+ if not content["alarm_delete_response"]["status"]:
+ raise ValueError(
+ "Error deleting alarm in MON. Response status is False."
+ )
+ alarm_uuid = content["alarm_delete_response"]["alarm_uuid"]
break
finally:
await consumer.stop()
if not alarm_uuid:
- raise ValueError('No alarm deletion response from MON. Is MON up?')
+ raise ValueError("No alarm deletion response from MON. Is MON up?")
return alarm_uuid
- def _build_create_alarm_payload(self, cor_id: int,
- metric_name: str,
- ns_id: str,
- vdu_name: str,
- vnf_member_index: int,
- threshold: int,
- statistic: str,
- operation: str):
+ def _build_create_alarm_payload(
+ self,
+ cor_id: int,
+ metric_name: str,
+ ns_id: str,
+ vdu_name: str,
+ vnf_member_index: str,
+ threshold: int,
+ statistic: str,
+ operation: str,
+ action: str,
+ ):
alarm_create_request = {
- 'correlation_id': cor_id,
- 'alarm_name': 'osm_alarm_{}_{}_{}_{}'.format(ns_id, vnf_member_index, vdu_name, metric_name),
- 'metric_name': metric_name,
- 'ns_id': ns_id,
- 'vdu_name': vdu_name,
- 'vnf_member_index': vnf_member_index,
- 'operation': operation,
- 'severity': 'critical',
- 'threshold_value': threshold,
- 'statistic': statistic
+ "correlation_id": cor_id,
+ "alarm_name": "osm_alarm_{}_{}_{}_{}".format(
+ ns_id, vnf_member_index, vdu_name, metric_name
+ ),
+ "metric_name": metric_name,
+ "operation": operation,
+ "severity": "critical",
+ "threshold_value": threshold,
+ "statistic": statistic,
+ "action": action,
+ "tags": {
+ "ns_id": ns_id,
+ "vdu_name": vdu_name,
+ "vnf_member_index": vnf_member_index,
+ },
}
msg = {
- 'alarm_create_request': alarm_create_request,
+ "alarm_create_request": alarm_create_request,
}
return msg
- def _build_delete_alarm_payload(self, cor_id: int, ns_id: str, vdu_name: str,
- vnf_member_index: int, alarm_uuid: str):
+ def _build_delete_alarm_payload(
+ self,
+ cor_id: int,
+ ns_id: str,
+ vdu_name: str,
+ vnf_member_index: str,
+ alarm_uuid: str,
+ ):
alarm_delete_request = {
- 'correlation_id': cor_id,
- 'alarm_uuid': alarm_uuid,
- 'ns_id': ns_id,
- 'vdu_name': vdu_name,
- 'vnf_member_index': vnf_member_index
+ "correlation_id": cor_id,
+ "alarm_uuid": alarm_uuid,
+ "tags": {
+ "ns_id": ns_id,
+ "vdu_name": vdu_name,
+ "vnf_member_index": vnf_member_index,
+ },
}
msg = {
- 'alarm_delete_request': alarm_delete_request,
+ "alarm_delete_request": alarm_delete_request,
}
return msg