From 1e2387474a449452b78520b9ad96a8b4b5e99722 Mon Sep 17 00:00:00 2001 From: Harald Pfeiffer Date: Wed, 17 Apr 2019 19:07:19 +0200 Subject: initial commit of source fetch --- .../check_graphite/check_graphite.py | 286 +++++++++++++++++++++ 1 file changed, 286 insertions(+) create mode 100755 nagios-plugins-contrib-24.20190301~bpo9+1/check_graphite/check_graphite.py (limited to 'nagios-plugins-contrib-24.20190301~bpo9+1/check_graphite/check_graphite.py') diff --git a/nagios-plugins-contrib-24.20190301~bpo9+1/check_graphite/check_graphite.py b/nagios-plugins-contrib-24.20190301~bpo9+1/check_graphite/check_graphite.py new file mode 100755 index 0000000..bedb7c1 --- /dev/null +++ b/nagios-plugins-contrib-24.20190301~bpo9+1/check_graphite/check_graphite.py @@ -0,0 +1,286 @@ +#!/usr/bin/env python +""" +check_graphite.py +~~~~~~~ + +:copyright: (c) 2012 DISQUS. +:license: Apache License 2.0, see LICENSE for more details. +""" + +import json +import optparse +import urllib +import urllib2 +import sys + +from numbers import Real + +NAGIOS_STATUSES = { + 'OK': 0, + 'WARNING': 1, + 'CRITICAL': 2, + 'UNKNOWN': 3 +} + +class Graphite(object): + + def __init__(self, url, targets, _from, _until): + self.url = url.rstrip('/') + self.targets = targets + self._from = _from + self._until = _until + params = [('target', t) for t in self.targets] +\ + [('from', self._from)] +\ + [('until', self._until)] +\ + [('format', 'json')] + self.full_url = self.url + '/render?' +\ + urllib.urlencode(params) + + def check_datapoints(self, datapoints, check_func, **kwargs): + """Find alerting datapoints + + Args: + datapoints (list): The list of datapoints to check + + Kwargs: + check_func (function): The function to find out of bounds datapoints + bounds (list): Compare against `datapoints` to find out of bounds list + compare (list): Used for comparison if `datapoints` is out of bounds + threshold (float): `check_func` is called for each datapoint against `threshold` + beyond (float): Return datapoint if `beyond` value in bounds list (percentage). + + Returns: + The list of out of bounds datapoints + """ + if 'threshold' in kwargs: + return [x for x in datapoints if isinstance(x, Real) and check_func(x, kwargs['threshold'])] + elif 'bounds' in kwargs: + if 'compare' in kwargs: + return [datapoints[x] for x in xrange(len(datapoints)) if all([datapoints[x], kwargs['bounds'][x], kwargs['compare'][x]]) and check_func(datapoints[x] / kwargs['bounds'][x], kwargs['beyond']) and check_func(datapoints[x], kwargs['compare'][x])] + else: + return [datapoints[x] for x in xrange(len(datapoints)) if all([datapoints[x], kwargs['bounds'][x]]) and check_func(datapoints[x], kwargs['bounds'][x])] + + def fetch_metrics(self): + try: + response = urllib2.urlopen(self.full_url) + + if response.code != 200: + return None + else: + return json.loads(response.read()) + except urllib2.URLError, TypeError: + return None + + def generate_output(self, datapoints, *args, **kwargs): + """Generate check output + + Args: + datapoints (list): The list of datapoints to check + warn_oob (list): Optional list of datapoints considered in warning state + crit_oob (list): Mandatory list of datapoints considered in warning state + + Kwargs: + count (int): Number of metrics that would generate an alert + warning (float): The check's warning threshold + critical (float): The check's critical threshold + target (str): The target for `datapoints` + + Returns: + A dictionary of datapoints grouped by their status ('CRITICAL', 'WARNING', 'OK') + """ + check_output = dict(OK=[], WARNING=[], CRITICAL=[]) + count = kwargs['count'] + warning = kwargs.get('warning', 0) + critical = kwargs.get('critical', 0) + target = kwargs.get('target', 'timeseries') + + if len(args) > 1: + (warn_oob, crit_oob) = args + else: + crit_oob = [x for x in args[0] if isinstance(x, Real)] + warn_oob = [] + + if self.has_numbers(crit_oob) and len(crit_oob) >= count: + check_output['CRITICAL'].append('%s [crit=%f|datapoints=%s]' %\ + (target, critical, ','.join(['%s' % str(x) for x in crit_oob]))) + elif self.has_numbers(warn_oob) and len(warn_oob) >= count: + check_output['WARNING'].append('%s [warn=%f|datapoints=%s]' %\ + (target, warning, ','.join(['%s' % str(x) for x in warn_oob]))) + else: + check_output['OK'].append('%s [warn=%0.3f|crit=%f|datapoints=%s]' %\ + (target, warning, critical, ','.join(['%s' % str(x) for x in datapoints]))) + + return check_output + + def has_numbers(self, lst): + try: + return any([isinstance(x, Real) for x in lst]) + except TypeError: + return False + + +if __name__ == '__main__': + parser = optparse.OptionParser() + parser.add_option('-U', '--graphite-url', dest='graphite_url', + default='http://localhost/', + metavar='URL', + help='Graphite URL [%default]') + parser.add_option('-t', '--target', dest='target', + action='append', + help='Target to check') + parser.add_option('--compare', dest='compare', + metavar='SERIES', + help='Compare TARGET against SERIES') + parser.add_option('--from', dest='_from', + help='From timestamp/date') + parser.add_option('--until', dest='_until', + default='now', + help='Until timestamp/date [%default]') + parser.add_option('-c', '--count', dest='count', + default=0, + type='int', + help='Alert on at least COUNT metrics [%default]') + parser.add_option('--beyond', dest='beyond', + default=0.7, + type='float', + help='Alert if metric is PERCENTAGE beyond comparison value [%default]') + parser.add_option('--percentile', dest='percentile', + default=0, + type='int', + metavar='PERCENT', + help='Use nPercentile Graphite function on the target (returns one datapoint)') + parser.add_option('--empty-ok', dest='empty_ok', + default=False, + action='store_true', + help='Empty data from Graphite is OK') + parser.add_option('--confidence', dest='confidence_bands', + default=False, + action='store_true', + help='Use holtWintersConfidenceBands Graphite function on the target') + parser.add_option('--over', dest='over', + default=True, + action='store_true', + help='Over specified WARNING or CRITICAL threshold [%default]') + parser.add_option('--under', dest='under', + default=False, + action='store_true', + help='Under specified WARNING or CRITICAL threshold [%default]') + parser.add_option('-W', dest='warning', + type='float', + metavar='VALUE', + help='Warning if datapoints beyond VALUE') + parser.add_option('-C', dest='critical', + type='float', + metavar='VALUE', + help='Critical if datapoints beyond VALUE') + + (options, args) = parser.parse_args() + + if not all([getattr(options, option) for option in ('_from', 'target')]): + parser.print_help() + sys.exit(NAGIOS_STATUSES['UNKNOWN']) + + real_from = options._from + + if options.under: + check_func = lambda x, y: x < y + options.over = False + else: + check_func = lambda x, y: x > y + + if options.confidence_bands: + targets = [options.target[0], 'holtWintersConfidenceBands(%s)' % options.target[0]] + check_threshold = None + from_slice = int(options._from) * -1 + real_from = '-2w' + + if options.compare: + targets.append(options.compare) + else: + if not all([getattr(options, option) for option in ('critical', 'warning')]): + parser.print_help() + sys.exit(NAGIOS_STATUSES['UNKNOWN']) + + if options.percentile: + targets = ['nPercentile(%s, %d)' % (options.target[0], options.percentile)] + else: + targets = options.target + + try: + warn = float(options.warning) + crit = float(options.critical) + except ValueError: + print 'ERROR: WARNING or CRITICAL threshold is not a number\n' + parser.print_help() + sys.exit(NAGIOS_STATUSES['UNKNOWN']) + + check_output = {} + graphite = Graphite(options.graphite_url, targets, real_from, options._until) + metric_data = graphite.fetch_metrics() + + if metric_data: + if options.confidence_bands: + actual = [x[0] for x in metric_data[0].get('datapoints', [])][from_slice:] + target_name = metric_data[0]['target'] + kwargs = {} + kwargs['beyond'] = options.beyond + + if options.over: + kwargs['bounds'] = [x[0] for x in metric_data[1].get('datapoints', [])][from_slice:] + elif options.under: + kwargs['bounds'] = [x[0] for x in metric_data[2].get('datapoints', [])][from_slice:] + + if options.compare: + kwargs['compare'] = [x[0] for x in metric_data[3].get('datapoints', [])][from_slice:] + + if not graphite.has_numbers(kwargs['compare']): + print 'CRITICAL: No compare target output from Graphite!' + sys.exit(NAGIOS_STATUSES['CRITICAL']) + + if graphite.has_numbers(actual) and graphite.has_numbers(kwargs['bounds']): + points_oob = graphite.check_datapoints(actual, check_func, **kwargs) + check_output[target_name] = graphite.generate_output(actual, + points_oob, + count=options.count, + target=target_name) + + else: + print 'CRITICAL: No output from Graphite for target(s): %s' % ', '.join(targets) + sys.exit(NAGIOS_STATUSES['CRITICAL']) + else: + for target in metric_data: + datapoints = [x[0] for x in target.get('datapoints', []) if isinstance(x[0], Real)] + if not graphite.has_numbers(datapoints) and not options.empty_ok: + print 'CRITICAL: No output from Graphite for target(s): %s' % ', '.join(targets) + sys.exit(NAGIOS_STATUSES['CRITICAL']) + + crit_oob = graphite.check_datapoints(datapoints, check_func, threshold=crit) + warn_oob = graphite.check_datapoints(datapoints, check_func, threshold=warn) + check_output[target['target']] = graphite.generate_output(datapoints, + warn_oob, + crit_oob, + count=options.count, + target=target['target'], + warning=warn, + critical=crit) + else: + if options.empty_ok and isinstance(metric_data, list): + print 'OK: No output from Graphite for target(s): %s' % ', '.join(targets) + sys.exit(NAGIOS_STATUSES['OK']) + + print 'CRITICAL: No output from Graphite for target(s): %s' % ', '.join(targets) + sys.exit(NAGIOS_STATUSES['CRITICAL']) + + for target, messages in check_output.iteritems(): + if messages['CRITICAL']: + exit_code = NAGIOS_STATUSES['CRITICAL'] + elif messages['WARNING']: + exit_code = NAGIOS_STATUSES['WARNING'] + else: + exit_code = NAGIOS_STATUSES['OK'] + + for status_code in ['CRITICAL', 'WARNING', 'OK']: + if messages[status_code]: + print '\n'.join(['%s: %s' % (status_code, status) for status in messages[status_code]]) + + sys.exit(exit_code) -- cgit v1.2.3