riak_pbc_connection

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

"""
=head1 NAME

riak_pbc_connection - Plugin to monitor the number of Protocol Buffers connections.

=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 Protocol Buffers connections')
    print ('graph_args --base 1000 -l 0')
    print ('graph_vlabel count')
    print ('graph_category Riak')
    print ('graph_period second')
    print ('graph_info This graph shows the number of Protocol Buffers connections')

    print ('pbc_connection.label pbc connections')
    print ('pbc_connection.type GAUGE')
    print ('pbc_connection.max 1000000')
    print ('pbc_connection.min 0')
    print ('pbc_connection.info The number of Protocol Buffers connections')

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 ('pbc_connection.value %s') % ret['pbc_active']

if __name__ == '__main__':
    main()