From acee709aab2b9fb6b83ac6a4da82ae1c0aeb03c5 Mon Sep 17 00:00:00 2001 From: Adam Sachs Date: Wed, 11 Dec 2024 09:23:49 -0500 Subject: [PATCH] revert disabling pooling on celery worker tasks (i.e., re-enable pooling) --- src/fides/api/tasks/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/fides/api/tasks/__init__.py b/src/fides/api/tasks/__init__.py index e757052ac1..ed5d9b8556 100644 --- a/src/fides/api/tasks/__init__.py +++ b/src/fides/api/tasks/__init__.py @@ -64,10 +64,11 @@ def get_new_session(self) -> ContextManager[Session]: if self._task_engine is None: self._task_engine = get_db_engine( config=CONFIG, + pool_size=CONFIG.database.task_engine_pool_size, + max_overflow=CONFIG.database.task_engine_max_overflow, keepalives_idle=CONFIG.database.task_engine_keepalives_idle, keepalives_interval=CONFIG.database.task_engine_keepalives_interval, keepalives_count=CONFIG.database.task_engine_keepalives_count, - disable_pooling=True, ) # same for the sessionmaker