| #!/usr/bin/python |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you 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. |
| |
| import pprint |
| import XenAPI |
| import urllib |
| from xml.dom import minidom |
| import time |
| import commands |
| |
| # Per VM dictionary (used by RRDUpdates to look up column numbers by variable names) |
| class VMReport(dict): |
| """Used internally by RRDUpdates""" |
| def __init__(self, uuid): |
| self.uuid = uuid |
| super(dict, self).__init__() |
| |
| |
| # Per Host dictionary (used by RRDUpdates to look up column numbers by variable names) |
| class HostReport(dict): |
| """Used internally by RRDUpdates""" |
| def __init__(self, uuid): |
| self.uuid = uuid |
| super(dict, self).__init__() |
| |
| |
| class PerfMonException(Exception): |
| pass |
| |
| |
| class XmlConfigException(PerfMonException): |
| pass |
| |
| |
| class UsageException(Exception): |
| pass |
| |
| |
| class RRDUpdates: |
| """ Object used to get and parse the output the http://localhost/rrd_udpates?... |
| """ |
| def __init__(self): |
| # params are what get passed to the CGI executable in the URL |
| self.params = dict() |
| self.params['start'] = int(time.time()) - 1000 # For demo purposes! |
| self.params['host'] = 'false' # include data for host (as well as for VMs) |
| self.params['cf'] = 'AVERAGE' # consolidation function, each sample averages 12 from the 5 second RRD |
| self.params['interval'] = '60' |
| |
| def get_nrows(self): |
| return self.rows |
| |
| def get_vm_list(self): |
| return self.vm_reports.keys() |
| |
| def get_vm_param_list(self, uuid): |
| report = self.vm_reports[uuid] |
| if not report: |
| return [] |
| return report.keys() |
| |
| def get_total_cpu_core(self, uuid): |
| report = self.vm_reports[uuid] |
| if not report: |
| return 0 |
| else: |
| param_keys = report.keys() |
| result = 0 |
| for param in param_keys: |
| if "cpu" in param: |
| result += 1 |
| return result |
| |
| def get_vm_data(self, uuid, param, row): |
| #pp = pprint.PrettyPrinter(indent=4) |
| #pp.pprint(self.vm_reports) |
| report = self.vm_reports[uuid] |
| col = report[param] |
| return self.__lookup_data(col, row) |
| |
| def get_host_uuid(self): |
| report = self.host_report |
| if not report: |
| return None |
| return report.uuid |
| |
| def get_host_param_list(self): |
| report = self.host_report |
| if not report: |
| return [] |
| return report.keys() |
| |
| def get_host_data(self, param, row): |
| report = self.host_report |
| col = report[param] |
| return self.__lookup_data(col, row) |
| |
| def get_row_time(self, row): |
| return self.__lookup_timestamp(row) |
| |
| # extract float from value (<v>) node by col,row |
| def __lookup_data(self, col, row): |
| # Note: the <rows> nodes are in reverse chronological order, and comprise |
| # a timestamp <t> node, followed by self.columns data <v> nodes |
| node = self.data_node.childNodes[self.rows - 1 - row].childNodes[col + 1] |
| return float(node.firstChild.toxml()) # node.firstChild should have nodeType TEXT_NODE |
| |
| # extract int from value (<t>) node by row |
| def __lookup_timestamp(self, row): |
| # Note: the <rows> nodes are in reverse chronological order, and comprise |
| # a timestamp <t> node, followed by self.columns data <v> nodes |
| node = self.data_node.childNodes[self.rows - 1 - row].childNodes[0] |
| return int(node.firstChild.toxml()) # node.firstChild should have nodeType TEXT_NODE |
| |
| def refresh(self, login, starttime, session, override_params): |
| self.params['start'] = starttime |
| params = override_params |
| params['session_id'] = session |
| params.update(self.params) |
| paramstr = "&".join(["%s=%s" % (k, params[k]) for k in params]) |
| # this is better than urllib.urlopen() as it raises an Exception on http 401 'Unauthorised' error |
| # rather than drop into interactive mode |
| for host in login.host.get_all(): |
| #print "http://" + str(login.host.get_address(host)) + "/rrd_updates?%s" % paramstr |
| sock = urllib.URLopener().open("http://" + str(login.host.get_address(host)) + "/rrd_updates?%s" % paramstr) |
| xmlsource = sock.read() |
| sock.close() |
| xmldoc = minidom.parseString(xmlsource) |
| self.__parse_xmldoc(xmldoc) |
| # Update the time used on the next run |
| self.params['start'] = self.end_time + 1 # avoid retrieving same data twice |
| |
| def __parse_xmldoc(self, xmldoc): |
| # The 1st node contains meta data (description of the data) |
| # The 2nd node contains the data |
| self.meta_node = xmldoc.firstChild.childNodes[0] |
| self.data_node = xmldoc.firstChild.childNodes[1] |
| |
| def lookup_metadata_bytag(name): |
| return int(self.meta_node.getElementsByTagName(name)[0].firstChild.toxml()) |
| # rows = number of samples per variable |
| # columns = number of variables |
| self.rows = lookup_metadata_bytag('rows') |
| self.columns = lookup_metadata_bytag('columns') |
| # These indicate the period covered by the data |
| self.start_time = lookup_metadata_bytag('start') |
| self.step_time = lookup_metadata_bytag('step') |
| self.end_time = lookup_metadata_bytag('end') |
| # the <legend> Node describes the variables |
| self.legend = self.meta_node.getElementsByTagName('legend')[0] |
| # vm_reports matches uuid to per VM report |
| if not hasattr(self,'vm_reports'): |
| self.vm_reports = {} |
| # There is just one host_report and its uuid should not change! |
| self.host_report = None |
| # Handle each column. (I.e. each variable) |
| for col in range(self.columns): |
| self.__handle_col(col) |
| |
| def __handle_col(self, col): |
| # work out how to interpret col from the legend |
| col_meta_data = self.legend.childNodes[col].firstChild.toxml() |
| # vm_or_host will be 'vm' or 'host'. Note that the Control domain counts as a VM! |
| (cf, vm_or_host, uuid, param) = col_meta_data.split(':') |
| if vm_or_host == 'vm': |
| # Create a report for this VM if it doesn't exist |
| if not uuid in self.vm_reports: |
| self.vm_reports[uuid] = VMReport(uuid) |
| # Update the VMReport with the col data and meta data |
| vm_report = self.vm_reports[uuid] |
| vm_report[param] = col |
| elif vm_or_host == 'host': |
| # Create a report for the host if it doesn't exist |
| if not self.host_report: |
| self.host_report = HostReport(uuid) |
| elif self.host_report.uuid != uuid: |
| raise PerfMonException("Host UUID changed: (was %s, is %s)" % (self.host_report.uuid, uuid)) |
| # Update the HostReport with the col data and meta data |
| self.host_report[param] = col |
| else: |
| raise PerfMonException("Invalid string in <legend>: %s" % col_meta_data) |
| |
| def getuuid(vm_name): |
| status, output = commands.getstatusoutput("xe vm-list | grep "+vm_name+" -B 1 | head -n 1 | awk -F':' '{print $2}' | tr -d ' '") |
| if (status != 0): |
| raise PerfMonException("Invalid vm name: %s" % vm_name) |
| return output |
| |
| def get_vm_group_perfmon(args={}): |
| login = XenAPI.xapi_local() |
| login.login_with_password("","") |
| result = "" |
| |
| total_vm = int(args['total_vm']) |
| total_counter = int(args['total_counter']) |
| now = int(time.time()) / 60 |
| |
| # Get pool's info of this host |
| #pool = login.xenapi.pool.get_all()[0] |
| # Get master node's address of pool |
| #master = login.xenapi.pool.get_master(pool) |
| #master_address = login.xenapi.host.get_address(master) |
| session = login._session |
| |
| max_duration = 0 |
| for counter_count in xrange(1, total_counter + 1): |
| duration = int(args['duration' + str(counter_count)]) |
| if duration > max_duration: |
| max_duration = duration |
| |
| rrd_updates = RRDUpdates() |
| rrd_updates.refresh(login.xenapi, now * 60 - max_duration, session, {}) |
| |
| #for uuid in rrd_updates.get_vm_list(): |
| for vm_count in xrange(1, total_vm + 1): |
| vm_name = args['vmname' + str(vm_count)] |
| vm_uuid = getuuid(vm_name) |
| #print "Got values for VM: " + str(vm_count) + " " + vm_uuid |
| for counter_count in xrange(1, total_counter + 1): |
| #refresh average |
| average_cpu = 0 |
| average_memory = 0 |
| counter = args['counter' + str(counter_count)] |
| total_row = rrd_updates.get_nrows() |
| duration = int(args['duration' + str(counter_count)]) / 60 |
| duration_diff = total_row - duration |
| if counter == "cpu": |
| total_cpu = rrd_updates.get_total_cpu_core(vm_uuid) |
| for row in xrange(duration_diff, total_row): |
| for cpu in xrange(0, total_cpu): |
| average_cpu += rrd_updates.get_vm_data(vm_uuid, "cpu" + str(cpu), row) |
| average_cpu /= (duration * total_cpu) |
| if result == "": |
| result += str(vm_count) + '.' + str(counter_count) + ':' + str(average_cpu) |
| else: |
| result += ',' + str(vm_count) + '.' + str(counter_count) + ':' + str(average_cpu) |
| elif counter == "memory": |
| for row in xrange(duration_diff, total_row): |
| average_memory += rrd_updates.get_vm_data(vm_uuid, "memory_target", row) / 1048576 - rrd_updates.get_vm_data(vm_uuid, "memory_internal_free", row) / 1024 |
| average_memory /= duration |
| if result == "": |
| result += str(vm_count) + '.' + str(counter_count) + ':' + str(average_memory) |
| else: |
| result += ',' + str(vm_count) + '.' + str(counter_count) + ':' + str(average_memory) |
| return result |
| |
