netx/netx_api/worker.py
oliver d56020d84c Harden runtime budgets: DB pool, CLI concurrency, timeouts, and metrics.
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>
2026-08-02 18:02:43 +08:00

54 lines
1.3 KiB
Python

"""Background worker process for long-running schedulers.
Optional when ``NETX_RUN_INLINE_SCHEDULERS=false`` (API does not start collectors):
python -m netx_api.worker
Starts: config_sync, lldp_collect, port_traffic tick loops.
UME WS / keepalive remain in the API process (token + alarm coordination).
By default the API runs collectors inline — no separate worker needed.
"""
from __future__ import annotations
import logging
import signal
import threading
import time
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s %(levelname)s %(name)s: %(message)s",
)
_log = logging.getLogger("netx.worker")
def main() -> None:
stop = threading.Event()
def _handle(_sig: int, _frame: object) -> None:
_log.info("shutdown signal received")
stop.set()
for sig in (signal.SIGINT, signal.SIGTERM):
try:
signal.signal(sig, _handle)
except Exception:
pass
from .ume_runtime import start_device_schedulers
start_device_schedulers()
_log.info("netx worker schedulers started (config_sync, lldp_collect, port_traffic)")
while not stop.is_set():
time.sleep(1.0)
from .app_shutdown import shutdown_runtime
shutdown_runtime(reason="worker_signal")
_log.info("netx worker exiting")
if __name__ == "__main__":
main()