Thunder/utils/keepalive.py

# Thunder/utils/keepalive.py

import asyncio
import aiohttp
from Thunder.vars import Var
from Thunder.utils.logger import logger

async def ping_server():
    try:
        async with aiohttp.ClientSession(
            timeout=aiohttp.ClientTimeout(total=10)
        ) as session:
            while True:
                try:
                    await asyncio.sleep(Var.PING_INTERVAL)
                    async with session.get(Var.URL) as resp:
                        if resp.status != 200:
                            logger.warning(f"Ping to {Var.URL} returned status {resp.status}.")
                except asyncio.CancelledError:
                    break
    except Exception as e:
        logger.error(f"Error in ping_server: {e}", exc_info=True)