109 lines
3.3 KiB
Python
109 lines
3.3 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
## TODO: Plugin output seems intermingled
|
||
|
|
## with the wrong plugin name
|
||
|
|
|
||
|
|
#from multiprocessing import Pool
|
||
|
|
#from datetime import datetime
|
||
|
|
import time
|
||
|
|
import socket
|
||
|
|
|
||
|
|
|
||
|
|
hostlist = ["localhost"]
|
||
|
|
|
||
|
|
class Munin():
|
||
|
|
def __init__(self, hostname="localhost", port=4949):
|
||
|
|
self.hostname = hostname
|
||
|
|
self.port = port
|
||
|
|
|
||
|
|
def connect(self):
|
||
|
|
# Use getdefaulttimeout() to figure out what the default timeout is.
|
||
|
|
self._sock = socket.create_connection((self.hostname, self.port), 10)
|
||
|
|
self._conn = self._sock.makefile()
|
||
|
|
self.hello_string = self._readline()
|
||
|
|
|
||
|
|
def _readline(self):
|
||
|
|
return self._conn.readline().strip()
|
||
|
|
|
||
|
|
def _iterline(self):
|
||
|
|
while True:
|
||
|
|
current_line = self._readline()
|
||
|
|
if not current_line:
|
||
|
|
break
|
||
|
|
if current_line.startswith("#"):
|
||
|
|
continue
|
||
|
|
if current_line == ".":
|
||
|
|
break
|
||
|
|
yield current_line
|
||
|
|
|
||
|
|
def fetch(self, plugin):
|
||
|
|
self._sock.sendall("fetch %s\n" % plugin)
|
||
|
|
plugin_data = {}
|
||
|
|
while True:
|
||
|
|
data = self._sock.recv(1024)
|
||
|
|
if not data:
|
||
|
|
break
|
||
|
|
response += data
|
||
|
|
|
||
|
|
return response
|
||
|
|
|
||
|
|
def close_connection(self):
|
||
|
|
self._sock.close()
|
||
|
|
|
||
|
|
def list_plugins(self):
|
||
|
|
"""Return a list of munin plugins configured on a node. """
|
||
|
|
self._sock.sendall("list\n")
|
||
|
|
return self._readline().split(" ")
|
||
|
|
|
||
|
|
def get_config(self, plugin):
|
||
|
|
self._sock.sendall("config %s\n" % plugin)
|
||
|
|
response = {}
|
||
|
|
for current_line in self._iterline():
|
||
|
|
key_name, key_value = current_line.split(" ", 1)
|
||
|
|
|
||
|
|
if "." in key_name:
|
||
|
|
# Some keys have periods in them.
|
||
|
|
# If so, make their own nested dictionary.
|
||
|
|
key_root, key_leaf = key_name.split(".", 1)
|
||
|
|
if key_root not in response:
|
||
|
|
response[key_root] = {}
|
||
|
|
response[key_root][key_leaf] = key_value
|
||
|
|
else:
|
||
|
|
response[key_name] = key_value
|
||
|
|
|
||
|
|
return response
|
||
|
|
|
||
|
|
def get_fetch(self, plugin):
|
||
|
|
self._sock.sendall("fetch %s\n" % plugin)
|
||
|
|
response = {}
|
||
|
|
for current_line in self._iterline():
|
||
|
|
full_key_name, key_value = current_line.split(" ")
|
||
|
|
key_name = full_key_name.split(".")[0]
|
||
|
|
response[key_name] = key_value
|
||
|
|
|
||
|
|
return response
|
||
|
|
|
||
|
|
|
||
|
|
def process_host_stats(self):
|
||
|
|
"""Given a Munin object, process its host data."""
|
||
|
|
plugins = self.list_plugins()
|
||
|
|
#print "Plugin List: %s" % plugins
|
||
|
|
for current_plugin in plugins:
|
||
|
|
#print "---------------"
|
||
|
|
#print "Fetching plugin: %s" % current_plugin
|
||
|
|
config = self.get_config(current_plugin)
|
||
|
|
#print "Plugin config: %s" % config
|
||
|
|
plugin_data = self.get_fetch(current_plugin)
|
||
|
|
#print "Plugin data: %s" % plugin_data
|
||
|
|
time.sleep(0.5)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
for current_host in hostlist:
|
||
|
|
print "querying host: %s" % current_host
|
||
|
|
munin_host = Munin(hostname=current_host)
|
||
|
|
munin_host.connect()
|
||
|
|
munin_host.process_host_stats()
|
||
|
|
munin_host.close_connection()
|
||
|
|
print "done querying host %s" % current_host
|