mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-21 19:07:46 +01:00
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
import json
|
|
import logging
|
|
import threading
|
|
import time
|
|
import psutil
|
|
import shutil
|
|
|
|
from frigate.config import FrigateConfig
|
|
from frigate.const import RECORD_DIR, CLIPS_DIR, CACHE_DIR
|
|
from frigate.version import VERSION
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def stats_init(camera_metrics, detectors):
|
|
stats_tracking = {
|
|
"camera_metrics": camera_metrics,
|
|
"detectors": detectors,
|
|
"started": int(time.time()),
|
|
}
|
|
return stats_tracking
|
|
|
|
|
|
def get_fs_type(path):
|
|
bestMatch = ""
|
|
fsType = ""
|
|
for part in psutil.disk_partitions(all=True):
|
|
if path.startswith(part.mountpoint) and len(bestMatch) < len(part.mountpoint):
|
|
fsType = part.fstype
|
|
bestMatch = part.mountpoint
|
|
return fsType
|
|
|
|
|
|
def stats_snapshot(stats_tracking):
|
|
camera_metrics = stats_tracking["camera_metrics"]
|
|
stats = {}
|
|
|
|
total_detection_fps = 0
|
|
|
|
for name, camera_stats in camera_metrics.items():
|
|
total_detection_fps += camera_stats["detection_fps"].value
|
|
stats[name] = {
|
|
"camera_fps": round(camera_stats["camera_fps"].value, 2),
|
|
"process_fps": round(camera_stats["process_fps"].value, 2),
|
|
"skipped_fps": round(camera_stats["skipped_fps"].value, 2),
|
|
"detection_fps": round(camera_stats["detection_fps"].value, 2),
|
|
"pid": camera_stats["process"].pid,
|
|
"capture_pid": camera_stats["capture_process"].pid,
|
|
}
|
|
|
|
stats["detectors"] = {}
|
|
for name, detector in stats_tracking["detectors"].items():
|
|
stats["detectors"][name] = {
|
|
"inference_speed": round(detector.avg_inference_speed.value * 1000, 2),
|
|
"detection_start": detector.detection_start.value,
|
|
"pid": detector.detect_process.pid,
|
|
}
|
|
stats["detection_fps"] = round(total_detection_fps, 2)
|
|
|
|
stats["service"] = {
|
|
"uptime": (int(time.time()) - stats_tracking["started"]),
|
|
"version": VERSION,
|
|
"storage": {},
|
|
}
|
|
|
|
for path in [RECORD_DIR, CLIPS_DIR, CACHE_DIR, "/dev/shm"]:
|
|
storage_stats = shutil.disk_usage(path)
|
|
stats["service"]["storage"][path] = {
|
|
"total": round(storage_stats.total / 1000000, 1),
|
|
"used": round(storage_stats.used / 1000000, 1),
|
|
"free": round(storage_stats.free / 1000000, 1),
|
|
"mount_type": get_fs_type(path),
|
|
}
|
|
|
|
return stats
|
|
|
|
|
|
class StatsEmitter(threading.Thread):
|
|
def __init__(
|
|
self,
|
|
config: FrigateConfig,
|
|
stats_tracking,
|
|
mqtt_client,
|
|
topic_prefix,
|
|
stop_event,
|
|
):
|
|
threading.Thread.__init__(self)
|
|
self.name = "frigate_stats_emitter"
|
|
self.config = config
|
|
self.stats_tracking = stats_tracking
|
|
self.mqtt_client = mqtt_client
|
|
self.topic_prefix = topic_prefix
|
|
self.stop_event = stop_event
|
|
|
|
def run(self):
|
|
time.sleep(10)
|
|
while True:
|
|
if self.stop_event.is_set():
|
|
logger.info(f"Exiting watchdog...")
|
|
break
|
|
stats = stats_snapshot(self.stats_tracking)
|
|
self.mqtt_client.publish(
|
|
f"{self.topic_prefix}/stats", json.dumps(stats), retain=False
|
|
)
|
|
time.sleep(self.config.mqtt.stats_interval)
|