109 lines
2.9 KiB
Python
109 lines
2.9 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from sensu_plugin import SensuPluginCheck
|
||
|
|
import json
|
||
|
|
import requests
|
||
|
|
|
||
|
|
def getGraphiteValue(host, port, target, from_time, to_time):
|
||
|
|
|
||
|
|
request_url = 'http://' + host + ':' + str(port) + '/render'
|
||
|
|
ranges = { 'from': from_time, 'to': to_time, 'target': target, 'format': 'json' }
|
||
|
|
|
||
|
|
|
||
|
|
r = requests.get(request_url, params=ranges)
|
||
|
|
graphite_dict = json.loads(r.content)
|
||
|
|
|
||
|
|
counter = 0
|
||
|
|
point_sum = 0
|
||
|
|
for point in graphite_dict:
|
||
|
|
for value, timestamp in point['datapoints']:
|
||
|
|
if value is not None:
|
||
|
|
point_sum = int(point_sum) + int(value)
|
||
|
|
counter += 1
|
||
|
|
|
||
|
|
avg_value = point_sum / counter
|
||
|
|
|
||
|
|
# return str(graphite_dict)
|
||
|
|
|
||
|
|
return avg_value
|
||
|
|
|
||
|
|
|
||
|
|
class GraphiteMetricCheck(SensuPluginCheck):
|
||
|
|
def setup(self):
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-w',
|
||
|
|
'--warning',
|
||
|
|
required=True,
|
||
|
|
type=int,
|
||
|
|
help='Integer warning level to output'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-c',
|
||
|
|
'--critical',
|
||
|
|
required=True,
|
||
|
|
type=int,
|
||
|
|
help='Integer critical level to output'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-s',
|
||
|
|
'--host',
|
||
|
|
required=True,
|
||
|
|
type=str,
|
||
|
|
help='graphite host. http://graphite.example.com'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-p',
|
||
|
|
'--port',
|
||
|
|
type=int,
|
||
|
|
default=8080,
|
||
|
|
help='Port'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-t',
|
||
|
|
'--target',
|
||
|
|
required=True,
|
||
|
|
type=str,
|
||
|
|
help='metric target: com.example.some.metric'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-f',
|
||
|
|
'--from_time',
|
||
|
|
default='-5min',
|
||
|
|
type=str,
|
||
|
|
help='from time. Default = -5min'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-o',
|
||
|
|
'--to_time',
|
||
|
|
type=str,
|
||
|
|
default='now',
|
||
|
|
help='to time. Default = now'
|
||
|
|
)
|
||
|
|
self.parser.add_argument(
|
||
|
|
'-m',
|
||
|
|
'--message',
|
||
|
|
type=str,
|
||
|
|
default=None,
|
||
|
|
help='message to display'
|
||
|
|
)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
# this method is called to_time perform the actual check
|
||
|
|
|
||
|
|
self.check_name('GraphiteMetricCheck') # defaults to_time class name
|
||
|
|
|
||
|
|
avg = getGraphiteValue(self.options.host, self.options.port, self.options.target, self.options.from_time, self.options.to_time)
|
||
|
|
|
||
|
|
if avg > self.options.warning and avg < self.options.critical:
|
||
|
|
self.warning(self.options.target + ' ' + str(avg))
|
||
|
|
elif avg < self.options.warning:
|
||
|
|
self.ok(self.options.target + ' ' + str(avg))
|
||
|
|
elif avg >= self.options.critical:
|
||
|
|
self.critical(self.options.target + ' ' + str(avg))
|
||
|
|
else:
|
||
|
|
self.unknown(self.options.message)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
f = GraphiteMetricCheck()
|