From 5bb83fe9049e263ec8a46d07a1b242c32a68c513 Mon Sep 17 00:00:00 2001 From: Vladless Date: Sun, 1 Dec 2024 23:40:37 +0300 Subject: [PATCH] servers in database --- handlers/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/handlers/utils.py b/handlers/utils.py index e3ca2857..95871cf6 100644 --- a/handlers/utils.py +++ b/handlers/utils.py @@ -5,7 +5,8 @@ from typing import Optional import asyncpg from bot import bot -from config import CLUSTERS, DATABASE_URL +from config import DATABASE_URL +from database import get_servers_from_db from logger import logger @@ -42,12 +43,13 @@ async def get_least_loaded_cluster() -> str: Returns: str: Идентификатор наименее загруженного кластера. """ - cluster_loads: dict[str, int] = {cluster_id: 0 for cluster_id in CLUSTERS.keys()} + servers = await get_servers_from_db() + + cluster_loads: dict[str, int] = {cluster_id: 0 for cluster_id in servers.keys()} async with asyncpg.create_pool(DATABASE_URL) as pool: async with pool.acquire() as conn: keys = await conn.fetch("SELECT server_id FROM keys") - for key in keys: cluster_id = key["server_id"] if cluster_id in cluster_loads: