1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
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
13 # http://www.apache.org/licenses/LICENSE-2.0
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
20 # For those usages not covered by the Apache License, Version 2.0 please
21 # contact: bdiaz@whitestack.com or glavado@whitestack.com
24 MON component in charge of CRUD operations for vim_accounts and alarms. It uses the message bus to communicate.
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
36 log
= logging
.getLogger(__name__
)
40 def __init__(self
, config
: Config
):
42 self
.msg_bus
= MessageBusClient(config
)
43 self
.service
= ServerService(config
)
44 self
.service
.populate_prometheus()
47 asyncio
.run(self
.start())
49 async def start(self
, wait_time
=5):
50 topics
= ["alarm_request"]
53 await self
.msg_bus
.aioread(topics
, self
._process
_msg
)
55 except Exception as e
:
56 # Failed to subscribe to kafka topic
57 log
.error("Error when subscribing to topic(s) %s", str(topics
))
58 log
.exception("Exception %s", str(e
))
59 # Wait for some time for kaka to stabilize and then reattempt to subscribe again
61 log
.info("Retrying to subscribe the kafka topic(s) %s", str(topics
))
63 async def _process_msg(self
, topic
, key
, values
):
64 log
.info("Message arrived: %s", values
)
66 if topic
== "alarm_request":
67 if key
== "create_alarm_request":
68 alarm_details
= values
["alarm_create_request"]
69 cor_id
= alarm_details
["correlation_id"]
70 response_builder
= ResponseBuilder()
72 alarm
= self
.service
.create_alarm(
73 alarm_details
["alarm_name"],
74 alarm_details
["threshold_value"],
75 alarm_details
["operation"].lower(),
76 alarm_details
["severity"].lower(),
77 alarm_details
["statistic"].lower(),
78 alarm_details
["metric_name"],
79 alarm_details
["action"],
80 alarm_details
["tags"],
82 response
= response_builder
.generate_response(
83 "create_alarm_response",
89 log
.exception("Error creating alarm: ")
90 response
= response_builder
.generate_response(
91 "create_alarm_response",
96 await self
._publish
_response
(
97 "alarm_response_" + str(cor_id
),
98 "create_alarm_response",
102 if key
== "delete_alarm_request":
103 alarm_details
= values
["alarm_delete_request"]
104 alarm_uuid
= alarm_details
["alarm_uuid"]
105 response_builder
= ResponseBuilder()
106 cor_id
= alarm_details
["correlation_id"]
108 self
.service
.delete_alarm(alarm_uuid
)
109 response
= response_builder
.generate_response(
110 "delete_alarm_response",
116 log
.exception("Error deleting alarm: ")
117 response
= response_builder
.generate_response(
118 "delete_alarm_response",
123 await self
._publish
_response
(
124 "alarm_response_" + str(cor_id
),
125 "delete_alarm_response",
130 log
.exception("Exception processing message: ")
132 async def _publish_response(self
, topic
: str, key
: str, msg
: dict):
134 "Sending response %s to topic %s with key %s", json
.dumps(msg
), topic
, key
136 await self
.msg_bus
.aiowrite(topic
, key
, msg
)