在 Redis 中实现分布式锁通常可以通过设置一个带有过期时间的键来完成。下面是实现分布式锁的一种常用方法：
1. 锁的获取
在 Redis 中，我们可以使用 SETNX 命令来尝试设置一个锁。如果返回值是 1，则表示成功获取到锁；如果返回值是 0，则表示锁已经被其他客户端持有。
import redis
import time
import uuid
class RedisLock:
def __init__(self, redis_client, lock_name, expire=10):
self.redis_client = redis_client
self.lock_name = lock_name
self.expire = expire
self.identifier = str(uuid.uuid4()) # 唯一标识符，防止误释放锁
def acquire(self):
# 尝试获取锁，如果获取成功则返回 True，否则返回 False
if self.redis_client.setnx(self.lock_name, self.identifier):
# 设置过期时间，避免死锁情况
self.redis_client.expire(self.lock_name, self.expire)
return True
return False
def release(self):
# 确保只有持有该锁的客户端才能释放它
script = """
if redis.call("get", KEYS[1]) == ARGV[1] then
return redis.call("del", KEYS[1])
else
return 0
end
"""
# 执行 Lua 脚本进行安全释放
result = self.redis_client.eval(script, 1, self.lock_name, self.identifier)
return result != 0
# 使用示例:
if __name__ == "__main__":
r = redis.StrictRedis(host='localhost', port=6379, db=0)
lock = RedisLock(r, "my_lock")
if lock.acquire():
try:
print("Lock acquired!")
# 在此处执行临界区代码
time.sleep(5) # 模拟长时间操作
finally:
lock.release()
print("Lock released!")
else:
print("Could not acquire the lock.")
2. 注意事项
- 唯一标识符：使用唯一标识符（如 UUID）来确保只有持有者能够释放该锁。
- 过期时间：为防止因为程序崩溃或异常而造成死锁，需要给锁设置一个过期时间。
- Lua 脚本：在释放锁时，使用 Lua 脚本来保证原子性，以确保只有拥有该锁的客户端才能释放它。
3. 高级功能（可选）
- 重入支持：根据需要可以扩展支持重入，即同一线程可以多次获得同一把锁。
- 加大超时时间：在长时间操作中，可以设计机制续租以延长超时时间。
这种方式简单易用，是实现分布式系统中对资源访问控制的一种有效方案。
内容由零声教学AI助手提供，问题来源于学员提问
