a10773e413
Add a dictionary as a data cache for the pollsters, creating it outside of the loop where they are invoked and passing it to them. Future changesets will include changes to the pollsters to put data in the cache. blueprint one-meter-per-plugin Change-Id: Ie65526dfe65a8880ad8683b62fae62f8e7f9e69b Signed-off-by: Doug Hellmann <doug.hellmann@dreamhost.com>
107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Author: François Rossigneux <francois.rossigneux@inria.fr>
|
|
#
|
|
# 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.
|
|
|
|
import datetime
|
|
|
|
from keystoneclient import exceptions
|
|
import requests
|
|
|
|
from ceilometer.central import plugin
|
|
from ceilometer import counter
|
|
from ceilometer.openstack.common import log
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
|
|
class KwapiClient(object):
|
|
"""Kwapi API client."""
|
|
|
|
def __init__(self, url, token=None):
|
|
"""Initializes client."""
|
|
self.url = url
|
|
self.token = token
|
|
|
|
def iter_probes(self):
|
|
"""Returns a list of dicts describing all probes."""
|
|
probes_url = self.url + '/probes/'
|
|
headers = {}
|
|
if self.token is not None:
|
|
headers = {'X-Auth-Token': self.token}
|
|
request = requests.get(probes_url, headers=headers)
|
|
message = request.json
|
|
probes = message['probes']
|
|
for key, value in probes.iteritems():
|
|
probe_dict = value
|
|
probe_dict['id'] = key
|
|
yield probe_dict
|
|
|
|
|
|
class _Base(plugin.CentralPollster):
|
|
"""Base class for the Kwapi pollster, derived from CentralPollster."""
|
|
|
|
@staticmethod
|
|
def get_kwapi_client(ksclient):
|
|
"""Returns a KwapiClient configured with the proper url and token."""
|
|
endpoint = ksclient.service_catalog.url_for(service_type='energy',
|
|
endpoint_type='internalURL'
|
|
)
|
|
return KwapiClient(endpoint, ksclient.auth_token)
|
|
|
|
def iter_probes(self, ksclient):
|
|
"""Iterate over all probes."""
|
|
try:
|
|
client = self.get_kwapi_client(ksclient)
|
|
except exceptions.EndpointNotFound:
|
|
LOG.debug(_("Kwapi endpoint not found"))
|
|
return []
|
|
return client.iter_probes()
|
|
|
|
|
|
class KwapiPollster(_Base):
|
|
"""Kwapi pollster derived from the base class."""
|
|
|
|
@staticmethod
|
|
def get_counter_names():
|
|
return ['energy', 'power']
|
|
|
|
def get_counters(self, manager, cache):
|
|
"""Returns all counters."""
|
|
for probe in self.iter_probes(manager.keystone):
|
|
yield counter.Counter(
|
|
name='energy',
|
|
type=counter.TYPE_CUMULATIVE,
|
|
unit='kWh',
|
|
volume=probe['kwh'],
|
|
user_id=None,
|
|
project_id=None,
|
|
resource_id=probe['id'],
|
|
timestamp=datetime.datetime.fromtimestamp(
|
|
probe['timestamp']).isoformat(),
|
|
resource_metadata={}
|
|
)
|
|
yield counter.Counter(
|
|
name='power',
|
|
type=counter.TYPE_GAUGE,
|
|
unit='W',
|
|
volume=probe['w'],
|
|
user_id=None,
|
|
project_id=None,
|
|
resource_id=probe['id'],
|
|
timestamp=datetime.datetime.fromtimestamp(
|
|
probe['timestamp']).isoformat(),
|
|
resource_metadata={}
|
|
)
|