48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
|
import socket
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
class Graphite(object):
|
||
|
|
|
||
|
|
def __init__(self, hostname, port):
|
||
|
|
self.hostname = hostname
|
||
|
|
self.port = int(port)
|
||
|
|
|
||
|
|
def flatten_dict(self, d):
|
||
|
|
def expand(key, value):
|
||
|
|
if isinstance(value, dict):
|
||
|
|
return [('{}.{}'.format(key, k), v) for k, v in self.flatten_dict(value).items()]
|
||
|
|
else:
|
||
|
|
return [(key, value)]
|
||
|
|
items = [item for k, v in d.items() for item in expand(k, v)]
|
||
|
|
return dict(items)
|
||
|
|
|
||
|
|
def update(self, prefix, metrics, nodes):
|
||
|
|
timestamp = int(time.time())
|
||
|
|
|
||
|
|
sock = socket.socket()
|
||
|
|
sock.connect((self.hostname, self.port))
|
||
|
|
|
||
|
|
for node in nodes:
|
||
|
|
try:
|
||
|
|
if node['flags']['online']:
|
||
|
|
stats = self.flatten_dict(node['statistics'])
|
||
|
|
for metric in metrics.split(','):
|
||
|
|
try:
|
||
|
|
msg = '{}{}.{} {} {}\n'.format(
|
||
|
|
prefix,
|
||
|
|
node['nodeinfo']['node_id'].replace(' ', '_'),
|
||
|
|
metric.replace(' ', '_'),
|
||
|
|
stats[metric],
|
||
|
|
timestamp
|
||
|
|
)
|
||
|
|
sock.send(msg.encode('utf-8'))
|
||
|
|
|
||
|
|
except KeyError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
except KeyError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
sock.close()
|