Munin contributed plugins

#!/usr/bin/env python

"""

=head1 NAME

mongo_btree - MongoDB btree Plugin

=head1 APPLICABLE SYSTEMS

Works until MongoDB 2.7. The "indexCounters" field was removed in 2.8 version.

=head1 CONFIGURATION

    [mongo_btree]

    env.MONGO_DB_URI mongodb://user:password@host:port/dbname

=head1 AUTHOR

Original script there : https://github.com/comerford/mongo-munin

Doc added by Alban Espie-Guillon <alban.espie@alterway.fr>

=cut

"""

import urllib2

import sys

try:

    import json

except ImportError:

    import simplejson as json

def getServerStatus():

    raw = urllib2.urlopen( "http://127.0.0.1:28017/_status" ).read()

    return json.loads( raw )["serverStatus"]

def get():

    status = getServerStatus()

    if status["version"] >= "2.4.0":

        return getServerStatus()["indexCounters"]

    else:

        return getServerStatus()["indexCounters"]["btree"]

def doData():

    for k,v in get().iteritems():

        print( str(k) + ".value " + str(int(v)) )

def doConfig():

    print "graph_title MongoDB btree stats"

    print "graph_args --base 1000 -l 0"

    print "graph_vlabel mb ${graph_period}"

    print "graph_category db"

    for k in get():

        print k + ".label " + k

        print k + ".min 0"

        print k + ".type COUNTER"

        print k + ".max 500000"

        print k + ".draw LINE1"

if __name__ == "__main__":

    if len(sys.argv) > 1 and sys.argv[1] == "config":

        doConfig()

    else:

        doData()