yieldbot/sensu-yieldbot-plugins

#!/usr/bin/env python

from optparse import OptionParser
import socket
import sys
import httplib
import json

PASS = 0
FAIL = 1

def get_bongo_host(server, app):
    try:
        con = httplib.HTTPConnection(server, timeout=45)
        con.request("GET","/v2/apps/" + app)
        data = con.getresponse()
        if data.status >= 300:
            print "Recieved non-2xx response= %s in get_bongo_host" % (data.status)
            sys.exit(FAIL)
        json_data = json.loads(data.read())
        host = json_data['app']['tasks'][0]['host']
        port = json_data['app']['tasks'][0]['ports'][0]
        con.close()
        return host, port
    except Exception, e:
        print "%s Exception caught in get_bongo_host" % (e)
        sys.exit(FAIL)

def get_status(host, group):
    try:
        con = httplib.HTTPConnection(host,timeout=45)
        con.request("GET","/v1/health/mismatch-" + group)
        data = con.getresponse()
        if data.status >= 300:
            print "Recieved non-2xx response= %s in get_status" % (data.status)
            sys.exit(FAIL)
        json_data = json.loads(data.read())
        con.close()
        if json_data['status'] == 1:
            print "%s" % (json_data['msg'])
            sys.exit(FAIL)
        else:
            print " `%s` nodes are fine" % group
            sys.exit(PASS)
    except Exception, e:
        print "%s Exception caught in get_status" % (e)
        sys.exit(FAIL)


if __name__=="__main__":
    parser = OptionParser()
    parser.add_option("-s", dest="server", action="store", default="localhost:8080", help="Marathon Cluster address with port no")
    parser.add_option("-a", dest="app", action="store", default="bongo.useast.prod", help="App Id to retrieve the slave address")
    parser.add_option("-c", dest="group", action="store", default="adserver", help="adserver/choosegoose")
    (options, args) = parser.parse_args()
    host, port = get_bongo_host(options.server, options.app)
    if "useast" in host:
        host = host.rsplit("prd",1)
        consul_host = "%snode.us-east-1.consul:%s" % (host[0], port)
    else:
        consul_host = "%s:%s" % (host, port)
    get_status(consul_host, options.group)