feat: added db connection retries

Author: An0nX

docs/getting-started/installation.md

@@ -95,6 +95,8 @@ volumes:
     # --- Application Settings (можно оставить по умолчанию) ---
     LOG_LEVEL=INFO
     GC_INTERVAL_SECONDS=60
+    DB_CONN_RETRIES=5
+    DB_CONN_RETRY_DELAY=5
     ```
 
 #### Шаг 4: Запустите DeBot

docs/system-architecture/configuration.md

@@ -20,6 +20,8 @@
 | `DB_NAME` | Имя базы данных. | `userbot_db` |
 | `DB_USER` | Имя пользователя для подключения к БД. | `userbot` |
 | `DB_PASS` | Пароль пользователя для подключения к БД. | `userbot_password` |
+| `DB_CONN_RETRIES` | Количество попыток подключения к БД при старте. | `5` |
+| `DB_CONN_RETRY_DELAY`| Задержка в секундах между попытками подключения. | `5` |
 
 ### Приложение
 

userbot/src/config.py

@@ -16,13 +16,17 @@
 DB_USER: str = os.getenv("DB_USER", "userbot")
 DB_PASS: str = os.getenv("DB_PASS", "userbot_password")
 DB_NAME: str = os.getenv("DB_NAME", "userbot_db")
+# New: Connection retry settings
+DB_CONN_RETRIES: int = int(os.getenv("DB_CONN_RETRIES", 5))
+DB_CONN_RETRY_DELAY: int = int(os.getenv("DB_CONN_RETRY_DELAY", 5))
+
 
 # --- Application Settings ---
 MODULE_FOLDER: str = "userbot.modules"
 LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO").upper()
 
 # --- Scheduler Settings ---
-GC_INTERVAL_SECONDS: int = int(os.getenv("GC_INTERVAL_SECONDS", 30))
+GC_INTERVAL_SECONDS: int = int(os.getenv("GC_INTERVAL_SECONDS", 60))
 AUTO_UPDATE_ENABLED: bool = os.getenv("AUTO_UPDATE_ENABLED", "False").lower() in ('true', '1', 't')
 AUTO_UPDATE_INTERVAL_MINUTES: int = int(os.getenv("AUTO_UPDATE_INTERVAL_MINUTES", 1440))
 DEPLOY_TYPE: str = os.getenv("DEPLOY_TYPE", "source") # 'source' or 'image'

userbot/src/db/session.py

@@ -1,11 +1,16 @@
 import logging
+import asyncio
+import sys
 from contextlib import asynccontextmanager
 from typing import AsyncGenerator
 
 from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession
 from sqlalchemy.exc import SQLAlchemyError
 
-from userbot.src.config import DB_TYPE, DB_HOST, DB_PORT, DB_USER, DB_PASS, DB_NAME
+from userbot.src.config import (
+    DB_TYPE, DB_HOST, DB_PORT, DB_USER, DB_PASS, DB_NAME,
+    DB_CONN_RETRIES, DB_CONN_RETRY_DELAY
+)
 from userbot.src.db.models import Base
 
 logger: logging.Logger = logging.getLogger(__name__)
@@ -31,11 +36,33 @@
 
 async def initialize_database() -> None:
     """
-    Creates all tables in the database based on the SQLAlchemy models.
+    Connects to the database and creates all tables based on the SQLAlchemy models.
+    Includes a retry mechanism to handle database startup delays.
     """
-    async with async_engine.begin() as conn:
-        await conn.run_sync(Base.metadata.create_all)
-    logger.info("Database schema initialization check complete.")
+    for attempt in range(DB_CONN_RETRIES):
+        try:
+            async with async_engine.begin() as conn:
+                await conn.run_sync(Base.metadata.create_all)
+            logger.info("Database schema initialization check complete.")
+            return  # Success, exit the function
+        except (ConnectionRefusedError, OSError) as e:
+            if attempt < DB_CONN_RETRIES - 1:
+                logger.warning(
+                    f"Database connection failed (attempt {attempt + 1}/{DB_CONN_RETRIES}): {e}. "
+                    f"Retrying in {DB_CONN_RETRY_DELAY} seconds..."
+                )
+                await asyncio.sleep(DB_CONN_RETRY_DELAY)
+            else:
+                logger.critical(
+                    f"Could not connect to the database after {DB_CONN_RETRIES} attempts. "
+                    "Please ensure the database is running and the .env file is configured correctly."
+                )
+                raise  # Re-raise the final exception to stop the application
+    
+    # This part should not be reachable if all retries fail, but as a safeguard:
+    logger.critical("Exhausted all retries to connect to the database. Exiting.")
+    sys.exit(1)
+
 
 @asynccontextmanager
 async def get_db() -> AsyncGenerator[AsyncSession, None]: