ansible/roles/nagios_client/files/scripts/check_haproxy_conns.py

76 lines
1.8 KiB
Python
Executable file

#!/usr/bin/python3
""" Nagios check for haproxy over-subscription.
fedmsg-gateway is the primary concern as it can eat up a ton of simultaneous
connections.
:Author: Ralph Bean <rbean@redhat.com>
"""
import socket
import sys
def _numeric(value):
""" Type casting utility """
try:
return int(value)
except ValueError:
try:
return float(value)
except ValueError:
return value
def query(sockname):
""" Read stats from the haproxy socket and return a dict """
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
sock.connect(sockname)
send_string = 'show info\n'
sock.send(send_string.encode())
try:
response = sock.recv(2048).strip().decode()
lines = response.split('\n')
data = dict(map(str.strip, line.split(':')) for line in lines)
data = {k:_numeric(v) for (k, v) in data.items()}
return data
except Exception as err:
print(str(err))
finally:
sock.close()
return None
def nagios_check(data):
""" Print warnings and return nagios exit codes. """
current = data['CurrConns']
maxconn = data['Maxconn']
percent = 100 * float(current) / float(maxconn)
details = "%.2f%% subscribed. %i current of %i maxconn." % (
percent, current, maxconn,
)
if percent < 50:
print("HAPROXY SUBS OK: " + details)
return 0
if percent < 75:
print("HAPROXY SUBS WARN: " + details)
return 1
if percent <= 100:
print("HAPROXY SUBS CRIT: " + details)
return 2
print("HAPROXY SUBS UNKNOWN: " + details)
return 3
if __name__ == '__main__':
try:
data = query(sockname="/var/run/haproxy-stat")
except Exception as err:
print("HAPROXY SUBS UNKNOWN: " + str(err))
sys.exit(3)
sys.exit(nagios_check(data))