Munin contributed plugins

#!/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()