72d6577796
When services are registered to keystone without using default service types, pollsters for the services don't work. This fix enables the pollsters to get service types from ceilometer.conf. If there is no entry in the file, the default service types will be used. Change-Id: Iab30b6e749f5d2af4ecb0bbf6fe1a137793f4c2c Closes-Bug: 1353356 DocImpact: Add new parameters in ceilometer.conf
132 lines
4.3 KiB
Python
132 lines
4.3 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
|
|
from oslo.config import cfg
|
|
import requests
|
|
import six
|
|
|
|
from ceilometer.central import plugin
|
|
from ceilometer.openstack.common.gettextutils import _
|
|
from ceilometer.openstack.common import log
|
|
from ceilometer import sample
|
|
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
service_types_opts = [
|
|
cfg.StrOpt('kwapi',
|
|
default='energy',
|
|
help='Kwapi service type.'),
|
|
]
|
|
|
|
cfg.CONF.register_opts(service_types_opts, group='service_types')
|
|
|
|
|
|
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 six.iteritems(probes):
|
|
probe_dict = value
|
|
probe_dict['id'] = key
|
|
yield probe_dict
|
|
|
|
|
|
class _Base(plugin.CentralPollster):
|
|
"""Base class for the Kwapi pollster, derived from CentralPollster."""
|
|
|
|
@property
|
|
def default_discovery(self):
|
|
return 'endpoint:%s' % cfg.CONF.service_types.kwapi
|
|
|
|
@staticmethod
|
|
def get_kwapi_client(ksclient, endpoint):
|
|
"""Returns a KwapiClient configured with the proper url and token."""
|
|
return KwapiClient(endpoint, ksclient.auth_token)
|
|
|
|
CACHE_KEY_PROBE = 'kwapi.probes'
|
|
|
|
def _iter_probes(self, ksclient, cache, endpoint):
|
|
"""Iterate over all probes."""
|
|
key = '%s-%s' % (endpoint, self.CACHE_KEY_PROBE)
|
|
if key not in cache:
|
|
cache[key] = self._get_probes(ksclient, endpoint)
|
|
return iter(cache[key])
|
|
|
|
def _get_probes(self, ksclient, endpoint):
|
|
try:
|
|
client = self.get_kwapi_client(ksclient, endpoint)
|
|
except exceptions.EndpointNotFound:
|
|
LOG.debug(_("Kwapi endpoint not found"))
|
|
return []
|
|
return list(client.iter_probes())
|
|
|
|
|
|
class EnergyPollster(_Base):
|
|
"""Measures energy consumption."""
|
|
def get_samples(self, manager, cache, resources):
|
|
"""Returns all samples."""
|
|
for endpoint in resources:
|
|
for probe in self._iter_probes(manager.keystone, cache, endpoint):
|
|
yield sample.Sample(
|
|
name='energy',
|
|
type=sample.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={}
|
|
)
|
|
|
|
|
|
class PowerPollster(_Base):
|
|
"""Measures power consumption."""
|
|
def get_samples(self, manager, cache, resources):
|
|
"""Returns all samples."""
|
|
for endpoint in resources:
|
|
for probe in self._iter_probes(manager.keystone, cache, endpoint):
|
|
yield sample.Sample(
|
|
name='power',
|
|
type=sample.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={}
|
|
)
|