Use util.Process facilities in AudioProcessor

Boilerplate begone!
This commit is contained in:
George Tsiamasiotis 2024-10-03 19:27:44 +03:00
parent fe84887095
commit 7092ca3505

View File

@ -2,8 +2,6 @@
import datetime import datetime
import logging import logging
import signal
import sys
import threading import threading
import time import time
from typing import Tuple from typing import Tuple
@ -73,36 +71,31 @@ class AudioProcessor(util.Process):
): ):
super().__init__(name="frigate.audio_manager", daemon=True) super().__init__(name="frigate.audio_manager", daemon=True)
self.logger = logging.getLogger(self.name)
self.camera_metrics = camera_metrics self.camera_metrics = camera_metrics
self.cameras = cameras self.cameras = cameras
def run(self) -> None: def run(self) -> None:
stop_event = threading.Event()
audio_threads: list[AudioEventMaintainer] = [] audio_threads: list[AudioEventMaintainer] = []
threading.current_thread().name = "process:audio_manager" threading.current_thread().name = "process:audio_manager"
signal.signal(signal.SIGTERM, lambda sig, frame: sys.exit())
if len(self.cameras) == 0: if len(self.cameras) == 0:
return return
try:
for camera in self.cameras: for camera in self.cameras:
audio_thread = AudioEventMaintainer( audio_thread = AudioEventMaintainer(
camera, camera,
self.camera_metrics, self.camera_metrics,
stop_event, self.stop_event,
) )
audio_threads.append(audio_thread) audio_threads.append(audio_thread)
audio_thread.start() audio_thread.start()
self.logger.info(f"Audio processor started (pid: {self.pid})") self.logger.info(f"Audio processor started (pid: {self.pid})")
while True: while not self.stop_event.wait():
signal.pause() pass
finally:
stop_event.set()
for thread in audio_threads: for thread in audio_threads:
thread.join(1) thread.join(1)
if thread.is_alive(): if thread.is_alive():
@ -112,6 +105,7 @@ class AudioProcessor(util.Process):
for thread in audio_threads: for thread in audio_threads:
if thread.is_alive(): if thread.is_alive():
self.logger.warning(f"Thread {thread.name} is still alive") self.logger.warning(f"Thread {thread.name} is still alive")
self.logger.info("Exiting audio processor") self.logger.info("Exiting audio processor")