mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 04:20:45 +08:00
oliver
d56020d84c
Add shared CLI budget/timeout with force-close, parallel port-traffic dispatch, unified shutdown, bounded audit queue, output/log caps, and /metrics probes. Co-authored-by: Cursor <cursoragent@cursor.com>
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""Centralized runtime shutdown for API / worker processes."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
_log = logging.getLogger("netx.shutdown")
|
|
|
|
|
|
def shutdown_runtime(*, reason: str = "lifespan") -> None:
|
|
"""Best-effort stop of schedulers, sidebands, pools, and sessions."""
|
|
_log.info("shutdown_runtime begin reason=%s", reason)
|
|
|
|
try:
|
|
from .config_sync_scheduler import stop_config_sync_scheduler
|
|
|
|
stop_config_sync_scheduler()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("stop_config_sync_scheduler failed")
|
|
|
|
try:
|
|
from .lldp_collect_scheduler import stop_lldp_collect_scheduler
|
|
|
|
stop_lldp_collect_scheduler()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("stop_lldp_collect_scheduler failed")
|
|
|
|
try:
|
|
from .port_traffic_scheduler import stop_port_traffic_scheduler
|
|
|
|
stop_port_traffic_scheduler()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("stop_port_traffic_scheduler failed")
|
|
|
|
try:
|
|
import netx_api.ume_support as ume_support
|
|
|
|
if ume_support._UME_WS_STOP_EVENT is not None:
|
|
ume_support._UME_WS_STOP_EVENT.set()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("set UME_WS_STOP_EVENT failed")
|
|
|
|
try:
|
|
from .ume_alarm_ws import shutdown_ws_consumer
|
|
|
|
shutdown_ws_consumer()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("shutdown_ws_consumer failed")
|
|
|
|
try:
|
|
from .oclaw_alarm_forwarder import shutdown_oclaw_alarm_forwarder
|
|
|
|
shutdown_oclaw_alarm_forwarder()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("shutdown_oclaw_alarm_forwarder failed")
|
|
|
|
try:
|
|
from .webcrt_session_registry import close_all_sessions
|
|
|
|
close_all_sessions()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("close_all_webcrt_sessions failed")
|
|
|
|
try:
|
|
from .cli_timeout import shutdown_cli_timeout_pool
|
|
|
|
shutdown_cli_timeout_pool(wait=False)
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("shutdown_cli_timeout_pool failed")
|
|
|
|
try:
|
|
from .audit_async import shutdown_audit_worker
|
|
|
|
shutdown_audit_worker(timeout_sec=2.0)
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("shutdown_audit_worker failed")
|
|
|
|
try:
|
|
from .ne_collect_runner import shutdown_ne_collect_executor
|
|
from .ne_connect import shutdown_ne_connect_executor
|
|
from .config_sync_runner import shutdown_config_sync_pools
|
|
|
|
shutdown_ne_collect_executor()
|
|
shutdown_ne_connect_executor()
|
|
shutdown_config_sync_pools()
|
|
except Exception: # noqa: BLE001
|
|
_log.exception("shutdown executors failed")
|
|
|
|
_log.info("shutdown_runtime done reason=%s", reason)
|