blob: 8b7d6ebb631881dea0e8638fe0859c8422f6e025 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
import time
from fastapi import FastAPI
from sqlalchemy import text
from app.routes.routes import router as api_router
from app.utils.create_tables import init_db
from app.utils.db import engine
from app.utils.logger_cfg import logger
app_start_time = time.perf_counter()
logger.debug("App start timestamp recorded")
app = FastAPI()
logger.info("FastAPI application instance created")
logger.debug("FastAPI instance created successfully")
app.include_router(api_router, prefix="/api")
logger.info("API router registered with prefix /api")
logger.debug("Router object: %s", api_router)
@app.on_event("startup")
async def startup():
logger.info("Application startup initiated")
logger.debug("Starting database connection attempt")
try:
async with engine.begin() as conn:
logger.debug("Executing test query: SELECT 1")
result = await conn.execute(text("SELECT 1"))
db_status = result.scalar()
logger.info(f"Database connection successful: {db_status}")
logger.debug("Database test query executed successfully")
logger.debug("Starting database table initialization")
await init_db()
logger.info("Database initialization completed")
logger.debug("Database tables initialized successfully")
except Exception as e:
logger.exception(f"Database startup failed: {e}")
elapsed = time.perf_counter() - app_start_time
logger.success(
f"Application startup completed in {elapsed*1000:.2f} ms ({elapsed:.6f} s)"
)
logger.debug("Startup complete timestamp recorded")
@app.on_event("shutdown")
async def shutdown():
logger.info("Application shutdown initiated")
logger.debug("Starting engine disposal")
try:
await engine.dispose()
logger.info("Database engine disposed successfully")
logger.debug("Engine disposal finished")
except Exception as e:
logger.exception(f"Error during shutdown: {e}")
@app.get("/")
async def read_root():
logger.info("Root endpoint accessed")
logger.debug("Processing root endpoint request")
return {"message": "Hello new asyncpg ci!"}
@app.get("/check-db")
async def check_db():
logger.info("Database health check endpoint called")
logger.debug("Starting database health check")
try:
async with engine.begin() as conn:
logger.debug("Executing test query: SELECT 1")
result = await conn.execute(text("SELECT 1"))
db_result = result.scalar()
logger.info(f"Database health check successful: {db_result}")
logger.debug("Database health check query executed successfully")
return {"db_check": db_result}
except Exception as e:
logger.exception(f"Database health check failed: {e}")
raise
|
