Add debug logging to the lock manager.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
2017-07-01 02:31:17 -04:00
parent 916c3497c4
commit c42a2c8a94
3 changed files with 27 additions and 3 deletions
@@ -8,10 +8,23 @@ from .base import LockingBackend
 class ModelLock(LockingBackend):
    @classmethod
    def acquire_lock(cls, name, timeout=None):
+        super(ModelLock, cls).acquire_lock(name=name, timeout=timeout)
        Lock = apps.get_model(app_label='lock_manager', model_name='Lock')
-        return Lock.objects.acquire_lock(name=name, timeout=timeout)
+        return ModelLock(
+            model_instance = Lock.objects.acquire_lock(
+                name=name, timeout=timeout
+            )
+        )

    @classmethod
    def purge_locks(cls):
+        super(ModelLock, cls).purge_locks()
        Lock = apps.get_model(app_label='lock_manager', model_name='Lock')
        Lock.objects.select_for_update().delete()
+
+    def __init__(self, model_instance):
+        self.model_instance = model_instance
+
+    def release(self):
+        super(ModelLock, self).release()
+        self.model_instance.release()