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

Fix async

Browse Source
This commit is contained in:
Nick Mowen 2022-11-13 14:20:24 -07:00
parent b93eb689fa
commit 39a5e2587e
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
frigate/stats.py
View File

@ -84,6 +84,20 @@ def get_temperatures() -> dict[str, float]:
return temps return temps
def get_processing_stats(config: FrigateConfig, stats: dict[str, str]) -> None:
"""Get stats for cpu / gpu."""
async def run_tasks():
await asyncio.wait(
asyncio.create_task(set_gpu_stats(config, stats)),
asyncio.create_task(set_cpu_stats(stats)),
)
loop = asyncio.get_event_loop()
loop.run_until_complete(run_tasks())
loop.close()
def set_cpu_stats(all_stats: dict[str, str]) -> None: def set_cpu_stats(all_stats: dict[str, str]) -> None:
"""Set cpu usage from top.""" """Set cpu usage from top."""
cpu_stats = get_cpu_stats() cpu_stats = get_cpu_stats()
@ -188,10 +202,7 @@ def stats_snapshot(
} }
stats["detection_fps"] = round(total_detection_fps, 2) stats["detection_fps"] = round(total_detection_fps, 2)
asyncio.gather( get_processing_stats(config, stats)
asyncio.create_task(set_gpu_stats(config, stats)),
asyncio.create_task(set_cpu_stats(stats)),
)
stats["service"] = { stats["service"] = {
"uptime": (int(time.time()) - stats_tracking["started"]), "uptime": (int(time.time()) - stats_tracking["started"]),