Munin contributed plugins

#!/usr/bin/env python3

# pylint: disable=invalid-name

# pylint: enable=invalid-name

"""Munin plugin to monitor MongoDB status.

=head1 NAME

mongodb_multi - monitor MongoDB connections, documents, memory and operations

=head1 APPLICABLE SYSTEMS

Systems with MongoDB installed. Tested up to MongoDB version 4.4.

=head1 CONFIGURATION

Defaults to mongodb://localhost, but can be configured:

    [mongodb_multi]

    env.uri mongodb://user:password@host:port/dbname?parameters

=head1 AUTHOR

Kim B. Heino <b@bbbs.net>

This is based heavily on non-multigraph MongoDB plugins.

=head1 LICENSE

GPLv2

=cut

"""

import os

import sys

import pymongo

URI = os.environ.get('uri', 'mongodb://localhost')

def server_status():

    """Get Mongo status."""

    conn = pymongo.MongoClient(URI)

    return conn.admin.command('serverStatus', workingSet=True)

def print_data(status):

    """Plugin values."""

    # conn

    print('multigraph mongo_conn')

    print('connections.value {}'.format(status['connections']['current']))

    # docs

    print('multigraph mongo_docs')

    for key, value in status['metrics']['document'].items():

        print('{}.value {}'.format(key, value))

    # mem

    print('multigraph mongo_mem')

    for key, value in status['mem'].items():

        if key in ('resident', 'virtual', 'mapped'):

            print('{}.value {}'.format(key, value * 1024 * 1024))

    # ops

    print('multigraph mongo_ops')

    for key, value in status['opcounters'].items():

        print('{}.value {}'.format(key, value))

def print_config(status):

    """Plugin config."""

    # conn

    print('multigraph mongo_conn')

    print('graph_title MongoDB connections')

    print('graph_args --base 1000 -l 0')

    print('graph_vlabel Connections')

    print('graph_category db')

    print('connections.label connections')

    # docs

    print('multigraph mongo_docs')

    print('graph_title MongoDB documents')

    print('graph_args --base 1000 -l 0')

    print('graph_vlabel Documents / ${graph_period}')

    print('graph_category db')

    for key in status['metrics']['document']:

        print('{0}.label {0}'.format(key))

        print('{}.type DERIVE'.format(key))

        print('{}.min 0'.format(key))

    # mem

    print('multigraph mongo_mem')

    print('graph_title MongoDB memory usage')

    print('graph_args --base 1024 -l 0')

    print('graph_vlabel Bytes')

    print('graph_category db')

    for key in status['mem']:

        if key in ('resident', 'virtual', 'mapped'):

            print('{0}.label {0}'.format(key))

    # ops

    print('multigraph mongo_ops')

    print('graph_title MongoDB operations')

    print('graph_args --base 1000 -l 0')

    print('graph_vlabel ops / ${graph_period}')

    print('graph_category db')

    print('graph_total total')

    for key in status['opcounters']:

        print('{0}.label {0}'.format(key))

        print('{}.type DERIVE'.format(key))

        print('{}.min 0'.format(key))

    if os.environ.get('MUNIN_CAP_DIRTYCONFIG') == '1':

        print_data(status)

if __name__ == '__main__':

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

        print_config(server_status())

    else:

        print_data(server_status())