Redis SETNX 实现可重入锁：原理、代码示例及注意事项

SETNX 命令是 Redis 中的一个原子操作，用于设置一个键值对，但只有在键不存在时才会设置成功。可重入锁是一种特殊的锁机制，允许同一个线程多次获取同一个锁而不会造成死锁。

要实现可重入锁，可以使用 Redis 的字符串数据结构和 Lua 脚本进行操作。以下是一个简单的示例代码：

import redis

class ReentrantLock:
    def __init__(self, redis_conn, lock_key):
        self.redis_conn = redis_conn
        self.lock_key = lock_key

    def acquire(self):
        # 尝试获取锁
        acquired = self.redis_conn.setnx(self.lock_key, 1)
        if acquired:
            # 锁获取成功，设置锁的拥有者为当前线程
            self.redis_conn.hset(self.lock_key, 'owner', 'thread_id')
            return True

        # 锁已被其他线程持有，检查是否为当前线程持有
        owner = self.redis_conn.hget(self.lock_key, 'owner')
        if owner == 'thread_id':
            # 当前线程已经持有锁，增加锁的重入次数
            self.redis_conn.hincrby(self.lock_key, 'reentrant_count', 1)
            return True

        return False

    def release(self):
        # 检查是否为当前线程持有锁
        owner = self.redis_conn.hget(self.lock_key, 'owner')
        if owner != 'thread_id':
            raise RuntimeError('Cannot release a lock that is not held by the current thread')

        # 减少锁的重入次数
        reentrant_count = self.redis_conn.hincrby(self.lock_key, 'reentrant_count', -1)
        if reentrant_count > 0:
            return

        # 重入次数为 0，释放锁
        self.redis_conn.delete(self.lock_key)

在上述代码中，acquire 方法用于获取锁，首先尝试使用 setnx 命令设置锁的键值对，如果设置成功则表示获取锁成功；如果锁已被其他线程持有，则检查是否为当前线程持有，如果是则增加锁的重入次数。

release 方法用于释放锁，首先检查是否为当前线程持有锁，如果不是则抛出异常。如果锁的重入次数大于 0，则减少锁的重入次数；如果重入次数为 0，则删除锁的键值对。

需要注意的是，上述代码只是一个简单的示例，实际应用中可能需要考虑更多的情况，比如超时时间、锁的可重入深度等。