From effbde407a8e8d4786ac73cbb52a2cc5825907ca Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Sat, 4 Feb 2023 18:28:17 +0100 Subject: [PATCH] increase db pool size --- backend/oasst_backend/config.py | 3 +++ backend/oasst_backend/database.py | 8 +++++++- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/backend/oasst_backend/config.py b/backend/oasst_backend/config.py index 1d260764..683133f7 100644 --- a/backend/oasst_backend/config.py +++ b/backend/oasst_backend/config.py @@ -166,6 +166,9 @@ class Settings(BaseSettings): DATABASE_URI: Optional[PostgresDsn] = None DATABASE_MAX_TX_RETRY_COUNT: int = 3 + DATABASE_POOL_SIZE = 128 + DATABASE_MAX_OVERFLOW = 128 + RATE_LIMIT: bool = True MESSAGE_SIZE_LIMIT: int = 2000 REDIS_HOST: str = "localhost" diff --git a/backend/oasst_backend/database.py b/backend/oasst_backend/database.py index 1d0e19b2..a77ca1f5 100644 --- a/backend/oasst_backend/database.py +++ b/backend/oasst_backend/database.py @@ -5,4 +5,10 @@ from sqlmodel import create_engine if settings.DATABASE_URI is None: raise OasstError("DATABASE_URI is not set", error_code=OasstErrorCode.DATABASE_URI_NOT_SET) -engine = create_engine(settings.DATABASE_URI, echo=settings.DEBUG_DATABASE_ECHO, isolation_level="REPEATABLE READ") +engine = create_engine( + settings.DATABASE_URI, + echo=settings.DEBUG_DATABASE_ECHO, + isolation_level="REPEATABLE READ", + pool_size=settings.DATABASE_POOL_SIZE, + max_overflow=settings.DATABASE_MAX_OVERFLOW, +)