如何利用setnx函数实现分布式锁

利用setnx函数实现分布式锁的基本思路是：

1. 使用setnx命令尝试在redis中设置一个指定的key，如果该key不存在则返回1，表示成功获取锁；如果该key已经存在则返回0，表示获取锁失败。

2. 在获取锁成功后，设置一个过期时间，确保锁在一定时间内自动释放，避免锁被长时间占用造成死锁。

3. 在释放锁时，使用del命令删除该key，确保锁释放成功。

以下是一个简单的示例代码来实现分布式锁：

import redis
import time

# 连接redis
redis_client = redis.Redis(host='localhost', port=6379, db=0)

# 获取分布式锁
def acquire_lock(lock_key, expire_time=10):
    lock = redis_client.setnx(lock_key, time.time() + expire_time)
    if lock:
        return True
    current_time = time.time()
    # 判断锁是否过期
    if current_time > float(redis_client.get(lock_key) or 0):
        # 如果锁已经过期，则重新设置锁
        new_expire_time = current_time + expire_time
        old_expire_time = float(redis_client.getset(lock_key, new_expire_time) or 0)
        if old_expire_time == current_time:
            return True
    return False

# 释放分布式锁
def release_lock(lock_key):
    current_time = time.time()
    if current_time < float(redis_client.get(lock_key) or 0):
        redis_client.delete(lock_key)
        return True
    return False

# 测试分布式锁
lock_key = "distributed_lock"
if acquire_lock(lock_key):
    try:
        # 业务逻辑处理
        print("获取锁成功，执行业务逻辑")
    finally:
        release_lock(lock_key)
else:
    print("获取锁失败")

在上面的示例中，首先通过acquire_lock函数尝试获取分布式锁，成功获取锁后执行业务逻辑，最后通过release_lock函数释放锁。通过这种方式可以确保在分布式环境下对资源的互斥访问。