Change zabbix_get_haproxy_stats.py to generate a dynamic list with metrics instead...

[uio-zabbix.git] / zabbix_get_haproxy_stats.py

#!/usr/bin/python2
#

#
# Authors:
# rafael@E-MC2.NET / https://e-mc2.net/
#
# Copyright (c) 2018 USIT-University of Oslo
#
# zabbix_get_haproxy_stats.py is free software: you can
# redistribute it and/or modify it under the terms of the GNU General
# Public License as published by the Free Software Foundation, either
# version 3 of the License, or (at your option) any later version.
#
# zabbix_get_haproxy_stats.py is distributed in the hope that it
# will be useful, but WITHOUT ANY WARRANTY; without even the implied
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with sms_send.  If not, see <http://www.gnu.org/licenses/>.
#

import os
import sys
import tempfile

from io import StringIO
import errno
import socket
import subprocess

# Zabbix proxy
zabbix_proxy = ['zabbix-proxy-prod03.uio.no','zabbix-proxy-prod04.uio.no']

# Path to zabbix_sender
zabbix_sender = '/usr/bin/zabbix_sender'

# Temp file with full json output
tmp_stat_file = tempfile.NamedTemporaryFile(delete=False,dir='/tmp')

# Socket parameters
SOCKET_HOSTNAME = socket.gethostname()
SOCKET_BUFFER_SIZE = 1024
SOCKET_PATH = "/var/lib/haproxy/stats"

parameters = "pxname,svname,qcur,qmax,scur,smax,slim,stot,bin,bout,dreq,dresp,ereq,econ,eresp,wretr,wredis,status,weight,act,bck,chkfail,chkdown,lastchg,downtime,qlimit,pid,iid,sid,throttle,lbtot,tracked,type,rate,rate_lim,rate_max,check_status,check_code,check_duration,hrsp_1xx,hrsp_2xx,hrsp_3xx,hrsp_4xx,hrsp_5xx,hrsp_other,hanafail,req_rate,req_rate_max,req_tot,cli_abrt,srv_abrt,comp_in,comp_out,comp_byp,comp_rsp,lastsess,last_chk,last_agt,qtime,ctime,rtime,ttime".split(",")


# ############################################
# get_haproxy_stats_data()
# ############################################

def get_haproxy_stats_data(command):

    try:

        sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        sock.connect(SOCKET_PATH)

        sock.sendall((command + "\n").encode())
        buff = StringIO()
        
        while True:
            chunk = sock.recv(SOCKET_BUFFER_SIZE)

            if chunk:
                buff.write(chunk.decode("ascii"))
            else:
                break

        sock.close()

        response = buff.getvalue()
        buff.close()
    
        return response.rstrip("\n")
        
    except Exception as e:
        raise Exception("[ERROR]: %s\n" % e)


# ############################################
# generate_zabbix_autodiscovery()
# ############################################

def generate_zabbix_sender_data(stats):

    try:

        result = ""
        
        lines = stats.split("\n")

        # Generate a dynamic list with all the metrics delivered by
        # haproxy stats

        lines[0] = lines[0].replace("# ","")
        fields = lines.pop(0).split(",")
        
        for line in lines:
            values = line.split(",")

            if values[1] == 'FRONTEND':

                for loop in xrange(2,len(values)-1):
                    if values[loop] != '':
                        result += SOCKET_HOSTNAME + " haproxy.frontend.stats[" + values[0] + "," + fields[loop] + "] " + values[loop] + "\n"

            if values[1] == 'BACKEND':
            
                for loop in xrange(2,len(values)-1):
                    if values[loop] != '':
                        result += SOCKET_HOSTNAME + " haproxy.backend.stats[" + values[0] + "," + fields[loop] + "] " + values[loop] + "\n"

            if values[1] not in ['FRONTEND','BACKEND']:

                for loop in xrange(2,len(values)-1):
                    if values[loop] != '':
                        result += SOCKET_HOSTNAME + " haproxy.server.stats[" + values[0] + "/" + values[1] + "," + fields[loop] + "] " + values[loop] + "\n"
                        

        return result

    except Exception as e:
        raise Exception("[ERROR]: %s\n" % e)


if __name__ == '__main__':

    try:

        stats = get_haproxy_stats_data("show stat")
        result = generate_zabbix_sender_data(stats)

        #
        # We create a file with the data that zabbix_sender will
        # send in a bulk execution.
        #

        with open(tmp_stat_file.name,'w') as f: 
            f.write(result)
           
        #
        # The monitoring of this host can be done by any of the
        # zabbix proxies defined in zabbix_proxy[]. We try all of
        # them until one of them accepts our data
        #
        
        for proxy in zabbix_proxy:

            command = zabbix_sender + ' -z ' + proxy + ' -i ' + tmp_stat_file.name + ' > /dev/null 2>&1'
            
            proc = subprocess.Popen([command],shell=True)
            proc.wait()
            
            if proc.returncode == 0:
                break
            
        # Delete temp file with zabbix_sender data
        os.remove(tmp_stat_file.name)
        
        # Return value 0 = execution OK
        print("0")

    except Exception, e:
        print("1")
        sys.exit(1)