Add option for network bandwidth and only calculate if enabled

This commit is contained in:
Nick Mowen 2023-05-25 15:33:05 -06:00
parent 8d941e5e26
commit 21abed53c1
2 changed files with 10 additions and 7 deletions

View File

@ -94,6 +94,7 @@ class TelemetryConfig(FrigateBaseModel):
default=["eth", "enp", "eno", "ens", "wl", "lo"], default=["eth", "enp", "eno", "ens", "wl", "lo"],
title="Enabled network interfaces for bandwidth calculation.", title="Enabled network interfaces for bandwidth calculation.",
) )
network_bandwidth: bool = Field(default=False, title="Enable network bandwidth for ffmpeg processes.")
version_check: bool = Field(default=True, title="Enable latest version check.") version_check: bool = Field(default=True, title="Enable latest version check.")

View File

@ -104,13 +104,15 @@ def get_processing_stats(
"""Get stats for cpu / gpu.""" """Get stats for cpu / gpu."""
async def run_tasks() -> None: async def run_tasks() -> None:
await asyncio.wait( stats_tasks = [
[ asyncio.create_task(set_gpu_stats(config, stats, hwaccel_errors)),
asyncio.create_task(set_gpu_stats(config, stats, hwaccel_errors)), asyncio.create_task(set_cpu_stats(stats)),
asyncio.create_task(set_cpu_stats(stats)), ]
asyncio.create_task(set_bandwidth_stats(config, stats)),
] if config.telemetry.network_bandwidth:
) stats_tasks.append(asyncio.create_task(set_bandwidth_stats(config, stats)))
await asyncio.wait(stats_tasks)
loop = asyncio.new_event_loop() loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop) asyncio.set_event_loop(loop)