2 # Copyright 2017 xFlow Research Pvt. Ltd
3 # This file is part of MON module
6 # Licensed under the Apache License, Version 2.0 (the "License"); you may
7 # not use this file except in compliance with the License. You may obtain
8 # a copy of the License at
10 # http://www.apache.org/licenses/LICENSE-2.0
12 # Unless required by applicable law or agreed to in writing, software
13 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
14 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
15 # License for the specific language governing permissions and limitations
18 # For those usages not covered by the Apache License, Version 2.0 please
19 # contact with: wajeeha.hamid@xflowresearch.com
23 AWS-Plugin implements all the methods of MON to interact with AWS using the BOTO client
26 __author__
= "Wajeeha Hamid"
27 __date__
= "18-September-2017"
31 from connection
import Connection
32 from metric_alarms
import MetricAlarm
33 from metrics
import Metrics
34 sys
.path
.append("../../core/message_bus")
35 from producer
import KafkaProducer
38 log
= logging
.getLogger(__name__
)
40 class plugin_metrics():
41 """Receives Alarm info from MetricAlarm and connects with the consumer/producer """
43 self
.metric
= Metrics()
44 self
.producer
= KafkaProducer('')
45 #---------------------------------------------------------------------------------------------------------------------------
46 def create_metric_request(self
,metric_info
):
47 '''Comaptible API using normalized parameters'''
48 metric_resp
= self
.metric
.createMetrics(self
.cloudwatch_conn
,metric_info
)
50 #---------------------------------------------------------------------------------------------------------------------------
51 def update_metric_request(self
,updated_info
):
52 '''Comaptible API using normalized parameters'''
53 update_resp
= self
.metric
.updateMetrics(self
.cloudwatch_conn
,updated_info
)
55 #---------------------------------------------------------------------------------------------------------------------------
56 def delete_metric_request(self
,delete_info
):
57 '''Comaptible API using normalized parameters'''
58 del_resp
= self
.metric
.deleteMetrics(self
.cloudwatch_conn
,delete_info
)
60 #---------------------------------------------------------------------------------------------------------------------------
61 def list_metrics_request(self
,list_info
):
62 '''Comaptible API using normalized parameters'''
63 list_resp
= self
.metric
.listMetrics(self
.cloudwatch_conn
,list_info
)
65 #---------------------------------------------------------------------------------------------------------------------------
66 def read_metrics_data(self
,list_info
):
67 '''Comaptible API using normalized parameters
68 Read all metric data related to a specified metric'''
69 data_resp
=self
.metric
.metricsData(self
.cloudwatch_conn
,list_info
)
71 #---------------------------------------------------------------------------------------------------------------------------
73 def metric_calls(self
,message
,aws_conn
):
74 """Gets the message from the common consumer"""
77 self
.cloudwatch_conn
= aws_conn
['cloudwatch_connection']
78 self
.ec2_conn
= aws_conn
['ec2_connection']
80 metric_info
= json
.loads(message
.value
)
81 metric_response
= dict()
83 if metric_info
['vim_type'] == 'AWS':
84 log
.debug ("VIM support : AWS")
86 # Check the Functionlity that needs to be performed: topic = 'alarms'/'metrics'/'Access_Credentials'
87 if message
.topic
== "metric_request":
88 log
.info("Action required against: %s" % (message
.topic
))
90 if message
.key
== "create_metric_request":
91 if self
.check_resource(metric_info
['metric_create']['resource_uuid']) == True:
92 metric_resp
= self
.create_metric_request(metric_info
['metric_create']) #alarm_info = message.value
93 metric_response
['schema_version'] = metric_info
['schema_version']
94 metric_response
['schema_type'] = "create_metric_response"
95 metric_response
['metric_create_response'] = metric_resp
96 payload
= json
.dumps(metric_response
)
97 file = open('../../core/models/create_metric_resp.json','wb').write((payload
))
98 self
.producer
.create_metrics_resp(key
='create_metric_response',message
=payload
,topic
= 'metric_response')
100 log
.info("Metric configured: %s", metric_resp
)
101 return metric_response
103 elif message
.key
== "update_metric_request":
104 if self
.check_resource(metric_info
['metric_create']['resource_uuid']) == True:
105 update_resp
= self
.update_metric_request(metric_info
['metric_create'])
106 metric_response
['schema_version'] = metric_info
['schema_version']
107 metric_response
['schema_type'] = "update_metric_response"
108 metric_response
['metric_update_response'] = update_resp
109 payload
= json
.dumps(metric_response
)
110 file = open('../../core/models/update_metric_resp.json','wb').write((payload
))
111 self
.producer
.update_metric_response(key
='update_metric_response',message
=payload
,topic
= 'metric_response')
113 log
.info("Metric Updates: %s",metric_response
)
114 return metric_response
116 elif message
.key
== "delete_metric_request":
117 if self
.check_resource(metric_info
['resource_uuid']) == True:
118 del_resp
=self
.delete_metric_request(metric_info
)
119 payload
= json
.dumps(del_resp
)
120 file = open('../../core/models/delete_metric_resp.json','wb').write((payload
))
121 self
.producer
.delete_metric_response(key
='delete_metric_response',message
=payload
,topic
= 'metric_response')
123 log
.info("Metric Deletion Not supported in AWS : %s",del_resp
)
126 elif message
.key
== "list_metric_request":
127 if self
.check_resource(metric_info
['metrics_list_request']['resource_uuid']) == True:
128 list_resp
= self
.list_metrics_request(metric_info
['metrics_list_request'])
129 metric_response
['schema_version'] = metric_info
['schema_version']
130 metric_response
['schema_type'] = "list_metric_response"
131 metric_response
['correlation_id'] = metric_info
['metrics_list_request']['correlation_id']
132 metric_response
['vim_type'] = metric_info
['vim_type']
133 metric_response
['metrics_list'] = list_resp
134 payload
= json
.dumps(metric_response
)
135 file = open('../../core/models/list_metric_resp.json','wb').write((payload
))
136 self
.producer
.list_metric_response(key
='list_metrics_response',message
=payload
,topic
= 'metric_response')
138 log
.info("Metric List: %s",metric_response
)
139 return metric_response
141 elif message
.key
== "read_metric_data_request":
142 if self
.check_resource(metric_info
['resource_uuid']) == True:
143 data_resp
= self
.read_metrics_data(metric_info
)
144 metric_response
['schema_version'] = metric_info
['schema_version']
145 metric_response
['schema_type'] = "read_metric_data_response"
146 metric_response
['metric_name'] = metric_info
['metric_name']
147 metric_response
['metric_uuid'] = metric_info
['metric_uuid']
148 metric_response
['correlation_id'] = metric_info
['correlation_uuid']
149 metric_response
['resource_uuid'] = metric_info
['resource_uuid']
150 metric_response
['tenant_uuid'] = metric_info
['tenant_uuid']
151 metric_response
['metrics_data'] = data_resp
152 payload
= json
.dumps(metric_response
)
153 file = open('../../core/models/read_metric_data_resp.json','wb').write((payload
))
154 self
.producer
.read_metric_data_response(key
='read_metric_data_response',message
=payload
,topic
= 'metric_response')
156 log
.info("Metric Data Response: %s",metric_response
)
157 return metric_response
160 log
.debug("Unknown key, no action will be performed")
162 log
.info("Message topic not relevant to this plugin: %s",
165 except Exception as e
:
166 log
.error("Consumer exception: %s", str(e
))
168 #---------------------------------------------------------------------------------------------------------------------------
169 def check_resource(self
,resource_uuid
):
171 '''Checking the resource_uuid is present in EC2 instances'''
174 instances
= self
.ec2_conn
.get_all_instance_status()
175 status_resource
= False
178 for instance_id
in instances
:
179 instance_id
= str(instance_id
).split(':')[1]
180 if instance_id
== resource_uuid
:
181 check_resp
['resource_uuid'] = resource_uuid
182 status_resource
= True
184 status_resource
= False
187 return status_resource
189 except Exception as e
:
190 log
.error("Error in Plugin Inputs %s",str(e
))
191 #---------------------------------------------------------------------------------------------------------------------------