redis分布式锁防止重复提交
时间: 2023-09-24 08:05:20 浏览: 100
redis分布式锁
使用Redis的分布式锁可以有效地防止重复提交。下面是一个更完整的示例代码:
```python
import redis
import time
# 创建Redis连接
redis_client = redis.Redis(host='localhost', port=6379, db=0)
def acquire_lock(lock_name, acquire_timeout=10, lock_timeout=10):
# 生成唯一的锁标识符
identifier = str(time.time())
end_time = time.time() + acquire_timeout
while time.time() < end_time:
# 尝试获取锁
if redis_client.set(lock_name, identifier, ex=lock_timeout, nx=True):
return identifier
time.sleep(0.001) # 等待一段时间后重试
return None
def release_lock(lock_name, identifier):
# 使用Lua脚本确保原子性操作
release_lock_script = """
if redis.call("get", KEYS[1]) == ARGV[1] then
return redis.call("del", KEYS[1])
else
return 0
end
"""
redis_client.eval(release_lock_script, 1, lock_name, identifier)
# 使用锁来执行防止重复提交的逻辑
def process_submission(submission_id):
lock_name = f"lock:submission:{submission_id}"
# 尝试获取锁
identifier = acquire_lock(lock_name)
if identifier:
try:
# 执行提交逻辑
print(f"Processing submission ID {submission_id}")
time.sleep(5) # 模拟处理过程
print(f"Submission ID {submission_id} processed successfully")
finally:
# 释放锁
release_lock(lock_name, identifier)
else:
print(f"Failed to acquire lock for submission ID {submission_id}")
```
在这个示例中,`acquire_lock`函数尝试获取一个带有指定名称的锁。它使用`set`命令来设置一个带有超时时间的键,只有在键不存在的情况下才会成功。`set`命令的`nx`选项确保只有一个线程能够成功获取锁。如果获取锁成功,函数将返回一个唯一的标识符,如果超时仍未获取到锁,则返回None。
`release_
阅读全文