78 lines
1.6 KiB
Python
Executable File
78 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
=head1 NAME
|
|
|
|
nutcracker_requests_ - monitor twemproxy (aka: nutcracker)
|
|
|
|
|
|
=head1 CONFIGURATION
|
|
|
|
Config in /etc/munin/plugin-conf.d/nutcracker.conf:
|
|
|
|
[nutcracker_*]
|
|
env.NUTCRACKER_STATS_HOST 127.0.0.1
|
|
env.NUTCRACKER_STATS_PORT 22222
|
|
|
|
|
|
=head1 AUTHORS
|
|
|
|
Copyright 2013 Edgar Veiga <edgarmveiga@gmail.com>
|
|
|
|
=cut
|
|
"""
|
|
|
|
|
|
import json
|
|
import socket
|
|
import os
|
|
import sys
|
|
|
|
|
|
def get_stats():
|
|
data = ''
|
|
|
|
HOST = os.environ.get('NUTCRACKER_STATS_HOST', '127.0.0.1')
|
|
PORT = int(os.environ.get('NUTCRACKER_STATS_PORT', 22222))
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
s.connect((HOST, PORT))
|
|
|
|
file = s.makefile('r')
|
|
data = file.readline()
|
|
s.close()
|
|
|
|
return json.loads(data)
|
|
|
|
|
|
def process_data():
|
|
data = get_stats()
|
|
# get pools
|
|
for key, value in data.items():
|
|
if isinstance(value, dict):
|
|
total = 0
|
|
# get server requests
|
|
for pool_key, pool_value in value.items():
|
|
if isinstance(pool_value, dict):
|
|
total += pool_value["requests"]
|
|
print("requests_" + key + ".value" + " " + str(total))
|
|
|
|
|
|
def process_config():
|
|
print("graph_title Nutcracker requests/s")
|
|
print("graph_category other")
|
|
print("graph_vlabel requests/s")
|
|
|
|
data = get_stats()
|
|
for key, value in data.items():
|
|
if isinstance(value, dict):
|
|
print("requests_" + key + ".label " + key)
|
|
print("requests_" + key + ".type COUNTER")
|
|
print("requests_" + key + ".min 0")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1 and sys.argv[1] == "config":
|
|
process_config()
|
|
else:
|
|
process_data()
|