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"],
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.")

View File

@ -104,13 +104,15 @@ def get_processing_stats(
"""Get stats for cpu / gpu."""
async def run_tasks() -> None:
await asyncio.wait(
[
asyncio.create_task(set_gpu_stats(config, stats, hwaccel_errors)),
asyncio.create_task(set_cpu_stats(stats)),
asyncio.create_task(set_bandwidth_stats(config, stats)),
]
)
stats_tasks = [
asyncio.create_task(set_gpu_stats(config, stats, hwaccel_errors)),
asyncio.create_task(set_cpu_stats(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()
asyncio.set_event_loop(loop)