OpenConceptLab/ocl_issues#1777 | added task post run signal to close …

…db connections

core/settings.py

@@ -99,7 +99,7 @@
     'core.importers',
     'core.pins',
     'core.client_configs',
-    'core.tasks',
+    'core.tasks.apps.TaskConfig',
     'core.toggles',
     'core.repos',
     'core.url_registry',

core/tasks/apps.py

@@ -0,0 +1,9 @@
+from django.apps import AppConfig
+
+
+class TaskConfig(AppConfig):
+    name = 'core.tasks'
+    verbose_name = "Task"
+
+    def ready(self):
+        from core.tasks import signals  # pylint: disable=unused-variable, unused-import

core/tasks/signals.py

@@ -0,0 +1,15 @@
+from celery.signals import task_postrun
+from django import db
+
+
+@task_postrun.connect
+def on_task_done(sender=None, headers=None, body=None, **kwargs):
+    for conn in db.connections.all():
+        try:
+            conn.close_if_unusable_or_obsolete()
+        except db.utils.InterfaceError:
+            pass
+        except db.DatabaseError as exc:
+            str_exc = str(exc)
+            if 'closed' not in str_exc and 'not connected' not in str_exc:
+                raise exc