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
  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                      vdur_name: str,
  47                      vnf_member_index: str,
  48                      nsr_id: str) -> Alarm:
  49         database.db.connect()
  50         try:
  51             with database.db.atomic():
  52                 return AlarmRepository.create(
  53                     uuid=str(uuid.uuid4()),
  54                     name=name,
  55                     threshold=threshold,
  56                     operation=operation.lower(),
  57                     severity=severity.lower(),
  58                     statistic=statistic.lower(),
  59                     monitoring_param=metric_name,
  60                     vdur_name=vdur_name,
  61                     vnf_member_index=vnf_member_index,
  62                     nsr_id=nsr_id
  63                 )
  64
  65         finally:
  66             database.db.close()
  67
  68     def delete_alarm(self,
  69                      alarm_uuid: str) -> None:
  70         database.db.connect()
  71         try:
  72             with database.db.atomic():
  73                 alarm = AlarmRepository.get(Alarm.uuid == alarm_uuid)
  74                 alarm.delete_instance()
  75         finally:
  76             database.db.close()