115 lines
4.1 KiB
Python
115 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
# pylint: disable=missing-module-docstring, unused-argument,
|
|
# pylint: disable=missing-function-docstring, line-too-long
|
|
|
|
import datetime
|
|
import time
|
|
|
|
|
|
# import necessary elements from API version 2
|
|
from cmk.agent_based.v2 import (
|
|
AgentSection,
|
|
CheckPlugin,
|
|
Service,
|
|
State,
|
|
Metric,
|
|
Result,
|
|
DiscoveryResult,
|
|
CheckResult,
|
|
render,
|
|
)
|
|
|
|
|
|
def get_state_upper(
|
|
levels: tuple[int | float, int | float], value: int | float
|
|
) -> State:
|
|
"""returns OK/WARN/CRIT depending on the given parameters"""
|
|
warn, crit = levels
|
|
if value >= crit:
|
|
return State.CRIT
|
|
if value >= warn:
|
|
return State.WARN
|
|
return State.OK
|
|
|
|
|
|
def parse_traefik_info(string_table):
|
|
"""the parse function"""
|
|
parsed_data = {}
|
|
for line in string_table:
|
|
if line[0] == "Traefik_Version":
|
|
parsed_data["version"] = line[1]
|
|
elif line[0] == "Traefik_CodeName":
|
|
parsed_data["codename"] = line[1]
|
|
elif line[0] == "Traefik_StartDate":
|
|
parsed_data["startdate"] = line[1]
|
|
# convert timestamp into seconds since epoch
|
|
dt = datetime.datetime.fromisoformat(line[1])
|
|
parsed_data["startdate_seconds_since_epoch"] = int(dt.timestamp())
|
|
elif line[0] == "Agent_Runtime":
|
|
# convert seconds into ms
|
|
parsed_data["agent_runtime"] = float(line[1]) * 1_000
|
|
elif line[0] == "Agent_Version":
|
|
parsed_data["agent_version"] = line[1]
|
|
return parsed_data
|
|
|
|
|
|
def discover_traefik_info(section) -> DiscoveryResult:
|
|
"""the discover function"""
|
|
yield Service()
|
|
|
|
|
|
def check_traefik_info(params, section) -> CheckResult:
|
|
"""the check function"""
|
|
_level_type, levels = params["levels_traefik_agent_execution_time"]
|
|
codename: str = section["codename"]
|
|
version: str = section["version"]
|
|
startdate: str = section["startdate"]
|
|
startdate_seconds_since_epoch: int = section["startdate_seconds_since_epoch"]
|
|
# calculate runtime of Traefik in seconds
|
|
current_epoch_time: int = int(time.time())
|
|
runtime: int = current_epoch_time - startdate_seconds_since_epoch
|
|
agent_version: str = section["agent_version"]
|
|
agent_runtime: float = round(section["agent_runtime"], 1)
|
|
state: State = get_state_upper(levels=levels, value=agent_runtime)
|
|
|
|
yield Metric(
|
|
name="traefik_agent_execution_time",
|
|
value=agent_runtime,
|
|
levels=levels,
|
|
)
|
|
summary: str = f"Traefik version: {version}, code name: {codename}"
|
|
details: str = f"Traefik start date: {startdate}\nRunning since: {render.timespan(runtime)}\n\nAgent version: {agent_version}\nAgent execution time: {agent_runtime}ms"
|
|
yield Result(state=state, summary=summary, details=details)
|
|
|
|
|
|
# create the new agent section, must begin with "agent_section_"
|
|
# and must be an instance of "AgentSection"
|
|
agent_section_traefik_info = AgentSection(
|
|
# "name" must exactly match the section name within the agent output
|
|
name="traefik_info",
|
|
# 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_info,
|
|
)
|
|
|
|
# create the new check plugin, must begin with "check_plugin_"
|
|
# and must be an instance of "CheckPlugin"
|
|
check_plugin_traefik_info = CheckPlugin(
|
|
# "name" should be the same as the corresponding section within the agent output
|
|
name="traefik_info",
|
|
service_name="Traefik Info",
|
|
# 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_info,
|
|
# 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_info,
|
|
# define the default parameters
|
|
check_default_parameters={
|
|
"levels_traefik_agent_execution_time": ("fixed", (500.0, 750.0))
|
|
},
|
|
# connect to the ruleset where parameters can be defined
|
|
# must match the name of the ruleset exactly
|
|
check_ruleset_name="traefik_info",
|
|
)
|