diff --git a/frigate/http.py b/frigate/http.py index cfddb8f11..83d35e8f7 100644 --- a/frigate/http.py +++ b/frigate/http.py @@ -85,11 +85,9 @@ def create_app( app.register_blueprint(bp) - - - app.wsgi_app = DispatcherMiddleware(app.wsgi_app, { - "/metrics": make_wsgi_app(registry=setupRegistry()) - }) + app.wsgi_app = DispatcherMiddleware( + app.wsgi_app, {"/metrics": make_wsgi_app(registry=setupRegistry())} + ) return app diff --git a/frigate/prometheus_exporter.py b/frigate/prometheus_exporter.py index 8b1f9ef27..66fc6dee0 100644 --- a/frigate/prometheus_exporter.py +++ b/frigate/prometheus_exporter.py @@ -5,7 +5,6 @@ from prometheus_client import CollectorRegistry from prometheus_client.metrics_core import GaugeMetricFamily, InfoMetricFamily - def setupRegistry(): myregistry = CollectorRegistry() myregistry.register(CustomCollector()) @@ -15,49 +14,68 @@ def setupRegistry(): def add_metric(metric, label, data, key, multiplier=1.0): try: string = str(data[key]) - value = float(re.findall(r'\d+', string)[0]) + value = float(re.findall(r"\d+", string)[0]) metric.add_metric([label], value * multiplier) except (KeyError, TypeError, IndexError): pass - -class CustomCollector(): +class CustomCollector: def __init__(self): self.stats_url = "http://localhost:5000/api/stats" - def collect(self): - - data = json.loads(urlopen(self.stats_url).read()) - # camera stats - ffmpeg_pid = GaugeMetricFamily('frigate_ffmpeg_pid', 'PID for ffmpeg process', labels=['camera']) - capture_pid = GaugeMetricFamily('frigate_capture_pid', 'PID for the ffmpeg process that consumes this camera', - labels=['camera']) - detect_pid = GaugeMetricFamily('frigate_detect_pid', 'PID for the process that runs detection for this camera', - labels=['camera']) - camera_fps = GaugeMetricFamily('frigate_camera_fps', 'Frames per second being consumed from your camera.', - labels=['camera']) - detection_fps = GaugeMetricFamily('frigate_detection_fps', 'Number of times detection is run per second.', - labels=['camera']) - process_fps = GaugeMetricFamily('frigate_process_fps', 'Frames per second being processed by frigate.', - labels=['camera']) - skipped_fps = GaugeMetricFamily('frigate_skipped_fps', 'Frames per second skip for processing by frigate.', - labels=['camera']) - detection_enabled = GaugeMetricFamily('frigate_detection_enabled', 'Detection enabled for camera', - labels=['camera']) + # camera stats + ffmpeg_pid = GaugeMetricFamily( + "frigate_ffmpeg_pid", "PID for ffmpeg process", labels=["camera"] + ) + capture_pid = GaugeMetricFamily( + "frigate_capture_pid", + "PID for the ffmpeg process that consumes this camera", + labels=["camera"], + ) + detect_pid = GaugeMetricFamily( + "frigate_detect_pid", + "PID for the process that runs detection for this camera", + labels=["camera"], + ) + camera_fps = GaugeMetricFamily( + "frigate_camera_fps", + "Frames per second being consumed from your camera.", + labels=["camera"], + ) + detection_fps = GaugeMetricFamily( + "frigate_detection_fps", + "Number of times detection is run per second.", + labels=["camera"], + ) + process_fps = GaugeMetricFamily( + "frigate_process_fps", + "Frames per second being processed by frigate.", + labels=["camera"], + ) + skipped_fps = GaugeMetricFamily( + "frigate_skipped_fps", + "Frames per second skip for processing by frigate.", + labels=["camera"], + ) + detection_enabled = GaugeMetricFamily( + "frigate_detection_enabled", + "Detection enabled for camera", + labels=["camera"], + ) for k, d in data.items(): - add_metric(ffmpeg_pid, k, d, 'ffmpeg_pid') - add_metric(detect_pid, k, d, 'pid') - add_metric(capture_pid, k, d, 'capture_pid') - add_metric(camera_fps, k, d, 'camera_fps') - add_metric(detection_fps, k, d, 'detection_fps') - add_metric(process_fps, k, d, 'process_fps') - add_metric(skipped_fps, k, d, 'skipped_fps') - add_metric(detection_enabled, k, d, 'detection_enabled') + add_metric(ffmpeg_pid, k, d, "ffmpeg_pid") + add_metric(detect_pid, k, d, "pid") + add_metric(capture_pid, k, d, "capture_pid") + add_metric(camera_fps, k, d, "camera_fps") + add_metric(detection_fps, k, d, "detection_fps") + add_metric(process_fps, k, d, "process_fps") + add_metric(skipped_fps, k, d, "skipped_fps") + add_metric(detection_enabled, k, d, "detection_enabled") yield ffmpeg_pid yield capture_pid @@ -70,25 +88,36 @@ class CustomCollector(): # detector stats try: - yield GaugeMetricFamily('frigate_detection_total_fps', - 'Sum of detection_fps across all cameras and detectors.', - value=data['detection_fps']) + yield GaugeMetricFamily( + "frigate_detection_total_fps", + "Sum of detection_fps across all cameras and detectors.", + value=data["detection_fps"], + ) except KeyError: pass - detector_inference_speed = GaugeMetricFamily('frigate_detector_inference_speed_seconds', - 'Time spent running object detection in seconds.', labels=['name']) - detector_pid = GaugeMetricFamily('frigate_detector_pid', - 'PID for the shared process that runs object detection on the detector', - labels=['name']) - detector_detection_start = GaugeMetricFamily('frigate_detection_start', - 'Detector start time (unix timestamp)', - labels=['name']) + detector_inference_speed = GaugeMetricFamily( + "frigate_detector_inference_speed_seconds", + "Time spent running object detection in seconds.", + labels=["name"], + ) + detector_pid = GaugeMetricFamily( + "frigate_detector_pid", + "PID for the shared process that runs object detection on the detector", + labels=["name"], + ) + detector_detection_start = GaugeMetricFamily( + "frigate_detection_start", + "Detector start time (unix timestamp)", + labels=["name"], + ) try: - for k, d in data['detectors'].items(): - add_metric(detector_inference_speed, k, d, 'inference_speed', 0.001) # ms to seconds - add_metric(detector_pid, k, d, 'pid') - add_metric(detector_detection_start, k, d, 'detection_start') + for k, d in data["detectors"].items(): + add_metric( + detector_inference_speed, k, d, "inference_speed", 0.001 + ) # ms to seconds + add_metric(detector_pid, k, d, "pid") + add_metric(detector_detection_start, k, d, "detection_start") except KeyError: pass @@ -97,13 +126,17 @@ class CustomCollector(): yield detector_detection_start # process stats - cpu_usages = GaugeMetricFamily('frigate_cpu_usage_percent', 'Process CPU usage %', labels=['pid']) - mem_usages = GaugeMetricFamily('frigate_mem_usage_percent', 'Process memory usage %', labels=['pid']) + cpu_usages = GaugeMetricFamily( + "frigate_cpu_usage_percent", "Process CPU usage %", labels=["pid"] + ) + mem_usages = GaugeMetricFamily( + "frigate_mem_usage_percent", "Process memory usage %", labels=["pid"] + ) try: - for k, d in data['cpu_usages'].items(): - add_metric(cpu_usages, k, d, 'cpu') - add_metric(mem_usages, k, d, 'mem') + for k, d in data["cpu_usages"].items(): + add_metric(cpu_usages, k, d, "cpu") + add_metric(mem_usages, k, d, "mem") except KeyError: pass @@ -111,13 +144,17 @@ class CustomCollector(): yield mem_usages # gpu stats - gpu_usages = GaugeMetricFamily('frigate_gpu_usage_percent', 'GPU utilisation %', labels=['gpu']) - gpu_mem_usages = GaugeMetricFamily('frigate_gpu_mem_usage_percent', 'GPU memory usage %', labels=['gpu']) + gpu_usages = GaugeMetricFamily( + "frigate_gpu_usage_percent", "GPU utilisation %", labels=["gpu"] + ) + gpu_mem_usages = GaugeMetricFamily( + "frigate_gpu_mem_usage_percent", "GPU memory usage %", labels=["gpu"] + ) try: - for k, d in data['gpu_usages'].items(): - add_metric(gpu_usages, k, d, 'gpu') - add_metric(gpu_usages, k, d, 'mem') + for k, d in data["gpu_usages"].items(): + add_metric(gpu_usages, k, d, "gpu") + add_metric(gpu_usages, k, d, "mem") except KeyError: pass @@ -125,17 +162,26 @@ class CustomCollector(): yield gpu_mem_usages # service stats - uptime_seconds = GaugeMetricFamily('frigate_service_uptime_seconds', 'Uptime seconds') - last_updated_timestamp = GaugeMetricFamily('frigate_service_last_updated_timestamp', - 'Stats recorded time (unix timestamp)') + uptime_seconds = GaugeMetricFamily( + "frigate_service_uptime_seconds", "Uptime seconds" + ) + last_updated_timestamp = GaugeMetricFamily( + "frigate_service_last_updated_timestamp", + "Stats recorded time (unix timestamp)", + ) try: - s = data['service'] - add_metric(uptime_seconds, '', s, 'uptime') - add_metric(last_updated_timestamp, '', s, 'last_updated') + s = data["service"] + add_metric(uptime_seconds, "", s, "uptime") + add_metric(last_updated_timestamp, "", s, "last_updated") - info = {'latest_version': data['service']['latest_version'], 'version': data['service']['version']} - yield InfoMetricFamily('frigate_service', 'Frigate version info', value=info) + info = { + "latest_version": data["service"]["latest_version"], + "version": data["service"]["version"], + } + yield InfoMetricFamily( + "frigate_service", "Frigate version info", value=info + ) except KeyError: pass @@ -145,17 +191,25 @@ class CustomCollector(): # service->temperatures: no data for me - storage_free = GaugeMetricFamily('frigate_storage_free_bytes', 'Storage free 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']) + storage_free = GaugeMetricFamily( + "frigate_storage_free_bytes", "Storage free 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: - for k, d in data['service']['storage'].items(): - add_metric(storage_free, k, d, 'free', 1e6) # MB to bytes - add_metric(storage_total, k, d, 'total', 1e6) # MB to bytes - add_metric(storage_used, k, d, 'used', 1e6) # MB to bytes - storage_mount_type.add_metric(k, {'mount_type': d['mount_type']}) + for k, d in data["service"]["storage"].items(): + add_metric(storage_free, k, d, "free", 1e6) # MB to bytes + add_metric(storage_total, k, d, "total", 1e6) # MB to bytes + add_metric(storage_used, k, d, "used", 1e6) # MB to bytes + storage_mount_type.add_metric(k, {"mount_type": d["mount_type"]}) except KeyError: pass @@ -163,4 +217,3 @@ class CustomCollector(): yield storage_mount_type yield storage_total yield storage_used -