administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Lazily initialize util.Process stop_event

Browse Source
This commit is contained in:
George Tsiamasiotis 2024-10-02 10:19:02 +03:00
parent e8aeaf7521
commit e204dfc1c3
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
frigate/util/process.py
View File

@ -2,6 +2,7 @@ import faulthandler
import logging import logging
import multiprocessing as mp import multiprocessing as mp
import signal import signal
import sys
from functools import wraps from functools import wraps
from logging.handlers import QueueHandler from logging.handlers import QueueHandler
from multiprocessing.synchronize import Event from multiprocessing.synchronize import Event
@ -52,6 +53,12 @@ class Process(BaseProcess):
logger: logging.Logger logger: logging.Logger
stop_event: Event stop_event: Event
@property
def stop_event(self) -> Event:
if "stop_event" not in self.__dict__:
self.__dict__["stop_event"] = mp.Event()
return self.__dict__["stop_event"]
def before_start(self) -> None: def before_start(self) -> None:
self.__log_queue = frigate.log.log_listener.queue self.__log_queue = frigate.log.log_listener.queue
@ -63,10 +70,14 @@ class Process(BaseProcess):
self.logger = logging.getLogger(self.name) self.logger = logging.getLogger(self.name)
self.stop_event = mp.Event()
def receiveSignal(signalNumber, frame): def receiveSignal(signalNumber, frame):
self.stop_event.set() # Make sure we do not trigger stop_event lazy initialization
stop_event = self.__dict__.get("stop_event")
if stop_event is not None:
stop_event.set()
else:
sys.exit()
signal.signal(signal.SIGTERM, receiveSignal) signal.signal(signal.SIGTERM, receiveSignal)
signal.signal(signal.SIGINT, receiveSignal) signal.signal(signal.SIGINT, receiveSignal)