fix: improved response time async fix

src/core/database.py

@@ -1,38 +1,31 @@
 from typing import AsyncGenerator
-
-from dotenv import load_dotenv
+from sqlmodel import SQLModel
 from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
-from sqlmodel import SQLModel, create_engine
 from sqlmodel.ext.asyncio.session import AsyncSession
-
-from src.core import *
 from src.core.config import settings
 
-load_dotenv()
-
-engine = create_engine(
-    settings.DATABASE_URL, echo=True
-)  # to false on prod just to chcek for now
 
 async_engine = create_async_engine(
-    url=settings.ASYNC_DATABASE_URL, future=True, connect_args={"ssl": True}
+    settings.ASYNC_DATABASE_URL,
+    echo=False,
+    future=True,
+    pool_size=20,
+    max_overflow=40,
+    pool_timeout=30,
+    connect_args={"ssl": True},
 )
 
 async_session = async_sessionmaker(
-    class_=AsyncSession, bind=async_engine, expire_on_commit=False
+    async_engine,
+    class_=AsyncSession,
+    expire_on_commit=False,
+    autoflush=False
 )
 
-
-def init_db():
-    SQLModel.metadata.create_all(engine)
-
+async def init_db():
+    async with async_engine.begin() as conn:
+        await conn.run_sync(SQLModel.metadata.create_all)
 
 async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
     async with async_session() as session:
         yield session
-
-
-if __name__ == "__main__":
-    print("Table creating")
-    init_db()
-    print("Table Created successfully!")

src/main.py

@@ -12,7 +12,10 @@ from fastapi.staticfiles import StaticFiles
 
 app = FastAPI(title="Yuvabe App API")
 
-
+@app.on_event("startup")
+async def on_startup():
+    await init_db()
+    
 app.include_router(home_router, prefix="/home", tags=["Home"])
 
 # init_db()