Lagt inn zabbix_gfs2_autodiscovery.pl

[uio-zabbix.git] / zabbix_haproxy_autodiscovery.py

#!/usr/bin/env python
#

#
# Authors:
# rafael@E-MC2.NET / https://e-mc2.net/
#
# Copyright (c) 2018 USIT-University of Oslo
#
# zabbix_haproxy_autodiscovery.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_haproxy_autodiscovery.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 json
import argparse

from io import StringIO
import errno
import socket


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_autodiscovery(stats,type):

    try:
        output_list = []
        
        lines = stats.split("\n")
        fields = lines.pop(0).split(",")

        for line in lines:
            values = line.split(",")

            if values[1] == 'FRONTEND' and type == 'FRONTEND':

                frontends = {"{#FRONTEND}": values[0]}
                output_list.append(frontends)
                
            if values[1] == 'BACKEND' and type == 'BACKEND':

                backends = {"{#BACKEND}": values[0]}
                output_list.append(backends)

            if values[1] not in ['FRONTEND','BACKEND'] and type == 'SERVER':

                servers = {"{#SERVER}": values[0] + "/" + values[1]}
                output_list.append(servers)    

        result = {"data":output_list}
        print json.dumps(result,sort_keys=True,indent=2)

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


if __name__ == '__main__':

    try:

        #
        # Type = FRONTEND | BACKEND | SERVER
        # 
        type = sys.argv[1].upper()
        
        stats = get_haproxy_stats_data("show stat")
        generate_zabbix_autodiscovery(stats,type)

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