revert disabling pooling on celery worker tasks (i.e., re-enable pool…

…ing)
ethyca · Dec 11, 2024 · acee709 · acee709
1 parent 20c5133
commit acee709
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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