Fix health check endpoint due to httpx session

This commit is contained in:
mhdzumair
2024-12-15 10:38:24 +05:30
parent e5e24f03b4
commit 7d44695a08
+7 -7
View File
@@ -141,14 +141,14 @@ async def get_home(request: Request):
@wrappers.exclude_rate_limit
async def health():
start_time = asyncio.get_event_loop().time()
async with httpx.AsyncClient(proxy=settings.requests_proxy_url) as session:
async with httpx.AsyncClient(proxy=settings.requests_proxy_url) as client:
try:
async with session.head("https://www.google.com", timeout=10) as response:
return {
"status": "healthy",
"status_code": response.status,
"time": asyncio.get_event_loop().time() - start_time,
}
response = await client.head("https://www.google.com", timeout=10)
return {
"status": "healthy",
"status_code": response.status_code,
"time": asyncio.get_event_loop().time() - start_time,
}
except Exception as e:
logging.error("Health check failed: %s", e)
raise HTTPException(status_code=503, detail="Health check failed.")