Gecko [ goons4good.org ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2022-10-11 05:09
__init__.py	18.16 KB	-rw-r--r--	2022-05-16 09:16
adb.py	5.2 KB	-rw-r--r--	2022-05-16 09:16
aix_account.py	1.33 KB	-rw-r--r--	2022-05-16 09:16
avahi_announce.py	8.53 KB	-rw-r--r--	2022-05-16 09:16
bonjour_announce.py	8.21 KB	-rw-r--r--	2022-05-16 09:16
btmp.py	8.33 KB	-rw-r--r--	2022-05-16 09:16
cert_info.py	5.82 KB	-rw-r--r--	2022-05-16 09:16
diskusage.py	3.35 KB	-rw-r--r--	2022-05-16 09:16
glxinfo.py	1.84 KB	-rw-r--r--	2022-05-16 09:16
haproxy.py	3.02 KB	-rw-r--r--	2022-05-16 09:16
inotify.py	11.88 KB	-rw-r--r--	2022-05-16 09:16
journald.py	2.61 KB	-rw-r--r--	2022-05-16 09:16
junos_rre_keys.py	723 B	-rw-r--r--	2022-05-16 09:16
load.py	5.95 KB	-rw-r--r--	2022-05-16 09:16
log_beacon.py	3.67 KB	-rw-r--r--	2022-05-16 09:16
memusage.py	1.57 KB	-rw-r--r--	2022-05-16 09:16
napalm_beacon.py	11.61 KB	-rw-r--r--	2022-05-16 09:16
network_info.py	4.66 KB	-rw-r--r--	2022-05-16 09:16
network_settings.py	6.53 KB	-rw-r--r--	2022-05-16 09:16
pkg.py	2.57 KB	-rw-r--r--	2022-05-16 09:16
proxy_example.py	1.48 KB	-rw-r--r--	2022-05-16 09:16
ps.py	2.23 KB	-rw-r--r--	2022-05-16 09:16
salt_monitor.py	4.06 KB	-rw-r--r--	2022-05-16 09:16
salt_proxy.py	1.81 KB	-rw-r--r--	2022-05-16 09:16
sensehat.py	2.82 KB	-rw-r--r--	2022-05-16 09:16
service.py	6.15 KB	-rw-r--r--	2022-05-16 09:16
sh.py	3.1 KB	-rw-r--r--	2022-05-16 09:16
smartos_imgadm.py	2.6 KB	-rw-r--r--	2022-05-16 09:16
smartos_vmadm.py	3.34 KB	-rw-r--r--	2022-05-16 09:16
status.py	4.11 KB	-rw-r--r--	2022-05-16 09:16
swapusage.py	1.57 KB	-rw-r--r--	2022-05-16 09:16
telegram_bot_msg.py	2.45 KB	-rw-r--r--	2022-05-16 09:16
twilio_txt_msg.py	2.65 KB	-rw-r--r--	2022-05-16 09:16
watchdog.py	4.79 KB	-rw-r--r--	2022-05-16 09:16
wtmp.py	10.17 KB	-rw-r--r--	2022-05-16 09:16

Rename

"""
Watch current connections of haproxy server backends.
Fire an event when over a specified threshold.

.. versionadded:: 2016.11.0
"""
import logging

import salt.utils.beacons

log = logging.getLogger(__name__)

__virtualname__ = "haproxy"

def __virtual__():
    """
    Only load the module if haproxyctl module is installed
    """
    if "haproxy.get_sessions" in __salt__:
        return __virtualname__
    else:
        log.debug("Not loading haproxy beacon")
        return False

def validate(config):
    """
    Validate the beacon configuration
    """
    if not isinstance(config, list):
        return False, "Configuration for haproxy beacon must be a list."
    else:
        config = salt.utils.beacons.list_to_dict(config)

if "backends" not in config:
            return False, "Configuration for haproxy beacon requires backends."
        else:
            if not isinstance(config["backends"], dict):
                return False, "Backends for haproxy beacon must be a dictionary."
            else:
                for backend in config["backends"]:
                    log.debug("config %s", config["backends"][backend])
                    if "servers" not in config["backends"][backend]:
                        return (
                            False,
                            "Backends for haproxy beacon require servers.",
                        )
                    else:
                        _servers = config["backends"][backend]["servers"]
                        if not isinstance(_servers, list):
                            return (
                                False,
                                "Servers for haproxy beacon must be a list.",
                            )
    return True, "Valid beacon configuration"

def beacon(config):
    """
    Check if current number of sessions of a server for a specific haproxy backend
    is over a defined threshold.

.. code-block:: yaml

beacons:
          haproxy:
            - backends:
                www-backend:
                    threshold: 45
                    servers:
                      - web1
                      - web2
            - interval: 120
    """
    ret = []

config = salt.utils.beacons.list_to_dict(config)

for backend in config.get("backends", ()):
        backend_config = config["backends"][backend]
        threshold = backend_config["threshold"]
        for server in backend_config["servers"]:
            scur = __salt__["haproxy.get_sessions"](server, backend)
            if scur:
                if int(scur) > int(threshold):
                    _server = {
                        "server": server,
                        "scur": scur,
                        "threshold": threshold,
                    }
                    log.debug(
                        "Emit because %s > %s for %s in %s",
                        scur,
                        threshold,
                        server,
                        backend,
                    )
                    ret.append(_server)
    return ret