Munin contributed plugins

#!/usr/bin/env python3

"""

=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 json

import urllib.request

import sys

def getServerStatus():

    raw = urllib.request.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().items():

        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()