Coverage for osm_mon/server/service.py: 0%

1# -*- coding: utf-8 -*- 

2 

3# Copyright 2018 Whitestack, LLC 

4# ************************************************************* 

5 

6# This file is part of OSM Monitoring module 

7# All Rights Reserved to Whitestack, LLC 

8 

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 

12 

13# http://www.apache.org/licenses/LICENSE-2.0 

14 

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 

19# under the License. 

20# For those usages not covered by the Apache License, Version 2.0 please 

21# contact: bdiaz@whitestack.com or glavado@whitestack.com 

22## 

23import logging 

24 

25from osm_mon.core.common_db import CommonDbClient 

26from osm_mon.core.config import Config 

27from osm_mon.core.models import Alarm 

28 

29log = logging.getLogger(__name__) 

30 

31 

32class ServerService: 

33 def __init__(self, config: Config): 

34 self.common_db = CommonDbClient(config) 

35 

36 def create_alarm( 

37 self, 

38 name: str, 

39 threshold: float, 

40 operation: str, 

41 severity: str, 

42 statistic: str, 

43 metric_name: str, 

44 action: str, 

45 tags: dict, 

46 ) -> Alarm: 

47 log.debug("create_alarm") 

48 alarm = Alarm( 

49 name, severity, threshold, operation, statistic, metric_name, action, tags 

50 ) 

51 self.common_db.create_alarm(alarm) 

52 log.info("Alarm %s created", alarm.name) 

53 return alarm 

54 

55 def delete_alarm(self, alarm_uuid: str) -> None: 

56 log.debug("delete_alarm") 

57 self.common_db.delete_alarm(alarm_uuid) 

58 log.info("Alarm %s is deleted", alarm_uuid) 

59 

60 def populate_prometheus(self): 

61 # This method can be executed infinite number of times 

62 # without side-effects. 

63 log.info("Setting up MON Exporter in prometheus") 

64 result = self.common_db.common_db.set_one( 

65 "prometheus_jobs", 

66 {"job_name": "mon_exporter"}, 

67 {"job_name": "mon_exporter", "static_configs": [{"targets": ["mon:8000"]}]}, 

68 fail_on_empty=False, 

69 upsert=True, 

70 ) 

71 log.info("Prometheus Jobs added > {}".format(result))