Traefik-CheckMK/agent_based/traefik_udp_components.py

138 lines
5.2 KiB
Python

#!/usr/bin/env python3
# pylint: disable=missing-module-docstring, unused-argument, consider-using-f-string
# pylint: disable=missing-function-docstring, line-too-long
# import necessary elements from API version 2
from cmk.agent_based.v2 import (
AgentSection,
CheckPlugin,
Service,
State,
Metric,
Result,
DiscoveryResult,
CheckResult,
check_levels,
)
def parse_traefik_udp_components(string_table):
"""the parse function"""
parsed_data = {}
for line in string_table:
if line[0] == "routers":
parsed_data["num_routers"] = int(line[1])
parsed_data["num_routers_warn"] = int(line[2])
parsed_data["num_routers_crit"] = int(line[3])
elif line[0] == "services":
parsed_data["num_services"] = int(line[1])
parsed_data["num_services_warn"] = int(line[2])
parsed_data["num_services_crit"] = int(line[3])
return parsed_data
def discover_traefik_udp_components(section) -> DiscoveryResult:
"""the discover function"""
yield Service()
def check_traefik_udp_components(params, section) -> CheckResult:
"""the check function"""
_level_type, levels_percent_not_ok = params["levels_traefik_udp_components_not_ok"]
levels_min_routers = params["levels_traefik_min_udp_routers"]
levels_max_routers = params["levels_traefik_max_udp_routers"]
levels_min_services = params["levels_traefik_min_udp_services"]
levels_max_services = params["levels_traefik_max_udp_services"]
num_routers: int = section["num_routers"]
num_routers_warn: int = section["num_routers_warn"]
num_routers_crit: int = section["num_routers_crit"]
num_services: int = section["num_services"]
num_services_warn: int = section["num_services_warn"]
num_services_crit: int = section["num_services_crit"]
num_components: int = num_routers + num_services
components_warn: int = num_routers_warn + num_services_warn
components_crit: int = num_routers_crit + num_services_crit
components_percent_not_ok: float = 0.0
if num_components > 0:
components_percent_not_ok = (
(components_warn + components_crit) * 100 / num_components
)
yield Metric(
name="traefik_percent_udp_components_not_ok",
value=components_percent_not_ok,
levels=levels_percent_not_ok,
)
summary: str = f"Number of UDP routers/services: {num_routers}/{num_services}"
details_routers: str = (
f"Routers WARN: {num_routers_warn}\nRouters CRIT: {num_routers_crit}"
)
details_services: str = (
f"Services WARN: {num_services_warn}\nServices CRIT: {num_services_crit}"
)
details = f"{details_routers}\n\n{details_services}"
state: State = State.OK
if components_warn > 0:
state = State.WARN
if components_crit > 0:
state = State.CRIT
yield Result(
state=state,
summary=summary,
details=details,
)
yield from check_levels(
metric_name="traefik_num_udp_routers",
value=num_routers,
levels_lower=levels_min_routers,
levels_upper=levels_max_routers,
label="Number of UDP routers",
notice_only=True,
render_func=lambda v: "%.0f" % v,
)
yield from check_levels(
metric_name="traefik_num_udp_services",
value=num_services,
levels_lower=levels_min_services,
levels_upper=levels_max_services,
label="Number of UDP services",
notice_only=True,
render_func=lambda v: "%.0f" % v,
)
# create the new agent section, must begin with "agent_section_"
# and must be an instance of "AgentSection"
agent_section_traefik_udp_components = AgentSection(
# "name" must exactly match the section name within the agent output
name="traefik_udp_components",
# define the parse function, name is arbitrary, a good choice is to choose
# "parse_" as prefix and append the section name
parse_function=parse_traefik_udp_components,
)
# create the new check plugin, must begin with "check_plugin_"
# and must be an instance of "CheckPlugin"
check_plugin_traefik_udp_components = CheckPlugin(
# "name" should be the same as the corresponding section within the agent output
name="traefik_udp_components",
service_name="Traefik UDP components",
# define the discovery function, name is arbitrary, a good choice is to choose
# "discover_" as prefix and append the section name
discovery_function=discover_traefik_udp_components,
# define the check function, name is arbitrary, a good choice is to choose
# "check_" as prefix and append the section name
check_function=check_traefik_udp_components,
# define the default parameters
check_default_parameters={
"levels_traefik_min_udp_routers": ("fixed", (0, 0)),
"levels_traefik_max_udp_routers": ("fixed", (25, 50)),
"levels_traefik_min_udp_services": ("fixed", (0, 0)),
"levels_traefik_max_udp_services": ("fixed", (25, 50)),
"levels_traefik_udp_components_not_ok": ("fixed", (0.5, 1.0)),
},
# connect to the ruleset where parameters can be defined
# must match the name of the ruleset exactly
check_ruleset_name="traefik_udp_components",
)