mirror of https://github.com/zulip/zulip.git
56 lines
1.3 KiB
Python
Executable File
56 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import re
|
|
import time
|
|
|
|
from os import path, environ
|
|
|
|
sys.path.append(path.join(path.dirname(__file__), '..'))
|
|
from zulip_tools import check_output
|
|
|
|
WARN_THRESHOLD = 50
|
|
CRIT_THRESHOLD = 100
|
|
|
|
states = {
|
|
0: "OK",
|
|
1: "WARNING",
|
|
2: "CRITICAL",
|
|
3: "UNKNOWN"
|
|
}
|
|
|
|
re = re.compile(r'(\w+)\t(\d+)')
|
|
output = check_output(['/usr/sbin/rabbitmqctl', 'list_queues'], shell=False)
|
|
|
|
status = 0
|
|
max_count = 0
|
|
warn_queues = []
|
|
|
|
if 'USER' in environ and not environ['USER'] in ['root', 'rabbitmq']:
|
|
print "This script must be run as the root or rabbitmq user"
|
|
|
|
for line in output.split("\n"):
|
|
line = line.strip()
|
|
m = re.match(line)
|
|
if m:
|
|
queue = m.group(1)
|
|
count = int(m.group(2))
|
|
this_status = 0
|
|
if count > CRIT_THRESHOLD:
|
|
this_status = 2
|
|
warn_queues.append(queue)
|
|
elif count > WARN_THRESHOLD:
|
|
this_status = max(status, 1)
|
|
warn_queues.append(queue)
|
|
|
|
status = max(status, this_status)
|
|
max_count = max(max_count, count)
|
|
|
|
warn_about = ", ".join(warn_queues)
|
|
now = int(time.time())
|
|
|
|
if status > 0:
|
|
print("%s|%s|%s|max count %s, queues affected: %s" % (now, status, states[status], max_count, warn_about))
|
|
else:
|
|
print("%s|%s|%s|queues normal, max count %s" % (now, status, states[status], max_count))
|