Set redis expiration for lock to ensure removal.

mcdonc · Jan 10, 2012 · 40453cd · 40453cd
1 parent 651c911
commit 40453cd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/retools/lock.py b/retools/lock.py
@@ -55,6 +55,7 @@ def __enter__(self):
             if redis.setnx(self.key, expires):
                 # We gained the lock; enter critical section
                 self.start_time = time.time()
+                redis.expire(self.key, expires)
                 return
 
             current_value = redis.get(self.key)
@@ -63,6 +64,7 @@ def __enter__(self):
             if current_value and float(current_value) < time.time() and \
                redis.getset(self.key, expires) == current_value:
                 self.start_time = time.time()
+                redis.expire(self.key, expires)
                 return
 
             timeout -= 1