Skip to content

Commit 83b4077

Browse files
committed
Remove close_db_connection_before_task
1 parent 4714bc4 commit 83b4077

1 file changed

Lines changed: 2 additions & 9 deletions

File tree

beagle_etl/celery.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -108,17 +108,10 @@ def setup_task_logger(logger, *args, **kwargs):
108108
}
109109

110110

111-
# Close database connections before and after each task to prevent stale connections
111+
# Close database connections after each task to prevent stale connections
112112
# Only do this in Celery workers, not during tests or regular Django operations
113-
@task_prerun.connect
114-
def close_db_connection_before_task(**kwargs):
115-
"""Close database connection before task starts"""
116-
if not settings.TESTING:
117-
connection.close()
118-
119-
120113
@task_postrun.connect
121114
def close_db_connection_after_task(**kwargs):
122-
"""Close database connection after task completes"""
115+
"""Close database connection after task completes to prevent connection leaks"""
123116
if not settings.TESTING:
124117
connection.close()

0 commit comments

Comments
 (0)