Andreas Wrede
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
"""Monitor helper for heartbeat daemon.
|
|
|
|
This module provides monitoring tasks for the heartbeat daemon.
|
|
The primary reachability monitoring is now event-driven (timers set/reset
|
|
on HTB arrival in udp.py) rather than periodic polling.
|
|
|
|
This module can be extended for additional monitoring tasks.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
import asyncio
|
|
import time
|
|
from . import notify as notify_mod
|
|
|
|
DROPOVERDUE = 7 * 24 * 3600
|
|
eventlog = notify_mod.eventlog
|
|
|
|
|
|
async def cleanup_connections(hbdclass):
|
|
"""Clean up connection timers on shutdown.
|
|
|
|
Cancels all active overdue timers to prevent callbacks after shutdown.
|
|
"""
|
|
for hostname, host in list(hbdclass.Host.hosts.items()):
|
|
for conn_type, conn in host.connections.items():
|
|
if hasattr(conn, 'cancel_overdue_timer'):
|
|
conn.cancel_overdue_timer()
|
|
|
|
|
|
async def start(
|
|
config: dict,
|
|
hbdclass: callable,
|
|
pushmsg=None,
|
|
msg_to_websockets=None,
|
|
):
|
|
"""Start monitor background tasks.
|
|
|
|
Note: Reachability monitoring is now timer-based and happens in udp.py
|
|
when HTB messages arrive. This function can be used for additional
|
|
monitoring tasks.
|
|
|
|
Currently runs a simple status logger every 5 minutes.
|
|
"""
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
logger_interval = 300 # Log status every 5 minutes
|
|
|
|
while True:
|
|
await asyncio.sleep(logger_interval)
|
|
|
|
# Log monitoring status
|
|
total_hosts = len(hbdclass.Host.hosts)
|
|
up_count = sum(
|
|
1 for h in hbdclass.Host.hosts.values()
|
|
for c in h.connections.values()
|
|
if c.state == hbdclass.Connection.UP
|
|
)
|
|
overdue_count = sum(
|
|
1 for h in hbdclass.Host.hosts.values()
|
|
for c in h.connections.values()
|
|
if c.state == hbdclass.Connection.OVERDUE
|
|
)
|
|
|
|
logger.debug(
|
|
f"Monitor status: {total_hosts} hosts, {up_count} UP, {overdue_count} OVERDUE"
|
|
)
|