Emma Foley cfcdbfdfd1 Add collectd_gnocchi support
- Create a new gnocchi dir for the collectd-gnocchi plugin.
- Get gnocchi endpoint from keystone
- Add new _base_url format
- Create metrics if they don't already exist
- Add instructions to doc/source/devstackGSG.rst

Change-Id: Id7ce8130cb22f33147b7f031cd65564375db10d6
2017-01-17 09:55:32 +00:00

243 lines
8.3 KiB
Python

# -*- coding: utf-8 -*-
# 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.
"""Gnocchi collectd plugin implementation"""
from __future__ import division
from __future__ import unicode_literals
import collectd_ceilometer
from collectd_ceilometer.common.keystone_light import ClientV3
from collectd_ceilometer.common.keystone_light import KeystoneException
from collectd_ceilometer.common.settings import Config
import json
import logging
import requests
from requests.exceptions import RequestException
import six
import threading
LOGGER = logging.getLogger(__name__)
ROOT_LOGGER = logging.getLogger(collectd_ceilometer.__name__)
# HTTP status codes
HTTP_CREATED = 201
HTTP_UNAUTHORIZED = 401
HTTP_NOT_FOUND = 404
class Sender(object):
"""Sends the JSON serialized data to Gnocchi"""
def __init__(self):
"""Create the Sender instance
The cofinguration must be initialized before the object is created.
"""
self._url_base = None
self._keystone = None
self._auth_token = None
self._auth_lock = threading.Lock()
self._failed_auth = False
self._meter_ids = {}
def _authenticate(self):
"""Authenticate and renew the authentication token"""
# if auth_token is available, just return it
if self._auth_token is not None:
return self._auth_token
# aquire the authentication lock
with self._auth_lock:
# re-check the auth_token as another thread could set it
if self._auth_token is not None:
return self._auth_token
LOGGER.debug('Authenticating request')
# pylint: disable=broad-except
try:
# create a keystone client if it doesn't exist
if self._keystone is None:
cfg = Config.instance()
self._keystone = ClientV3(
auth_url=cfg.OS_AUTH_URL,
username=cfg.OS_USERNAME,
password=cfg.OS_PASSWORD,
tenant_name=cfg.OS_TENANT_NAME
)
# store the authentication token
self._auth_token = self._keystone.auth_token
# get the uri of service endpoint
endpoint = self._get_endpoint("gnocchi")
self._url_base = "{}/v1/metric/%s/measures".format(endpoint)
LOGGER.info('Authenticating request - success')
self._failed_auth = False
except KeystoneException as exc:
log_level = logging.DEBUG
if not self._failed_auth:
log_level = logging.ERROR
LOGGER.error(
'Suspending error logs until successful auth'
)
LOGGER.log(log_level, 'Authentication error: %s',
six.text_type(exc),
exc_info=0)
if exc.response:
LOGGER.debug('Response: %s', exc.response)
self._auth_token = None
self._failed_auth = True
return self._auth_token
def send(self, metername, payload, unit):
"""Send the payload to Gnocchi"""
# get the auth_token
auth_token = self._authenticate()
LOGGER.info('Auth_token: %s',
auth_token,
)
# if auth_token is not set, there is nothing to do
if auth_token is None:
LOGGER.debug('Unable to send data. Not authenticated')
return
if self._url_base is None:
LOGGER.debug(
'Unable to send data. Missing endpoint from ident server')
return
# create request URL
metric_id = self._get_metric_id(metername, unit)
url = self._url_base % (metric_id)
# send the POST request
result = self._perform_request(url, payload, auth_token)
if result is None:
return
LOGGER.info('Result: %s %s',
six.text_type(result.status_code),
result.text)
# if the request failed due to an auth error
if result.status_code == HTTP_UNAUTHORIZED:
# reset the auth token in order to force the subsequent
# _authenticate() call to renew it
# Here, it can happen that the token is reset right after
# another thread has finished the authentication and thus
# the authentication may be performed twice
self._auth_token = None
# renew the authentication token
auth_token = self._authenticate()
if auth_token is not None:
# and try to repost
result = self._perform_request(url, payload, auth_token)
if result.status_code == HTTP_NOT_FOUND:
LOGGER.debug("Received 404 error when submitting %s sample, \
creating a new metric",
metername)
# create metric (endpoint, metername)
metric_id = self._get_metric_id(metername, unit)
LOGGER.info('metername: %s, meter_id: %s', metername, metric_id)
# Set a new url for the request
url = self._url_base % (metric_id)
# TODO(emma-l-foley): Add error checking
# Submit the sample
result = self._perform_request(url, payload, auth_token)
if result.status_code == HTTP_CREATED:
LOGGER.debug('Result: %s', HTTP_CREATED)
else:
LOGGER.info('Result: %s %s',
result.status_code,
result.text)
def _get_endpoint(self, service):
# get the uri of service endpoint
endpoint = self._keystone.get_service_endpoint(
service,
Config.instance().CEILOMETER_URL_TYPE)
return endpoint
def _get_metric_id(self, metername, unit):
try:
return self._meter_ids[metername]
except KeyError as ke:
LOGGER.warn(ke)
LOGGER.warn('No known ID for %s', metername)
endpoint = self._get_endpoint("gnocchi")
self._meter_ids[metername] = \
self._create_metric(metername, endpoint, unit)
return self._meter_ids[metername]
def _create_metric(self, metername, endpoint, unit):
url = "{}/v1/metric/".format(endpoint)
payload = json.dumps({"archive_policy_name": "high",
"name": metername,
"unit": unit,
})
result = self._perform_request(url, payload, self._auth_token)
metric_id = json.loads(result.text)['id']
LOGGER.debug("metric_id=%s", metric_id)
return metric_id
@classmethod
def _perform_request(cls, url, payload, auth_token):
"""Perform the POST request"""
LOGGER.debug('Performing request to %s', url)
# request headers
headers = {'X-Auth-Token': auth_token,
'Content-type': 'application/json'}
# perform request and return its result
response = None
try:
LOGGER.debug(
"Performing request to: %s with data=%s and headers=%s",
url, payload, headers)
response = requests.post(
url, data=payload, headers=headers,
timeout=(Config.instance().CEILOMETER_TIMEOUT / 1000.))
LOGGER.info('Response: %s: %s',
response.status_code, response.text
)
except RequestException as exc:
LOGGER.error('gnocchi request error: %s', six.text_type(exc))
finally:
LOGGER.debug('Returning response from _perform_request(): %s',
response.status_code)
return response