Redis cluster lock (#238)

bbangert · Apr 9, 2024 · 9ad1806 · 9ad1806
1 parent 0dfc877
commit 9ad1806
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/beaker/ext/redisclusternm.py b/beaker/ext/redisclusternm.py
@@ -75,6 +75,13 @@ class RedisClusterSynchronizer(RedisSynchronizer):
 
     This Synchronizer only supports 1 reader or 1 writer at time, not concurrent readers.
     """
+    RELEASE_LOCK_LUA = """
+        if redis.call('get', KEYS[1]) == ARGV[1] then
+            return redis.call('del', KEYS[1])
+        else
+            return 0
+        end
+    """
 
     def __init__(self, identifier, urls, nodes=None, **kwargs):
         super(RedisSynchronizer, self).__init__()
@@ -85,3 +92,9 @@ def __init__(self, identifier, urls, nodes=None, **kwargs):
             )
         else:
             self.client = urls
+        self._release_lock = self.client.register_script(self.RELEASE_LOCK_LUA)
+
+    def do_release_write_lock(self):
+        identifier = self.identifier
+        owner_id = self._get_owner_id()
+        self._release_lock(keys=[identifier], args=[owner_id])