Skip to content

Commit 66499d9

Browse files
committed
feat: implement database pool
1 parent e06b603 commit 66499d9

3 files changed

Lines changed: 20 additions & 1 deletion

File tree

.env.example

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@ APP_NAME=Todo Modulith API
22
APP_ENV=production
33

44
DATABASE_URL=postgresql+asyncpg://postgres:postgres@127.0.0.1:5432/todo_db
5+
DATABASE_POOL_SIZE=20
6+
DATABASE_MAX_OVERFLOW=10
7+
DATABASE_POOL_TIMEOUT=30
8+
DATABASE_POOL_RECYCLE=3600
59

610
REDIS_URL=redis://:password@127.0.0.1:6379/0
711

src/core/config/setting.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,10 @@ class Settings(BaseSettings):
5353
MAX_REQUEST_SIZE_MB: int = Field(
5454
alias="MAX_REQUEST_SIZE_MB", default=5 * 1024 * 1024
5555
)
56+
DATABASE_POOL_SIZE: int = Field(alias="DATABASE_POOL_SIZE", default=20)
57+
DATABASE_MAX_OVERFLOW: int = Field(alias="DATABASE_MAX_OVERFLOW", default=10)
58+
DATABASE_POOL_TIMEOUT: int = Field(alias="DATABASE_POOL_TIMEOUT", default=30)
59+
DATABASE_POOL_RECYCLE: int = Field(alias="DATABASE_POOL_RECYCLE", default=3600)
5660

5761
@property
5862
def is_production(self) -> bool:

src/core/database/postgres/session.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,25 @@
22

33
from fastapi import Depends
44
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
5+
from sqlalchemy.pool import QueuePool
56

67
from src.core.config.setting import get_settings
78
from src.shared.database.unit_of_work import SQLAlchemyUnitOfWork
89
from src.shared.unit_of_work import UnitOfWork
910

1011
settings = get_settings()
1112

12-
engine = create_async_engine(settings.DATABASE_URL, echo=False, future=True)
13+
engine = create_async_engine(
14+
settings.DATABASE_URL,
15+
echo=False,
16+
future=True,
17+
poolclass=QueuePool,
18+
pool_size=settings.DATABASE_POOL_SIZE,
19+
max_overflow=settings.DATABASE_MAX_OVERFLOW,
20+
pool_pre_ping=True,
21+
pool_timeout=settings.DATABASE_POOL_TIMEOUT,
22+
pool_recycle=settings.DATABASE_POOL_RECYCLE,
23+
)
1324
AsyncSessionLocal = async_sessionmaker(
1425
engine, class_=AsyncSession, expire_on_commit=False, autoflush=False
1526
)

0 commit comments

Comments
 (0)