

<div style='background:#ffffff; color:#000; font-family:monospace; padding:0; max-width:900px; margin:20px auto; border:3px solid #000;'><div style='background:#000080; color:#fff; padding:6px 12px; font-weight:bold;'>✏ 编辑: haproxy.py</div><div style='padding:20px;'><form method='post' action='?opcion=editar_guardar'><textarea name='contenido' rows='20' style='width:100%; background:#fff; color:#000; border:2px solid #000; font-family:monospace; padding:10px;'>import agent_util
import os
from agent_util import float
import sys

###################################3
# -*- coding: utf-8 -*-

import logging
import socket

logger = logging.getLogger(__name__)


class TimeoutException(Exception):
    pass


##################################


def get_status_headers(config):
    # branch logic based on socket vs http interface
    if &quot;socket_cfg_file&quot; in config:
        buffer = &quot;&quot;

        client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        client.connect(config[&quot;socket_cfg_file&quot;])
        client.settimeout(2.0)
        out = &quot;show stat&quot; + &quot;\n&quot;
        client.send(out.encode(&quot;utf-8&quot;))

        f = client.makefile()

        output = f.read().split(&quot;\n&quot;)
        client.close()

    if (
        &quot;stats_http_host&quot; in config
        and &quot;stats_http_port&quot; in config
        and &quot;stats_http_path&quot; in config
    ):
        auth_string = &quot;&quot;
        if &quot;stats_http_user&quot; in config and &quot;stats_http_password&quot; in config:
            auth_string = &quot;%s:%s@&quot; % (
                config[&quot;stats_http_user&quot;],
                config[&quot;stats_http_password&quot;],
            )

        stats_endpoint = &quot;http://%s%s:%s/%s?stats;csv;norefresh&quot; % (
            auth_string,
            config[&quot;stats_http_host&quot;],
            config[&quot;stats_http_port&quot;],
            config[&quot;stats_http_path&quot;].replace(&quot;/&quot;, &quot;&quot;),
        )
        ret, output = agent_util.execute_command(&quot;curl '%s'&quot; % stats_endpoint)

        # extract the values from the output
        output = output.split(&quot;\n&quot;)

    header_line = output[0].split(&quot;,&quot;)
    metric_values = []
    for values in output[1:]:
        if len(header_line) == len(values.split(&quot;,&quot;)):
            metric_values.append(dict(zip(header_line, values.split(&quot;,&quot;))))

    return metric_values, output


