Nagios check for Rabbit cluster partitioning

check_rabbit_partitioning.py

#!/usr/bin/env python
"""
Nagios check to check RabbitMQ cluster for partitioning (split-brain)
"""

import json
import optparse
import logging
import requests
import sys

OK = 0
WARNING = 1
CRITICAL = 2
UNKNOWN = 3

LOGGER = logging.getLogger(__name__)
LOGGER.addHandler(logging.StreamHandler(sys.stdout))
LOGGER.setLevel(logging.INFO)

parse = optparse.OptionParser()
parse.add_option('-s', '--server', type='string', dest='rmq_server',
                 help='RabbitMQ hostname')
parse.add_option('-u', '--user', type='string', dest='rmq_user',
                 help='RabbitMQ username', default='guest')
parse.add_option('-p', '--password', type='string', dest='rmq_pass',
                 help='RabbitMQ password', default='guest')
parse.add_option('-P', '--port', type='int', dest='rmq_port',
                 help='RabbitMQ port', default='15672')

(opts, args) = parse.parse_args()

# Require at least one argument
if not opts.rmq_server:
    parse.print_help()
    exit(OK)

try:
    api_url = 'http://{0}:{1}@{2}:{3}/api/nodes/'.format(
        opts.rmq_user,
        opts.rmq_pass,
        opts.rmq_server,
        opts.rmq_port
    )
    data = requests.get(api_url).text

    if not data:
        LOGGER.info('UNKNOWN - No data returned from RabbitMQ')
        exit(UNKNOWN)
    if len(json.loads(data)[0]['partitions']) > 0:
        LOGGER.info('CRITICAL - Partition detected')
        exit(CRITICAL)
    else:
        LOGGER.info('OK - No partition detected')
        exit(OK)
except requests.ConnectionError:
    LOGGER.critical(
        'CRITICAL - Error connecting to %s:%s',
        opts.rmq_server,
        opts.rmq_port
    )
    exit(CRITICAL)