Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | # -*- coding: utf-8 -*- |
| 3 | |
| 4 | from ceilometerclient import client |
| 5 | from os import environ as env |
| 6 | import keystoneclient.v2_0.client as ksclient |
| 7 | import re |
| 8 | import datetime |
| 9 | import time |
| 10 | from monitor.monitordriver import * |
Sapan Bhatia | 6e08c3e | 2015-01-07 01:09:58 -0500 | [diff] [blame^] | 11 | from core.models import * |
Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 12 | |
Sapan Bhatia | 6e08c3e | 2015-01-07 01:09:58 -0500 | [diff] [blame^] | 13 | def object_to_filter(model_name, pk): |
| 14 | filter_dict = { |
| 15 | 'Slice':[Slice, 'tenant_id', 'project_id'], |
| 16 | 'Sliver':[Sliver, 'instance_id', 'resource_id'], |
| 17 | 'Site':[Site, 'tenant_id', 'project_id'] |
| 18 | } |
| 19 | |
| 20 | mod,field,tag = filter_dict[model_name] |
| 21 | obj = mod.objects.get(pk=pk) |
| 22 | return '%s=%s'%(tag,mod[field]) |
| 23 | |
Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 24 | |
| 25 | def cli_to_array(cli_query): |
| 26 | '''This converts from the cli list of queries to what is required |
| 27 | by the python api. |
| 28 | so from: |
| 29 | "this<=34;that=foo" |
| 30 | to |
| 31 | "[{field=this,op=le,value=34},{field=that,op=eq,value=foo}]" |
| 32 | ''' |
| 33 | if cli_query is None: |
| 34 | return None |
| 35 | |
| 36 | op_lookup = {'!=': 'ne', |
| 37 | '>=': 'ge', |
| 38 | '<=': 'le', |
| 39 | '>': 'gt', |
| 40 | '<': 'lt', |
| 41 | '=': 'eq'} |
| 42 | |
| 43 | def split_by_op(string): |
| 44 | # two character split (<=,!=) |
| 45 | frags = re.findall(r'([[a-zA-Z0-9_.]+)([><!]=)([^ -,\t\n\r\f\v]+)', |
| 46 | string) |
| 47 | if len(frags) == 0: |
| 48 | #single char split (<,=) |
| 49 | frags = re.findall(r'([a-zA-Z0-9_.]+)([><=])([^ -,\t\n\r\f\v]+)', |
| 50 | string) |
| 51 | return frags |
| 52 | |
| 53 | opts = [] |
| 54 | queries = cli_query.split(';') |
| 55 | for q in queries: |
| 56 | frag = split_by_op(q) |
| 57 | if len(frag) > 1: |
| 58 | raise ValueError('incorrect seperator %s in query "%s"' % |
| 59 | ('(should be ";")', q)) |
| 60 | if len(frag) == 0: |
| 61 | raise ValueError('invalid query %s' % q) |
| 62 | query = frag[0] |
| 63 | opt = {} |
| 64 | opt['field'] = query[0] |
| 65 | opt['op'] = op_lookup[query[1]] |
| 66 | opt['value'] = query[2] |
| 67 | opts.append(opt) |
| 68 | return opts |
| 69 | |
| 70 | def meters_to_stats(meters): |
| 71 | stats = DashboardStatistics() |
| 72 | for m in meters: |
| 73 | timestamp = datetime.datetime.strptime(m.duration_start,'%Y-%m-%dT%H:%M:%S') |
| 74 | stats.stat_list.append({'timestamp':timestamp, 'value':m.sum}) |
| 75 | stats.sum+=m.sum |
| 76 | stats.average+=m.sum |
| 77 | stats.unit = 'ns' |
| 78 | |
| 79 | stats.average/=len(meters) |
| 80 | return stats |
| 81 | |
| 82 | |
| 83 | |
| 84 | class CeilometerDriver(MonitorDriver): |
Sapan Bhatia | 6e08c3e | 2015-01-07 01:09:58 -0500 | [diff] [blame^] | 85 | def get_meter(self, meter, obj, pk, keystone=None): |
Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 86 | if (not keystone): |
| 87 | keystone = {} |
| 88 | keystone['username']=env['OS_USERNAME'] |
| 89 | keystone['password']=env['OS_PASSWORD'] |
| 90 | keystone['auth_url']=env['OS_AUTH_URL'] |
| 91 | keystone['tenant_name']=env['OS_TENANT_NAME'] |
Sapan Bhatia | 6e08c3e | 2015-01-07 01:09:58 -0500 | [diff] [blame^] | 92 | keystone['os_cacert']=env['OS_CACERT'] |
Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 93 | |
| 94 | ceilometer_client = client._get_ksclient(**keystone) |
| 95 | token = ceilometer_client.auth_token |
| 96 | |
| 97 | ceilo_endpoint = client._get_endpoint(ceilometer_client, **keystone) |
Sapan Bhatia | 6e08c3e | 2015-01-07 01:09:58 -0500 | [diff] [blame^] | 98 | #ceilometer = client.get_client(2, username=keystone['username'], password=keystone['password'], tenant_name=keystone['tenant_name'], auth_url=keystone['auth_url']) |
Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 99 | |
| 100 | ceilometer = client.Client('2',endpoint = ceilo_endpoint, token = lambda: token) |
| 101 | |
| 102 | cur_ts = datetime.datetime.fromtimestamp(time.time()-86400) |
| 103 | str_ts = cur_ts.strftime('%Y-%m-%dT%H:%M:%S') |
| 104 | |
Sapan Bhatia | 6e08c3e | 2015-01-07 01:09:58 -0500 | [diff] [blame^] | 105 | object_filter = object_to_filter(obj, pk) |
Sapan Bhatia | 56410d5 | 2014-12-16 01:08:23 -0500 | [diff] [blame] | 106 | filter=';'.join([object_filter,'timestamp>%s'%str_ts]) |
| 107 | #query = cli_to_array("project_id=124de34266b24f57957345cdb43cc9ff;timestamp>2014-12-11T00:00:00") |
| 108 | query = cli_to_array(filter) |
| 109 | |
| 110 | meters = ceilometer.statistics.list(meter,q=query,period=3600) |
| 111 | |
| 112 | stats = meters_to_stats(meters) |
| 113 | return stats |
| 114 | |