From 2a47fc47779d59100758ad27d0dcda25cb0567fe Mon Sep 17 00:00:00 2001
From: Leo Franchi <leo@humbughq.com>
Date: Mon, 28 Jan 2013 15:59:27 -0500
Subject: [PATCH] Add munin plugin for send-receive timing

(imported from commit e2ae0775379ce59ab43213e68ade4d3f88b578e6)
---
 humbug/bots/check_send_receive.py | 137 ++++++++++++++++++++++++++++++
 1 file changed, 137 insertions(+)
 create mode 100755 humbug/bots/check_send_receive.py

diff --git a/humbug/bots/check_send_receive.py b/humbug/bots/check_send_receive.py
new file mode 100755
index 0000000..5d0d6ce
--- /dev/null
+++ b/humbug/bots/check_send_receive.py
@@ -0,0 +1,137 @@
+#!/usr/bin/env python
+
+"""
+Script to provide information about send-receive times.
+
+It supports both munin and nagios outputs
+
+It must be run on a machine that is using the live database for the
+Django ORM.
+"""
+
+import datetime
+import os
+import sys
+import optparse
+import random
+
+
+def total_seconds(timedelta):
+    return (timedelta.microseconds + (timedelta.seconds + timedelta.days * 24 * 3600) * 10**6) / 10.**6
+
+usage = """Usage: send-receive.py [options] [config]
+
+       'config' is optional, if present will return config info.
+        Otherwise, returns the output data."""
+
+parser = optparse.OptionParser(usage=usage)
+parser.add_option('--site',
+                  dest='site',
+                  default="https://humbughq.com",
+                  action='store')
+
+parser.add_option('--nagios',
+                  dest='nagios',
+                  action='store_true')
+
+parser.add_option('--munin',
+                  dest='munin',
+                  action='store_true')
+(options, args) = parser.parse_args()
+
+if not options.nagios and not options.munin:
+    print 'No output options specified! Please provide --munin or --nagios'
+    sys.exit(0)
+
+if len(args) > 2:
+    print usage
+    sys.exit(0)
+
+if options.munin:
+    if len(args) and args[0] == 'config':
+        print \
+"""graph_title Send-Receive times
+graph_info The number of seconds it takes to send and receive a message from the server
+graph_args -u 5 -l 0
+graph_vlabel RTT (seconds)
+sendreceive.label Send-receive round trip time
+sendreceive.warning 3
+sendreceive.critical 5"""
+        sys.exit(0)
+
+sys.path.append('/home/humbug/humbug/api')
+import humbug
+
+states = {
+    "OK": 0,
+    "WARNING": 1,
+    "CRITICAL": 2,
+    "UNKNOWN": 3
+    }
+
+def report(state, time, msg=None):
+    if msg:
+        print "%s: %s" % (state, msg)
+    else:
+        print "%s: send time was %s" % (state, time)
+    exit(states[state])
+
+def send_humbug(sender, message, nagios):
+    result = sender.send_message(message)
+    if result["result"] != "success" and nagios:
+        report("CRITICAL", "Error sending Humbug, args were: %s, %s" % (message, result))
+
+def get_humbug(recipient, max_message_id):
+    return recipient.get_messages({'last': str(max_message_id)})['messages']
+
+# hamlet and othello are default users
+sender = "hamlet@humbughq.com"
+recipient = "othello@humbughq.com"
+
+humbug_sender = humbug.Client(
+    email=sender,
+    api_key="dfe1c934d555f4b9538d0d4cfd3069c2",
+    verbose=True,
+    client="test: Humbug API",
+    site=options.site)
+
+humbug_recipient = humbug.Client(
+    email=recipient,
+    api_key="4e5d97591bec64bf57d2698ffbb563e3",
+    verbose=True,
+    client="test: Humbug API",
+    site=options.site)
+
+
+max_message_id = humbug_recipient.get_profile().get('max_message_id')
+msg_to_send = str(random.getrandbits(64))
+time_start = datetime.datetime.now()
+
+send_humbug(humbug_sender, {
+    "type": 'private',
+    "content": msg_to_send,
+    "subject": "time to send",
+    "to": recipient,
+    }, options.nagios)
+
+msg_content = []
+
+while msg_to_send not in msg_content:
+    messages = get_humbug(humbug_recipient, max_message_id)
+    time_diff = datetime.datetime.now() - time_start
+
+    # Prevents checking the same messages everytime in the conditional
+    # statement of the while loop
+    max_message_id = max([msg['id'] for msg in messages])
+    msg_content = [m['content'] for m in messages]
+
+    if options.nagios:
+        if time_diff.seconds > 3:
+            report('WARNING', time_diff)
+        if time_diff.seconds > 6:
+            report('CRITICAL', time_diff)
+
+if options.munin:
+    print "sendreceive.value %s" % total_seconds(time_diff)
+elif options.nagios:
+    report('OK', time_diff)