581f337021
Wrap sync_config, send_heartbeat, and queue.flush operations in separate try/except to ensure daemon survives periodic operation exceptions. Critical for unattended 24/7 pilot devices where a locked database or disk error would otherwise kill the entire loop. Per-frame step() handling unchanged.
56 lines
2 KiB
Python
56 lines
2 KiB
Python
from edge.daemon import run_loop
|
|
|
|
|
|
class _Runtime:
|
|
def __init__(self): self.steps = 0; self.applied = None
|
|
def step(self, now): self.steps += 1; return []
|
|
def apply_zones(self, zones): self.applied = zones
|
|
|
|
|
|
class _Client:
|
|
def __init__(self): self.configs = 0; self.beats = 0
|
|
def fetch_config(self): self.configs += 1; return {"zones": [], "thresholds": {}}
|
|
def send_heartbeat(self, m): self.beats += 1; return True
|
|
|
|
|
|
class _Queue:
|
|
def __init__(self): self.flushes = 0
|
|
def flush(self): self.flushes += 1; return 0
|
|
|
|
|
|
def test_loop_steps_and_periodics():
|
|
rt, cl, q = _Runtime(), _Client(), _Queue()
|
|
ticks = iter(range(1000))
|
|
counters = run_loop(rt, cl, q, clock=lambda: float(next(ticks)),
|
|
sleep=lambda s: None, iterations=10,
|
|
sync_every=5, heartbeat_every=5, flush_every=5)
|
|
assert rt.steps == 10
|
|
assert cl.configs == 2 and cl.beats == 2 and q.flushes == 2 # at frames 5 and 10
|
|
assert rt.applied == [] # zones applied from fetched config
|
|
|
|
|
|
def test_loop_survives_step_exception():
|
|
class Boom(_Runtime):
|
|
def step(self, now):
|
|
self.steps += 1
|
|
if self.steps == 2: raise RuntimeError("frame glitch")
|
|
return []
|
|
rt = Boom()
|
|
counters = run_loop(rt, _Client(), _Queue(), clock=lambda: 0.0,
|
|
sleep=lambda s: None, iterations=3,
|
|
sync_every=99, heartbeat_every=99, flush_every=99)
|
|
assert rt.steps == 3 # kept going past the exception
|
|
assert counters["errors"] == 1
|
|
|
|
|
|
def test_loop_survives_flush_exception():
|
|
class BoomQueue(_Queue):
|
|
def flush(self):
|
|
self.flushes += 1
|
|
raise RuntimeError("db locked")
|
|
rt = _Runtime()
|
|
counters = run_loop(rt, _Client(), BoomQueue(), clock=lambda: 0.0,
|
|
sleep=lambda s: None, iterations=5,
|
|
sync_every=99, heartbeat_every=99, flush_every=2)
|
|
assert rt.steps == 5 # loop kept running past the flush exceptions
|
|
assert counters["errors"] >= 1
|