mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-12-29 00:06:19 +01:00
a468ed316d
* Added stop_event to util.Process util.Process will take care of receiving signals when the stop_event is accessed in the subclass. If it never is, SystemExit is raised instead. This has the effect of still behaving like multiprocessing.Process when stop_event is not accessed, while still allowing subclasses to not deal with the hassle of setting it up. * Give each util.Process their own logger This will help to reduce boilerplate in subclasses. * Give explicit types to util.Process.__init__ This gives better type hinting in the editor. * Use util.Process facilities in AudioProcessor Boilerplate begone! * Removed pointless check in util.Process The log_listener.queue should never be None, unless something has gone extremely wrong in the log setup code. If we're that far gone, crashing is better. * Make sure faulthandler is enabled in all processes This has no effect currently since we're using the fork start_method. However, when we inevidably switch to forkserver (either by choice, or by upgrading to python 3.14+) not having this makes for some really fun failure modes :D
96 lines
2.6 KiB
Python
96 lines
2.6 KiB
Python
import faulthandler
|
|
import logging
|
|
import multiprocessing as mp
|
|
import signal
|
|
import sys
|
|
import threading
|
|
from functools import wraps
|
|
from logging.handlers import QueueHandler
|
|
from typing import Any, Callable, Optional
|
|
|
|
import frigate.log
|
|
|
|
|
|
class BaseProcess(mp.Process):
|
|
def __init__(
|
|
self,
|
|
*,
|
|
name: Optional[str] = None,
|
|
target: Optional[Callable] = None,
|
|
args: tuple = (),
|
|
kwargs: dict = {},
|
|
daemon: Optional[bool] = None,
|
|
):
|
|
super().__init__(
|
|
name=name, target=target, args=args, kwargs=kwargs, daemon=daemon
|
|
)
|
|
|
|
def start(self, *args, **kwargs):
|
|
self.before_start()
|
|
super().start(*args, **kwargs)
|
|
self.after_start()
|
|
|
|
def __getattribute__(self, name: str) -> Any:
|
|
if name == "run":
|
|
run = super().__getattribute__("run")
|
|
|
|
@wraps(run)
|
|
def run_wrapper(*args, **kwargs):
|
|
try:
|
|
self.before_run()
|
|
return run(*args, **kwargs)
|
|
finally:
|
|
self.after_run()
|
|
|
|
return run_wrapper
|
|
|
|
return super().__getattribute__(name)
|
|
|
|
def before_start(self) -> None:
|
|
pass
|
|
|
|
def after_start(self) -> None:
|
|
pass
|
|
|
|
def before_run(self) -> None:
|
|
pass
|
|
|
|
def after_run(self) -> None:
|
|
pass
|
|
|
|
|
|
class Process(BaseProcess):
|
|
logger: logging.Logger
|
|
|
|
@property
|
|
def stop_event(self) -> threading.Event:
|
|
# Lazily create the stop_event. This allows the signal handler to tell if anyone is
|
|
# monitoring the stop event, and to raise a SystemExit if not.
|
|
if "stop_event" not in self.__dict__:
|
|
self.__dict__["stop_event"] = threading.Event()
|
|
return self.__dict__["stop_event"]
|
|
|
|
def before_start(self) -> None:
|
|
self.__log_queue = frigate.log.log_listener.queue
|
|
|
|
def before_run(self) -> None:
|
|
faulthandler.enable()
|
|
|
|
def receiveSignal(signalNumber, frame):
|
|
# Get the stop_event through the dict to bypass lazy initialization.
|
|
stop_event = self.__dict__.get("stop_event")
|
|
if stop_event is not None:
|
|
# Someone is monitoring stop_event. We should set it.
|
|
stop_event.set()
|
|
else:
|
|
# Nobody is monitoring stop_event. We should raise SystemExit.
|
|
sys.exit()
|
|
|
|
signal.signal(signal.SIGTERM, receiveSignal)
|
|
signal.signal(signal.SIGINT, receiveSignal)
|
|
|
|
self.logger = logging.getLogger(self.name)
|
|
|
|
logging.basicConfig(handlers=[], force=True)
|
|
logging.getLogger().addHandler(QueueHandler(self.__log_queue))
|