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

formatter

Browse Source
This commit is contained in:
Mitch Ross 2023-04-24 21:59:22 -04:00
parent 1948fa6225
commit 5685a0ddb4
1 changed files with 184 additions and 69 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

253
frigate/monitoring/prometheus.py
View File

@ -14,19 +14,27 @@ def setupRegistry():
def add_metric(metric, label, stats, key, multiplier=1.0): def add_metric(metric, label, stats, key, multiplier=1.0):
try: try:
string = str(stats[key]) string = str(stats[key])
value = float(re.findall(r'\d+', string)[0]) value = float(re.findall(r"\d+", string)[0])
metric.add_metric([label], value * multiplier) metric.add_metric([label], value * multiplier)
except (KeyError, TypeError, IndexError): except (KeyError, TypeError, IndexError):
pass pass
class CustomCollector: class CustomCollector:
def __init__(self): def __init__(self):
self.stats_url = "http://localhost:5000/api/stats" self.stats_url = "http://localhost:5000/api/stats"
self.process_stats = {} self.process_stats = {}
def add_metric_process(self, metric, stats, camera_name, pid_name, process_name, cpu_or_memory, process_type): def add_metric_process(
self,
metric,
stats,
camera_name,
pid_name,
process_name,
cpu_or_memory,
process_type,
):
try: try:
pid = str(stats[camera_name][pid_name]) pid = str(stats[camera_name][pid_name])
label_values = [pid, camera_name, process_name, process_type] label_values = [pid, camera_name, process_name, process_type]
@ -37,40 +45,87 @@ class CustomCollector:
def collect(self): def collect(self):
stats = json.loads(urlopen(self.stats_url).read()) stats = json.loads(urlopen(self.stats_url).read())
self.process_stats = stats['cpu_usages'] self.process_stats = stats["cpu_usages"]
# process stats for cameras, detectors and other # process stats for cameras, detectors and other
cpu_usages = GaugeMetricFamily('frigate_cpu_usage_percent', 'Process CPU usage %', cpu_usages = GaugeMetricFamily(
labels=['pid', 'name', 'process', 'type']) "frigate_cpu_usage_percent",
mem_usages = GaugeMetricFamily('frigate_mem_usage_percent', 'Process memory usage %', "Process CPU usage %",
labels=['pid', 'name', 'process', 'type']) labels=["pid", "name", "process", "type"],
)
mem_usages = GaugeMetricFamily(
"frigate_mem_usage_percent",
"Process memory usage %",
labels=["pid", "name", "process", "type"],
)
# camera stats # camera stats
camera_fps = GaugeMetricFamily('frigate_camera_fps', 'Frames per second being consumed from your camera.', camera_fps = GaugeMetricFamily(
labels=['camera_name']) "frigate_camera_fps",
detection_fps = GaugeMetricFamily('frigate_detection_fps', 'Number of times detection is run per second.', "Frames per second being consumed from your camera.",
labels=['camera_name']) labels=["camera_name"],
process_fps = GaugeMetricFamily('frigate_process_fps', 'Frames per second being processed by frigate.', )
labels=['camera_name']) detection_fps = GaugeMetricFamily(
skipped_fps = GaugeMetricFamily('frigate_skipped_fps', 'Frames per second skip for processing by frigate.', "frigate_detection_fps",
labels=['camera_name']) "Number of times detection is run per second.",
detection_enabled = GaugeMetricFamily('frigate_detection_enabled', 'Detection enabled for camera', labels=["camera_name"],
labels=['camera_name']) )
process_fps = GaugeMetricFamily(
"frigate_process_fps",
"Frames per second being processed by frigate.",
labels=["camera_name"],
)
skipped_fps = GaugeMetricFamily(
"frigate_skipped_fps",
"Frames per second skip for processing by frigate.",
labels=["camera_name"],
)
detection_enabled = GaugeMetricFamily(
"frigate_detection_enabled",
"Detection enabled for camera",
labels=["camera_name"],
)
for camera_name, camera_stats in stats.items(): for camera_name, camera_stats in stats.items():
add_metric(camera_fps, camera_name, camera_stats, 'camera_fps') add_metric(camera_fps, camera_name, camera_stats, "camera_fps")
add_metric(detection_fps, camera_name, camera_stats, 'detection_fps') add_metric(detection_fps, camera_name, camera_stats, "detection_fps")
add_metric(process_fps, camera_name, camera_stats, 'process_fps') add_metric(process_fps, camera_name, camera_stats, "process_fps")
add_metric(skipped_fps, camera_name, camera_stats, 'skipped_fps') add_metric(skipped_fps, camera_name, camera_stats, "skipped_fps")
add_metric(detection_enabled, camera_name, camera_stats, 'detection_enabled') add_metric(
detection_enabled, camera_name, camera_stats, "detection_enabled"
)
self.add_metric_process(cpu_usages, stats, camera_name, 'ffmpeg_pid', 'ffmpeg', 'cpu', 'Camera') self.add_metric_process(
self.add_metric_process(cpu_usages, stats, camera_name, 'capture_pid', 'capture', 'cpu', 'Camera') cpu_usages, stats, camera_name, "ffmpeg_pid", "ffmpeg", "cpu", "Camera"
self.add_metric_process(cpu_usages, stats, camera_name, 'pid', 'detect', 'cpu', 'Camera') )
self.add_metric_process(
cpu_usages,
stats,
camera_name,
"capture_pid",
"capture",
"cpu",
"Camera",
)
self.add_metric_process(
cpu_usages, stats, camera_name, "pid", "detect", "cpu", "Camera"
)
self.add_metric_process(mem_usages, stats, camera_name, 'ffmpeg_pid', 'ffmpeg', 'mem', 'Camera') self.add_metric_process(
self.add_metric_process(mem_usages, stats, camera_name, 'capture_pid', 'capture', 'mem', 'Camera') mem_usages, stats, camera_name, "ffmpeg_pid", "ffmpeg", "mem", "Camera"
self.add_metric_process(mem_usages, stats, camera_name, 'pid', 'detect', 'mem', 'Camera') )
self.add_metric_process(
mem_usages,
stats,
camera_name,
"capture_pid",
"capture",
"mem",
"Camera",
)
self.add_metric_process(
mem_usages, stats, camera_name, "pid", "detect", "mem", "Camera"
)
yield camera_fps yield camera_fps
yield detection_fps yield detection_fps
@ -80,28 +135,59 @@ class CustomCollector:
# detector stats # detector stats
try: try:
yield GaugeMetricFamily('frigate_detection_total_fps', yield GaugeMetricFamily(
'Sum of detection_fps across all cameras and detectors.', "frigate_detection_total_fps",
value=stats['detection_fps']) "Sum of detection_fps across all cameras and detectors.",
value=stats["detection_fps"],
)
except KeyError: except KeyError:
pass pass
detector_inference_speed = GaugeMetricFamily('frigate_detector_inference_speed_seconds', detector_inference_speed = GaugeMetricFamily(
'Time spent running object detection in seconds.', labels=['name']) "frigate_detector_inference_speed_seconds",
"Time spent running object detection in seconds.",
labels=["name"],
)
detector_detection_start = GaugeMetricFamily('frigate_detection_start', detector_detection_start = GaugeMetricFamily(
'Detector start time (unix timestamp)', "frigate_detection_start",
labels=['name']) "Detector start time (unix timestamp)",
labels=["name"],
)
try: try:
for detector_name, detector_stats in stats['detectors'].items(): for detector_name, detector_stats in stats["detectors"].items():
add_metric(detector_inference_speed, detector_name, detector_stats, 'inference_speed', add_metric(
0.001) # ms to seconds detector_inference_speed,
add_metric(detector_detection_start, detector_name, detector_stats, 'detection_start') detector_name,
self.add_metric_process(cpu_usages, stats['detectors'], detector_name, 'pid', 'detect', 'cpu', detector_stats,
'Detector') "inference_speed",
self.add_metric_process(mem_usages, stats['detectors'], detector_name, 'pid', 'detect', 'mem', 0.001,
'Detector') ) # ms to seconds
add_metric(
detector_detection_start,
detector_name,
detector_stats,
"detection_start",
)
self.add_metric_process(
cpu_usages,
stats["detectors"],
detector_name,
"pid",
"detect",
"cpu",
"Detector",
)
self.add_metric_process(
mem_usages,
stats["detectors"],
detector_name,
"pid",
"detect",
"mem",
"Detector",
)
except KeyError: except KeyError:
pass pass
@ -111,8 +197,8 @@ class CustomCollector:
# remaining process stats # remaining process stats
try: try:
for process_id, pid_stats in self.process_stats.items(): for process_id, pid_stats in self.process_stats.items():
add_metric(cpu_usages, process_id, pid_stats, 'cpu') add_metric(cpu_usages, process_id, pid_stats, "cpu")
add_metric(mem_usages, process_id, pid_stats, 'mem') add_metric(mem_usages, process_id, pid_stats, "mem")
except KeyError: except KeyError:
pass pass
@ -120,13 +206,17 @@ class CustomCollector:
yield mem_usages yield mem_usages
# gpu stats # gpu stats
gpu_usages = GaugeMetricFamily('frigate_gpu_usage_percent', 'GPU utilisation %', labels=['gpu_name']) gpu_usages = GaugeMetricFamily(
gpu_mem_usages = GaugeMetricFamily('frigate_gpu_mem_usage_percent', 'GPU memory usage %', labels=['gpu_name']) "frigate_gpu_usage_percent", "GPU utilisation %", labels=["gpu_name"]
)
gpu_mem_usages = GaugeMetricFamily(
"frigate_gpu_mem_usage_percent", "GPU memory usage %", labels=["gpu_name"]
)
try: try:
for gpu_name, gpu_stats in stats['gpu_usages'].items(): for gpu_name, gpu_stats in stats["gpu_usages"].items():
add_metric(gpu_usages, gpu_name, gpu_stats, 'gpu') add_metric(gpu_usages, gpu_name, gpu_stats, "gpu")
add_metric(gpu_usages, gpu_name, gpu_stats, 'mem') add_metric(gpu_usages, gpu_name, gpu_stats, "mem")
except KeyError: except KeyError:
pass pass
@ -134,17 +224,26 @@ class CustomCollector:
yield gpu_mem_usages yield gpu_mem_usages
# service stats # service stats
uptime_seconds = GaugeMetricFamily('frigate_service_uptime_seconds', 'Uptime seconds') uptime_seconds = GaugeMetricFamily(
last_updated_timestamp = GaugeMetricFamily('frigate_service_last_updated_timestamp', "frigate_service_uptime_seconds", "Uptime seconds"
'Stats recorded time (unix timestamp)') )
last_updated_timestamp = GaugeMetricFamily(
"frigate_service_last_updated_timestamp",
"Stats recorded time (unix timestamp)",
)
try: try:
service_stats = stats['service'] service_stats = stats["service"]
add_metric(uptime_seconds, '', service_stats, 'uptime') add_metric(uptime_seconds, "", service_stats, "uptime")
add_metric(last_updated_timestamp, '', service_stats, 'last_updated') add_metric(last_updated_timestamp, "", service_stats, "last_updated")
info = {'latest_version': stats['service']['latest_version'], 'version': stats['service']['version']} info = {
yield InfoMetricFamily('frigate_service', 'Frigate version info', value=info) "latest_version": stats["service"]["latest_version"],
"version": stats["service"]["version"],
}
yield InfoMetricFamily(
"frigate_service", "Frigate version info", value=info
)
except KeyError: except KeyError:
pass pass
@ -154,17 +253,33 @@ class CustomCollector:
# service->temperatures: no data for me # service->temperatures: no data for me
storage_free = GaugeMetricFamily('frigate_storage_free_bytes', 'Storage free bytes', labels=['storage']) storage_free = GaugeMetricFamily(
storage_mount_type = InfoMetricFamily('frigate_storage_mount_type', 'Storage mount type', labels=['storage']) "frigate_storage_free_bytes", "Storage free bytes", labels=["storage"]
storage_total = GaugeMetricFamily('frigate_storage_total_bytes', 'Storage total bytes', labels=['storage']) )
storage_used = GaugeMetricFamily('frigate_storage_used_bytes', 'Storage used bytes', labels=['storage']) storage_mount_type = InfoMetricFamily(
"frigate_storage_mount_type", "Storage mount type", labels=["storage"]
)
storage_total = GaugeMetricFamily(
"frigate_storage_total_bytes", "Storage total bytes", labels=["storage"]
)
storage_used = GaugeMetricFamily(
"frigate_storage_used_bytes", "Storage used bytes", labels=["storage"]
)
try: try:
for storage_path, storage_stats in stats['service']['storage'].items(): for storage_path, storage_stats in stats["service"]["storage"].items():
add_metric(storage_free, storage_path, storage_stats, 'free', 1e6) # MB to bytes add_metric(
add_metric(storage_total, storage_path, storage_stats, 'total', 1e6) # MB to bytes storage_free, storage_path, storage_stats, "free", 1e6
add_metric(storage_used, storage_path, storage_stats, 'used', 1e6) # MB to bytes ) # MB to bytes
storage_mount_type.add_metric(storage_path, {'mount_type': storage_stats['mount_type']}) add_metric(
storage_total, storage_path, storage_stats, "total", 1e6
) # MB to bytes
add_metric(
storage_used, storage_path, storage_stats, "used", 1e6
) # MB to bytes
storage_mount_type.add_metric(
storage_path, {"mount_type": storage_stats["mount_type"]}
)
except KeyError: except KeyError:
pass pass