Merge pull request #10 from Volumental/close_all_connections

close all connection before loading tasks
Volumental · Oct 3, 2018 · defc109 · defc109
2 parents 51a591a + 7150090
commit defc109
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/django_leek/server.py b/django_leek/server.py
@@ -5,6 +5,7 @@
 from .helpers import load_task
 from . import worker
 from . import helpers
+import django
 
 log = logging.getLogger(__name__)
 
@@ -41,6 +42,7 @@ def handle(self):
                 log.info('Got a task')
                 try:
                     task_id = int(data.decode())
+                    django.db.connections.close_all()
                     queued_task = load_task(task_id=task_id)
 
                     # Ensure pool got a worker processing it