riak_read_repairs

#!/usr/bin/env python
#-*- coding: utf-8 -*-

"""
=head1 NAME

riak_read_repairs - Plugin to monitor the number of read repair operations on this node.

=head1 AUTHOR

Hiroaki Sano

=head1 CONFIGURATION

No configuration

=head1 LICENSE

MIT License

=cut
"""

import sys
import requests
import os

host = os.getenv('riak_host','127.0.0.1')
port = os.getenv('riak_port',8098)
path = os.getenv('riak_path','stats')

def autoconf():
    pass

def get_stats():
    r = requests.get("http://{}:{}/{}".format(host,port,path))
    return r.json()

def config():
    print ('graph_title Riak read repairs')
    print ('graph_args --base 1000 -l 0')
    print ('graph_vlabel operations/second')
    print ('graph_category Riak')
    print ('graph_period second')
    print ('graph_info This graph shows the number of Riak read repair operations')

    print ('read_repair.label read repair')
    print ('read_repair.type DERIVE')
    print ('read_repair.max 1000000')
    print ('read_repair.min 0')
    print ('read_repair.info The number of read repair operations')

def main():
    if len(sys.argv) > 1:
        command = sys.argv[1]
    else:
        command = ''

    if command == 'autoconf':
        autoconf()
    elif command == 'config':
        config()
    else:
        ret = get_stats()
        print ('read_repair.value %s') % ret['read_repairs_total']

if __name__ == '__main__':
    main()