diff --git a/app/tasks/record_usages.py b/app/tasks/record_usages.py index cdecf01..c9380e1 100644 --- a/app/tasks/record_usages.py +++ b/app/tasks/record_usages.py @@ -2,7 +2,6 @@ import asyncio from collections import defaultdict from datetime import datetime -from grpclib import GRPCError from sqlalchemy import and_, select, insert, update, bindparam from app import marznode @@ -115,7 +114,7 @@ async def get_users_stats( ) -> tuple[int, list[dict]]: try: params = list() - for stat in await node.fetch_users_stats(): + for stat in await asyncio.wait_for(node.fetch_users_stats(), 10): if stat.usage: params.append({"uid": stat.uid, "value": stat.usage}) return node_id, params