Close old django connections before task body.

Replaces /refs #633
coleifer · Nov 12, 2021 · e77acf3 · e77acf3
1 parent 3ce5e6d
commit e77acf3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/huey/contrib/djhuey/__init__.py b/huey/contrib/djhuey/__init__.py
@@ -129,6 +129,8 @@ def close_db(fn):
     """Decorator to be used with tasks that may operate on the database."""
     @wraps(fn)
     def inner(*args, **kwargs):
+        if not HUEY.immediate:
+            close_old_connections()
         try:
             return fn(*args, **kwargs)
         finally: