mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-26 19:06:11 +01:00
57864f2be6
Generally eliminate the `while True` loops while waiting for a stop event and prefer to condition the loops on if the stop event is set, blocking on that where it makes sense. This generally comes in 3 flavors. First and simplest, when there is a sleep and the stop event is the only thing the loop blocks on, instead do a check using `stop_event.wait(timeout)` to instead block on the stop event for the designated amount of time. Second, when there is a different event that is blocking in the loop, condition the loop on `stop_event.is_set()` rather than breaking when it is set. Finally, when there is a separate internal condition that requires a counter, have the loop iterate over the counter and use `if stop_event.wait(timeout)` internal to the loop.
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import datetime
|
|
import logging
|
|
import threading
|
|
import time
|
|
import os
|
|
import signal
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class FrigateWatchdog(threading.Thread):
|
|
def __init__(self, detectors, stop_event):
|
|
threading.Thread.__init__(self)
|
|
self.name = "frigate_watchdog"
|
|
self.detectors = detectors
|
|
self.stop_event = stop_event
|
|
|
|
def run(self):
|
|
time.sleep(10)
|
|
while not self.stop_event.wait(10):
|
|
now = datetime.datetime.now().timestamp()
|
|
|
|
# check the detection processes
|
|
for detector in self.detectors.values():
|
|
detection_start = detector.detection_start.value
|
|
if detection_start > 0.0 and now - detection_start > 10:
|
|
logger.info(
|
|
"Detection appears to be stuck. Restarting detection process..."
|
|
)
|
|
detector.start_or_restart()
|
|
elif not detector.detect_process.is_alive():
|
|
logger.info("Detection appears to have stopped. Exiting frigate...")
|
|
os.kill(os.getpid(), signal.SIGTERM)
|
|
|
|
logger.info(f"Exiting watchdog...")
|