Refactor audio capture and processing threads to use more descriptive thread names

This commit is contained in:
Sergey Krashevich 2023-06-19 10:48:41 +03:00
parent 386e388f75
commit 3bd46db1ae
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
2 changed files with 7 additions and 8 deletions

View File

@ -10,10 +10,9 @@ import threading
import numpy as np
from setproctitle import setproctitle
from frigate.config import CameraConfig, AudioModelConfig
from frigate.config import AudioModelConfig, CameraConfig
from frigate.object_detection import RemoteObjectDetector
from frigate.util import listen, SharedMemoryFrameManager
from frigate.util import SharedMemoryFrameManager, listen
logger = logging.getLogger(__name__)
@ -31,8 +30,8 @@ def capture_audio(
signal.signal(signal.SIGTERM, receiveSignal)
signal.signal(signal.SIGINT, receiveSignal)
threading.current_thread().name = f"capture:{name}"
setproctitle(f"frigate.capture:{name}")
threading.current_thread().name = f"capture_audio:{name}"
setproctitle(f"frigate.capture_audio:{name}")
listen()
chunk_size = int(round(model_config.duration * model_config.sample_rate * 2))
@ -84,8 +83,8 @@ def process_audio(
signal.signal(signal.SIGTERM, receiveSignal)
signal.signal(signal.SIGINT, receiveSignal)
threading.current_thread().name = f"process:{name}"
setproctitle(f"frigate.process:{name}")
threading.current_thread().name = f"process_audio:{name}"
setproctitle(f"frigate.process_audio:{name}")
listen()
shape = (int(round(model_config.duration * model_config.sample_rate)),)

View File

@ -2,13 +2,13 @@
import logging
import multiprocessing as mp
import numpy as np
import signal
import subprocess as sp
import threading
from types import FrameType
from typing import Optional
import numpy as np
from setproctitle import setproctitle
from frigate.config import CameraConfig, FrigateConfig