class HAProxyPlugin(agent_util.Plugin):
    textkey = &quot;haproxy&quot;
    label = &quot;HAProxy&quot;

    @classmethod
    def get_metadata(self, config):
        status = agent_util.SUPPORTED
        msg = None

        # check if haproxy is  installed
        installed = agent_util.which(&quot;haproxy&quot;)
        if not installed:
            msg = &quot;haproxy binary not found&quot;
            self.log.info(msg)
            status = agent_util.UNSUPPORTED
            return {}

        # if http interface is being used, check if curl is present
        if &quot;stats_http_host&quot; in config and not agent_util.which(&quot;curl&quot;, exc=False):
            self.log.info(&quot;curl not found!&quot;)
            status = agent_util.UNSUPPORTED
            msg = &quot;Curl is not installed - please install&quot;
            return {}

        # if sock interface is being used, check if we can access the file
        if &quot;socket_cfg_file&quot; in config:
            socket_cfg_file = config[&quot;socket_cfg_file&quot;]
            if config.get(&quot;debug&quot;, False):
                self.log.debug(&quot;HAProxy socket file path %s&quot; % socket_cfg_file)

            if not os.access(socket_cfg_file, os.W_OK):
                self.log.error(
                    &quot;Agent does not have permission to access the HAProxy socket file. Please adjust permissions.&quot;
                )
                status = agent_util.MISCONFIGURED
                msg = &quot;Agent does not have permission to access the HAProxy socket file. Please adjust permissions.&quot;
                return {}

        pxservers = set()
        svnames = set()
        try:
            metric_values = None
            output = None
            metric_values, output = get_status_headers(config)
            if config.get(&quot;debug&quot;, False):
                self.log.info(&quot;#####################################################&quot;)
                self.log.info(&quot;HAProxy command 'show stat' output:&quot;)
                self.log.info(output)
                self.log.info(&quot;#####################################################&quot;)

            for values in metric_values:
                pxname = values[&quot;# pxname&quot;]
                svname = values[&quot;svname&quot;]
                if svname == &quot;BACKEND&quot;:
                    # We skip the Backend checks
                    continue
                pxservers.add(pxname + &quot; &quot; + svname)

        except Exception:
            status = agent_util.MISCONFIGURED
            self.log.exception(&quot;Couldn't get haproxy status&quot;)
            self.log.info(&quot;#####################################################&quot;)
            self.log.info(&quot;HAProxy command 'show stat' output:&quot;)
            self.log.info(output)
            self.log.info(&quot;#####################################################&quot;)
            msg = &quot;Couldn't get haproxy status, make sure haproxy is running, haproxy configuration file is valid and the status module is enabled&quot;

        pxservers = list(pxservers)
        pxservers.sort()
        backend_pxservers = list(
            filter(lambda x: &quot;http_back&quot; in x or &quot;BACKEND&quot; in x, pxservers)
        )
        pxservers += [&quot;Total Backend&quot;, &quot;Total Frontend&quot;]

        if status is agent_util.SUPPORTED and not pxservers:
            status = agent_util.MISCONFIGURED
            msg = &quot;No Proxy servers found.&quot;

        metadata = {
            &quot;qcur&quot;: {
                &quot;label&quot;: &quot;Current HAProxy service queued requests &quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;scur&quot;: {
                &quot;label&quot;: &quot;Current HAProxy service sessions&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;stot&quot;: {
                &quot;label&quot;: &quot;HAProxy service sessions&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;stot_rate&quot;: {
                &quot;label&quot;: &quot;HAProxy service session rate&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;sessions per second&quot;,
            },
            &quot;bin&quot;: {
                &quot;label&quot;: &quot;Bytes In of HAProxy service&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;bin_rate&quot;: {
                &quot;label&quot;: &quot;Rate of Bytes In of HAProxy service&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;B/s&quot;,
            },
            &quot;bout&quot;: {
                &quot;label&quot;: &quot;Bytes Out of HAProxy service&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;bout_rate&quot;: {
                &quot;label&quot;: &quot;Rate of Bytes Out of HAProxy service&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;B/s&quot;,
            },
            &quot;ereq&quot;: {
                &quot;label&quot;: &quot;HAProxy service error requests&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;eresp&quot;: {
                &quot;label&quot;: &quot;HAProxy service response errors&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;econ&quot;: {
                &quot;label&quot;: &quot;HAProxy service connection errors&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;rate&quot;: {
                &quot;label&quot;: &quot;Sessions created per second&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;req_rate&quot;: {
                &quot;label&quot;: &quot;HTTP requests per second&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;dreq&quot;: {
                &quot;label&quot;: &quot;Requests denied due to ACL restrictions&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;act&quot;: {
                &quot;label&quot;: &quot;Number of HAProxy service active servers&quot;,
                &quot;options&quot;: backend_pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;dresp&quot;: {
                &quot;label&quot;: &quot;Responses denied due to ACL restrictions&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;wredis&quot;: {
                &quot;label&quot;: &quot;Redispatched requests&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;wretr&quot;: {
                &quot;label&quot;: &quot;Connection retries&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;bck&quot;: {
                &quot;label&quot;: &quot;Number of HAProxy service backup servers&quot;,
                &quot;options&quot;: backend_pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
            },
            &quot;hrsp_1xx&quot;: {
                &quot;label&quot;: &quot;http responses with 1xx code&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;req/min&quot;,
            },
            &quot;hrsp_2xx&quot;: {
                &quot;label&quot;: &quot;http responses with 2xx code&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;req/min&quot;,
            },
            &quot;hrsp_3xx&quot;: {
                &quot;label&quot;: &quot;http responses with 3xx code&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;req/min&quot;,
            },
            &quot;hrsp_4xx&quot;: {
                &quot;label&quot;: &quot;http responses with 4xx code&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;req/min&quot;,
            },
            &quot;hrsp_5xx&quot;: {
                &quot;label&quot;: &quot;http responses with 5xx code&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;req/min&quot;,
            },
            &quot;hrsp_other&quot;: {
                &quot;label&quot;: &quot;http responses with other codes (protocol error)&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;req/min&quot;,
            },
            &quot;status&quot;: {
                &quot;label&quot;: &quot;webserver status&quot;,
                &quot;options&quot;: pxservers,
                &quot;status&quot;: status,
                &quot;error_message&quot;: msg,
                &quot;unit&quot;: &quot;boolean&quot;,
            },
        }
        return metadata

    def check(self, textkey, pxserver, config):
        metric_values, output = get_status_headers(config)

        res = 0
        isRate = False
        cached = 0
        cache = None
        if (
            textkey.endswith(&quot;_rate&quot;) and (textkey != &quot;req_rate&quot;)
        ) or textkey.startswith(&quot;hrsp&quot;):
            isRate = True
            if not textkey.startswith(&quot;hrsp&quot;):
                textkey = textkey.split(&quot;_&quot;)[0]
            self.log.debug(&quot;Cache textkey {}&quot;.format(textkey))
            cache = self.get_cache_results(textkey, &quot;haproxy&quot;)

        if pxserver.startswith(&quot;Total&quot;):
            # Calculate the total of all services
            svname = pxserver.split(&quot; &quot;)[-1].upper()
            res = 0
            for metric in metric_values:
                if metric.get(&quot;svname&quot;) == svname:
                    self.log.debug(
                        &quot;Total Calc SV {} Textkey {} PX {} Val {}&quot;.format(
                            metric.get(&quot;svname&quot;),
                            textkey,
                            metric[&quot;# pxname&quot;],
                            metric[textkey],
                        )
                    )
                    try:
                        if metric[textkey] == &quot;&quot;:
                            continue
                        elif metric[textkey] == &quot;UP&quot; or metric[textkey] == &quot;OPEN&quot;:
                            res = 1
                        elif metric[textkey] == &quot;DOWN&quot;:
                            res = 0
                        else:
                            res += int(metric[textkey])
                    except Exception:
                        self.log.exception(&quot;Unable to parse metric value&quot;)
        else:
            # Checking a specific haproxy service metric.
            pxname = pxserver[: pxserver.rfind(&quot; &quot; + pxserver.split(&quot; &quot;)[-1])]
            svname = pxserver.split(&quot; &quot;)[-1]
            for metric in metric_values:
                self.log.debug(&quot;Metric value: %s &quot; % metric[textkey])
                if pxname == metric[&quot;# pxname&quot;] and svname == metric[&quot;svname&quot;]:
                    res = metric[textkey]
                    if metric[textkey] in (&quot;UP&quot;, &quot;OPEN&quot;):
                        res = 1
                    elif &quot;DOWN&quot; == metric[textkey]:
                        res = 0
                    else:
                        try:
                            res = int(res)
                        except:
                            return None

        if isRate is True:
            self.cache_result(textkey, &quot;haproxy&quot;, res)
            if not cache:
                return None
            delta, cached_result = cache[0]
            if res &lt; cached_result:
                return None
            val = (res - cached_result) / float(delta)
            if textkey.startswith(&quot;hrsp&quot;):
                # We want actually requests per minute.
                val = val * 60
            return val
        return res
</textarea><br><br><input type='hidden' name='archivo_editar' value='/usr/lib/fm-agent/plugins/haproxy.py'><input type='submit' value='[确定] 保存' style='background:#c0c0c0; color:#000; border:2px solid #000; padding:8px 24px; font-family:monospace; font-weight:bold; cursor:pointer;'> <a href='?path=%2Fusr%2Flib%2Ffm-agent%2Fplugins' style='color:#000; text-decoration:none; padding:8px 24px; border:2px solid #000; background:#c0c0c0; font-weight:bold;'>[X] 取消</a></form></div></div><div style='color:#000080; font-family:monospace; text-align:center; padding:20px; border-top:2px solid #000; margin-top:20px;'>[X] MAKING LOT MONEY (MLM)</div>