fuel-plugin-lma-collector/deployment_scripts/puppet/modules/lma_collector/files/collectd/openstack_cinder.py

131 lines
4.6 KiB
Python

#!/usr/bin/python
# Copyright 2015 Mirantis, Inc.
#
# 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.
#
# Collectd plugin for getting statistics from Cinder
import collectd
from collections import Counter
from collections import defaultdict
import re
import collectd_openstack as openstack
PLUGIN_NAME = 'cinder'
INTERVAL = openstack.INTERVAL
class CinderStatsPlugin(openstack.CollectdPlugin):
""" Class to report the statistics on Cinder service.
state of agents
number of volumes broken down by state
total size of volumes usable and in error state
"""
states = {'up': 0, 'down': 1, 'disabled': 2}
cinder_re = re.compile('^cinder-')
def collect(self):
# Get information of the state per service
# State can be: 'up', 'down' or 'disabled'
aggregated_workers = defaultdict(Counter)
for worker in self.iter_workers('cinder'):
host = worker['host'].split('.')[0]
service = self.cinder_re.sub('', worker['service'])
state = worker['state']
aggregated_workers[service][state] += 1
self.dispatch_value('cinder_service', '',
self.states[state],
{'host': host,
'service': service,
'state': state})
for service in aggregated_workers:
totalw = sum(aggregated_workers[service].values())
for state in self.states:
prct = (100.0 * aggregated_workers[service][state]) / totalw
self.dispatch_value('cinder_services_percent', '',
prct,
{'state': state, 'service': service})
self.dispatch_value('cinder_services', '',
aggregated_workers[service][state],
{'state': state, 'service': service})
volumes_details = self.get_objects_details('cinder', 'volumes')
def groupby(d):
return d.get('status', 'unknown').lower()
def count_size_bytes(d):
return d.get('size', 0) * 10 ** 9
status = self.count_objects_group_by(volumes_details,
group_by_func=groupby)
for s, nb in status.iteritems():
self.dispatch_value('volumes', s, nb)
sizes = self.count_objects_group_by(volumes_details,
group_by_func=groupby,
count_func=count_size_bytes)
for n, size in sizes.iteritems():
self.dispatch_value('volumes_size', n, size)
snaps_details = self.get_objects_details('cinder', 'snapshots')
status_snaps = self.count_objects_group_by(snaps_details,
group_by_func=groupby)
for s, nb in status_snaps.iteritems():
self.dispatch_value('snapshots', s, nb)
sizes = self.count_objects_group_by(snaps_details,
group_by_func=groupby,
count_func=count_size_bytes)
for n, size in sizes.iteritems():
self.dispatch_value('snapshots_size', n, size)
def dispatch_value(self, plugin_instance, name, value, meta=None):
v = collectd.Values(
plugin=PLUGIN_NAME, # metric source
plugin_instance=plugin_instance,
type='gauge',
type_instance=name,
interval=INTERVAL,
# w/a for https://github.com/collectd/collectd/issues/716
meta=meta or {'0': True},
values=[value]
)
v.dispatch()
plugin = CinderStatsPlugin(collectd, PLUGIN_NAME)
def config_callback(conf):
plugin.config_callback(conf)
def notification_callback(notification):
plugin.notification_callback(notification)
def read_callback():
plugin.conditional_read_callback()
collectd.register_config(config_callback)
collectd.register_notification(notification_callback)
collectd.register_read(read_callback, INTERVAL)