Code Coverage

Cobertura Coverage Report > osm_policy_module.alarming >

service.py

Trend

Classes100%
 
Lines56%
   
Conditionals100%
 

File Coverage summary

NameClassesLinesConditionals
service.py
100%
1/1
56%
71/127
100%
0/0

Coverage Breakdown by Class

NameLinesConditionals
service.py
56%
71/127
N/A

Source

osm_policy_module/alarming/service.py
1 # -*- coding: utf-8 -*-
2 # pylint: disable=no-member
3
4 # Copyright 2018 Whitestack, LLC
5 # *************************************************************
6
7 # This file is part of OSM Monitoring module
8 # All Rights Reserved to Whitestack, LLC
9
10 # Licensed under the Apache License, Version 2.0 (the "License"); you may
11 # not use this file except in compliance with the License. You may obtain
12 # a copy of the License at
13
14 #         http://www.apache.org/licenses/LICENSE-2.0
15
16 # Unless required by applicable law or agreed to in writing, software
17 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
18 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
19 # License for the specific language governing permissions and limitations
20 # under the License.
21
22 # For those usages not covered by the Apache License, Version 2.0 please
23 # contact: bdiaz@whitestack.com or glavado@whitestack.com
24 ##
25 1 import asyncio
26 1 import json
27 1 import logging
28
29 1 import requests
30 1 from requests.exceptions import ConnectionError
31
32 1 from osm_policy_module.common.common_db_client import CommonDbClient
33 1 from osm_policy_module.common.lcm_client import LcmClient
34 1 from osm_policy_module.common.mon_client import MonClient
35 1 from osm_policy_module.core import database
36 1 from osm_policy_module.core.config import Config
37 1 from osm_policy_module.core.database import VnfAlarm, VnfAlarmRepository, AlarmActionRepository
38 1 from osm_policy_module.core.exceptions import VdurNotFound
39
40 1 log = logging.getLogger(__name__)
41
42
43 1 class AlarmingService:
44
45 1     def __init__(self, config: Config, loop=None):
46 1         self.conf = config
47 1         if not loop:
48 1             loop = asyncio.get_event_loop()
49 1         self.loop = loop
50 1         self.db_client = CommonDbClient(config)
51 1         self.mon_client = MonClient(config, loop=self.loop)
52 1         self.lcm_client = LcmClient(config, loop=self.loop)
53
54 1     async def configure_vnf_alarms(self, nsr_id: str):
55 1         log.info("Configuring vnf alarms for network service %s", nsr_id)
56 1         alarms_created = []
57 1         database.db.connect()
58 1         try:
59 1             with database.db.atomic():
60 1                 vnfrs = self.db_client.get_vnfrs(nsr_id)
61 1                 for vnfr in vnfrs:
62 1                     log.debug("Processing vnfr: %s", vnfr)
63 1                     vnfd = self.db_client.get_vnfd(vnfr['vnfd-id'])
64 1                     for vdur in vnfr['vdur']:
65 1                         vdu = next(
66                             filter(
67                                 lambda vdu: vdu['id'] == vdur['vdu-id-ref'],
68                                 vnfd['vdu']
69                             )
70                         )
71 1                         if 'alarm' in vdu:
72 1                             alarm_descriptors = vdu['alarm']
73 1                             for alarm_descriptor in alarm_descriptors:
74 1                                 try:
75 1                                     VnfAlarmRepository.get(
76                                         VnfAlarm.alarm_id == alarm_descriptor['alarm-id'],
77                                         VnfAlarm.vnf_member_index == vnfr['member-vnf-index-ref'],
78                                         VnfAlarm.vdu_name == vdur['name'],
79                                         VnfAlarm.nsr_id == nsr_id
80                                     )
81 0                                     log.debug("vdu %s already has an alarm configured with same id %s", vdur['name'],
82                                               alarm_descriptor['alarm-id'])
83 0                                     continue
84 1                                 except VnfAlarm.DoesNotExist:
85 1                                     pass
86 1                                 vnf_monitoring_param = next(
87                                     filter(
88                                         lambda param: param['id'] == alarm_descriptor['vnf-monitoring-param-ref'],
89                                         vnfd['monitoring-param'])
90                                 )
91 1                                 metric_name = self._get_metric_name(vnf_monitoring_param, vdur, vnfd)
92 1                                 alarm_uuid = await self.mon_client.create_alarm(
93                                     metric_name=metric_name,
94                                     ns_id=nsr_id,
95                                     vdu_name=vdur['name'],
96                                     vnf_member_index=vnfr['member-vnf-index-ref'],
97                                     threshold=alarm_descriptor['value'],
98                                     operation=alarm_descriptor['operation'],
99                                     statistic=vnf_monitoring_param['aggregation-type']
100                                 )
101 1                                 alarm = VnfAlarmRepository.create(
102                                     alarm_id=alarm_descriptor['alarm-id'],
103                                     alarm_uuid=alarm_uuid,
104                                     nsr_id=nsr_id,
105                                     vnf_member_index=vnfr['member-vnf-index-ref'],
106                                     vdu_name=vdur['name']
107                                 )
108 1                                 for action_type in ['ok', 'insufficient-data', 'alarm']:
109 1                                     if 'actions' in alarm_descriptor and action_type in alarm_descriptor['actions']:
110 1                                         for url in alarm_descriptor['actions'][action_type]:
111 1                                             AlarmActionRepository.create(
112                                                 type=action_type,
113                                                 url=url['url'],
114                                                 alarm=alarm
115                                             )
116 1                                 alarms_created.append(alarm)
117
118 0         except Exception as e:
119 0             log.exception("Error configuring VNF alarms:")
120 0             if len(alarms_created) > 0:
121 0                 log.debug("Cleaning alarm resources in MON")
122 0                 for alarm in alarms_created:
123 0                     try:
124 0                         await self.mon_client.delete_alarm(alarm.nsr_id,
125                                                            alarm.vnf_member_index,
126                                                            alarm.vdu_name,
127                                                            alarm.alarm_uuid)
128 0                     except ValueError:
129 0                         log.exception("Error deleting alarm in MON %s", alarm.alarm_uuid)
130 0             raise e
131         finally:
132 1             database.db.close()
133
134 1     async def delete_orphaned_alarms(self, nsr_id):
135 0         log.info("Deleting orphaned vnf alarms for network service %s", nsr_id)
136 0         database.db.connect()
137 0         try:
138 0             with database.db.atomic():
139 0                 for alarm in VnfAlarmRepository.list(VnfAlarm.nsr_id == nsr_id):
140 0                     try:
141 0                         self.db_client.get_vdur(nsr_id, alarm.vnf_member_index, alarm.vdu_name)
142 0                     except VdurNotFound:
143 0                         log.debug("Deleting orphaned alarm %s", alarm.alarm_uuid)
144 0                         try:
145 0                             await self.mon_client.delete_alarm(
146                                 alarm.nsr_id,
147                                 alarm.vnf_member_index,
148                                 alarm.vdu_name,
149                                 alarm.alarm_uuid)
150 0                         except ValueError:
151 0                             log.exception("Error deleting alarm in MON %s", alarm.alarm_uuid)
152 0                         alarm.delete_instance()
153 0         except Exception as e:
154 0             log.exception("Error deleting orphaned alarms:")
155 0             raise e
156         finally:
157 0             database.db.close()
158
159 1     async def delete_vnf_alarms(self, nsr_id):
160 0         log.info("Deleting vnf alarms for network service %s", nsr_id)
161 0         database.db.connect()
162 0         try:
163 0             with database.db.atomic():
164 0                 for alarm in VnfAlarmRepository.list(VnfAlarm.nsr_id == nsr_id):
165 0                     log.debug("Deleting vnf alarm %s", alarm.alarm_uuid)
166 0                     try:
167 0                         await self.mon_client.delete_alarm(
168                             alarm.nsr_id,
169                             alarm.vnf_member_index,
170                             alarm.vdu_name,
171                             alarm.alarm_uuid)
172 0                     except ValueError:
173 0                         log.exception("Error deleting alarm in MON %s", alarm.alarm_uuid)
174 0                     alarm.delete_instance()
175
176 0         except Exception as e:
177 0             log.exception("Error deleting vnf alarms:")
178 0             raise e
179         finally:
180 0             database.db.close()
181
182 1     async def handle_alarm(self, alarm_uuid: str, status: str, payload: dict):
183 1         database.db.connect()
184 1         try:
185 1             with database.db.atomic():
186 1                 alarm = VnfAlarmRepository.get(VnfAlarm.alarm_uuid == alarm_uuid)
187 1                 log.debug("Handling vnf alarm %s with status %s", alarm.alarm_id, status)
188 1                 for action in alarm.actions:
189 1                     if action.type == status:
190 1                         log.info("Executing request to url %s for vnf alarm %s with status %s", action.url,
191                                  alarm.alarm_id, status)
192 1                         try:
193 1                             requests.post(url=action.url, json=json.dumps(payload))
194 0                         except ConnectionError:
195 0                             log.exception("Error connecting to url %s", action.url)
196
197 0         except VnfAlarm.DoesNotExist:
198 0             log.debug("There is no alarming action configured for alarm %s.", alarm_uuid)
199         finally:
200 1             database.db.close()
201
202 1     def _get_metric_name(self, vnf_monitoring_param: dict, vdur: dict, vnfd: dict):
203 1         vdu = next(
204             filter(lambda vdu: vdu['id'] == vdur['vdu-id-ref'], vnfd['vdu'])
205         )
206 1         if 'vdu-monitoring-param' in vnf_monitoring_param:
207 1             vdu_monitoring_param = next(filter(
208                 lambda param: param['id'] == vnf_monitoring_param['vdu-monitoring-param'][
209                     'vdu-monitoring-param-ref'], vdu['monitoring-param']))
210 1             nfvi_metric = vdu_monitoring_param['nfvi-metric']
211 1             return nfvi_metric
212 0         if 'vdu-metric' in vnf_monitoring_param:
213 0             vnf_metric_name = vnf_monitoring_param['vdu-metric']['vdu-metric-name-ref']
214 0             return vnf_metric_name
215 0         if 'vnf-metric' in vnf_monitoring_param:
216 0             vnf_metric_name = vnf_monitoring_param['vnf-metric']['vnf-metric-name-ref']
217 0             return vnf_metric_name
218 0         raise ValueError('No metric name found for vnf_monitoring_param %s' % vnf_monitoring_param['id'])