CollectD-python-haproxy-plugin/haproxy-plugin.py at master · AlexeyStolyarov/CollectD-python-haproxy-plugin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#!/usr/bin/python
# -*- coding: utf-8 -*-


import sys
import os
import commands
import socket
import time

try:
        import collectd
except:
        import tools.collectd_mockup    as collectd

from haproxy     import haproxy
from tools.data_list import data_gathered
from tools. data_list import data_stat_index


PLUGIN_NAME    = os.path.basename(__file__).split(".")[0]
stats                = haproxy.HAProxyStats('/var/run/haproxy/haproxy.sock')
VERBOSE_LOGGING = True

# convert list of "arg: val" to dict[arg]=val
def str2dict(arg):
    rez = {}
    for x in arg:
        if x:
            rez[(x.split(':')[0]).strip()] = x.split(':')[1].strip()

    return rez

def log_verbose(msg):
    if not VERBOSE_LOGGING:
        return
    collectd.info('haproxy plugin [verbose]: %s' % msg)


# callback funcion for collectd Daemon
def read_callback(data=None):

        metric = collectd.Values()

        #
        # get Haproxy stat info
        #
        rez = str2dict(stats.execute('show info'))
        instance = rez['node']    or socket.gethostname()
        # loop over keys that we need
        for i in    data_gathered['info']:

            metric.plugin = PLUGIN_NAME
            metric.interval = 60
            metric.plugin_instance = instance
            metric.type = 'status'
            metric.type_instance = i
            metric.values = (unicodedata.numeric(rez[i]),)    #values must be list or tuple
#            metric.dispatch()

        #
        # get stat info
        #
        f_ends = {}
        b_ends = {}
        b_ends_summ = {}
        for x in stats.execute('show stat'):

            # skiping some trash
            if x.startswith("#"):
                continue
            if x.startswith("stats"):
                continue
            if x == '':
                continue

            # convert string into array
            tmp = x.split(',')

            # fill frontends
            if tmp[1] == 'FRONTEND':
                f_ends[tmp[0]] = tmp
                continue

            # fill backend summary info
            if tmp[1] == 'BACKEND':
                b_ends_summ[tmp[0]] = tmp
                continue

            # fill backen workers info
            b_ends["%s@%s" % (tmp[1],tmp[0])] = tmp


            # fill data for backends
        for z in data_gathered['stat']:
            # fill info for backend/workers
            for k, v    in b_ends.iteritems():
                # skip empty values
                if v[data_stat_index[z]]:
                    metric.plugin = PLUGIN_NAME
                    metric.interval = 60

                    metric.plugin_instance = instance
                    metric.type =    'be_workers'
                    metric.type_instance = "%s_%s" % (k,z)

                    metric.values = (v[data_stat_index[z]],) #values must be list or tuple
                    metric.dispatch()
            # fill info for backend summary
            for k, v    in b_ends_summ.iteritems():
                # skip empty values
                if v[data_stat_index[z]]:
                    metric.plugin = PLUGIN_NAME
                    metric.interval = 60

                    metric.plugin_instance = instance
                    metric.type =    "backend"
                    metric.type_instance = "%s_%s" % (k,z)
                    metric.values = (v[data_stat_index[z]],)    # values must be list or tuple
                    metric.dispatch()
            # fill info for backend summary
            for k, v    in f_ends.iteritems():
                # skip empty values
                if v[data_stat_index[z]]:
                    metric.plugin = PLUGIN_NAME
                    metric.interval = 60

                    metric.plugin_instance = instance
                    metric.type =    "frontend"
                    metric.type_instance = "%s_%s" % (k,z)

                    metric.values = (v[data_stat_index[z]],) #values must be list or tuple
                    metric.dispatch()


#for debugging purposes
read_callback()

collectd.register_read(read_callback)