osm_mon/server/service.py

   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 ##
  23 import logging
  24 import uuid
  25
  26 from osm_mon.core import database
  27 from osm_mon.core.common_db import CommonDbClient
  28 from osm_mon.core.config import Config
  29 from osm_mon.core.database import AlarmRepository, Alarm, AlarmTagRepository
  30
  31 log = logging.getLogger(__name__)
  32
  33
  34 class ServerService:
  35
  36     def __init__(self, config: Config):
  37         self.common_db = CommonDbClient(config)
  38
  39     def create_alarm(self,
  40                      name: str,
  41                      threshold: str,
  42                      operation: str,
  43                      severity: str,
  44                      statistic: str,
  45                      metric_name: str,
  46                      tags: dict) -> Alarm:
  47         database.db.connect()
  48         try:
  49             with database.db.atomic():
  50                 alarm = AlarmRepository.create(
  51                     uuid=str(uuid.uuid4()),
  52                     name=name,
  53                     threshold=threshold,
  54                     operation=operation.lower(),
  55                     severity=severity.lower(),
  56                     statistic=statistic.lower(),
  57                     metric=metric_name
  58                 )
  59                 for k, v in tags.items():
  60                     AlarmTagRepository.create(
  61                         name=k,
  62                         value=v,
  63                         alarm=alarm
  64                     )
  65                 return alarm
  66
  67         finally:
  68             database.db.close()
  69
  70     def delete_alarm(self,
  71                      alarm_uuid: str) -> None:
  72         database.db.connect()
  73         try:
  74             with database.db.atomic():
  75                 alarm = AlarmRepository.get(Alarm.uuid == alarm_uuid)
  76                 alarm.delete_instance()
  77         finally:
  78             database.db.close()