-##
-# Copyright 2017 xFlow Research Pvt. Ltd
-# This file is part of MON module
-# All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License. You may obtain
-# a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
-# License for the specific language governing permissions and limitations
-# under the License.
-#
-# For those usages not covered by the Apache License, Version 2.0 please
-# contact with: wajeeha.hamid@xflowresearch.com
-##
-
-"""
-AWS-Plugin implements all the methods of MON to interact with AWS using the BOTO client
-"""
-from io import UnsupportedOperation
-
-from osm_mon.core.settings import Config
-from osm_mon.plugins.CloudWatch.metric_alarms import MetricAlarm
-from osm_mon.plugins.CloudWatch.metrics import Metrics
-
-__author__ = "Wajeeha Hamid"
-__date__ = "18-September-2017"
-
-import logging
-
-log = logging.getLogger(__name__)
-
-
-class plugin_alarms:
- """Receives Alarm info from MetricAlarm and connects with the consumer/producer"""
-
- def __init__(self):
- self._cfg = Config.instance()
- self.metricAlarm = MetricAlarm()
- self.metric = Metrics()
-
- def configure_alarm(self, alarm_info):
- alarm_id = self.metricAlarm.config_alarm(self.cloudwatch_conn, alarm_info)
- return alarm_id
-
- def update_alarm_configuration(self, test):
- alarm_id = self.metricAlarm.update_alarm(self.cloudwatch_conn, test)
- return alarm_id
-
- def delete_alarm(self, alarm_id):
- return self.metricAlarm.delete_Alarm(self.cloudwatch_conn, alarm_id)
-
- def get_alarms_list(self, instance_id):
- return self.metricAlarm.alarms_list(self.cloudwatch_conn, instance_id)
-
- def get_ack_details(self, ack_info):
- return self.metricAlarm.alarm_details(self.cloudwatch_conn, ack_info)
-
- def get_metrics_data(self, metric_name, period, instance_id):
- # TODO: Investigate and fix this call
- return self.metric.metricsData(self.cloudwatch_conn, metric_name, period, instance_id)
-
- def alarm_calls(self, key: str, alarm_info: dict, aws_conn: dict):
- """Gets the message from the common consumer"""
- try:
- self.cloudwatch_conn = aws_conn['cloudwatch_connection']
- self.ec2_conn = aws_conn['ec2_connection']
-
- if key == "create_alarm_request":
- alarm_inner_dict = alarm_info['alarm_create_request']
- metric_status = self.check_metric(alarm_inner_dict['metric_name'])
-
- if self.check_resource(alarm_inner_dict['resource_uuid']) and metric_status['status']:
- log.debug("Resource and Metrics exists")
-
- alarm_info['alarm_create_request']['metric_name'] = metric_status['metric_name']
- # Generate a valid response message, send via producer
- config_resp = self.configure_alarm(alarm_info) # alarm_info = message.value
-
- if config_resp is None:
- log.debug("Alarm Already exists")
- # TODO: This should return a response with status False
- return config_resp
-
- else:
- log.info("New alarm created with alarm info: %s", config_resp)
- return config_resp
-
- else:
- log.error("Resource ID doesn't exists")
-
- elif key == "acknowledge_alarm":
- alarm_inner_dict = alarm_info['ack_details']
-
- if self.check_resource(alarm_inner_dict['resource_uuid']):
- ack_details = self.get_ack_details(alarm_info)
- log.info("Acknowledge sent: %s", ack_details)
- return ack_details
-
- else:
- log.error("Resource ID is Incorrect")
-
- elif key == "update_alarm_request":
- alarm_inner_dict = alarm_info['alarm_update_request']
- metric_status = self.check_metric(alarm_inner_dict['metric_name'])
-
- if metric_status['status']:
- log.debug("Resource and Metrics exists")
- alarm_info['alarm_update_request']['metric_name'] = metric_status['metric_name']
- # Generate a valid response message, send via producer
- update_resp = self.update_alarm_configuration(alarm_info)
-
- if update_resp is None:
- # TODO: This should return a response with status False
- log.debug("Alarm Already exists")
- return update_resp
-
- else:
- log.info("Alarm Updated with alarm info: %s", update_resp)
- return update_resp
-
- else:
- log.info("Metric Not Supported")
-
- elif key == "delete_alarm_request":
- # Generate a valid response message, send via producer
- del_resp = self.delete_alarm(alarm_info)
- log.info("Alarm Deleted with alarm info: %s", del_resp)
- return del_resp
-
- elif key == "alarm_list_request":
- alarm_inner_dict = alarm_info['alarm_list_request']
-
- if self.check_resource(alarm_inner_dict['resource_uuid']) or alarm_inner_dict['resource_uuid'] == "":
- # Generate a valid response message, send via producer
- list_resp = self.get_alarms_list(alarm_info) # ['alarm_names']
- return list_resp
- else:
- log.error("Resource ID is Incorrect")
-
- else:
- raise UnsupportedOperation("Unknown key, no action will be performed")
-
- except Exception as e:
- log.error("Message retrieval exception: %s", str(e))
-
- def check_resource(self, resource_uuid):
- """Finding Resource with the resource_uuid"""
- try:
- check_resp = dict()
- instances = self.ec2_conn.get_all_instance_status()
-
- # resource_id
- for instance_id in instances:
- instance_id = str(instance_id).split(':')[1]
-
- if instance_id == resource_uuid:
- check_resp['resource_uuid'] = resource_uuid
- return True
- return False
-
- except Exception as e:
- log.error("Error in Plugin Inputs %s", str(e))
-
- def check_metric(self, metric_name):
- """ Checking whether the metric is supported by AWS """
- try:
- check_resp = dict()
-
- # metric_name
- if metric_name == 'CPU_UTILIZATION':
- metric_name = 'CPUUtilization'
- metric_status = True
-
- elif metric_name == 'DISK_READ_OPS':
- metric_name = 'DiskReadOps'
- metric_status = True
-
- elif metric_name == 'DISK_WRITE_OPS':
- metric_name = 'DiskWriteOps'
- metric_status = True
-
- elif metric_name == 'DISK_READ_BYTES':
- metric_name = 'DiskReadBytes'
- metric_status = True
-
- elif metric_name == 'DISK_WRITE_BYTES':
- metric_name = 'DiskWriteBytes'
- metric_status = True
-
- elif metric_name == 'PACKETS_RECEIVED':
- metric_name = 'NetworkPacketsIn'
- metric_status = True
-
- elif metric_name == 'PACKETS_SENT':
- metric_name = 'NetworkPacketsOut'
- metric_status = True
-
- else:
- metric_name = None
- metric_status = False
- check_resp['metric_name'] = metric_name
- # status
-
- if metric_status:
- check_resp['status'] = True
- return check_resp
-
- except Exception as e:
- log.error("Error in Plugin Inputs %s", str(e))