可重入redis 分布式锁-Go服务器开发

可重入Redis分布式锁是一种特殊的锁，它允许同一个线程在持有锁的情况下再次获得该锁，而不会造成死锁。实现可重入锁时，通常会为每个锁跟踪持有它的线程，并增加计数器以记录当前获取的次数。

以下是使用Python和redis-py库来实现可重入Redis分布式锁的示例：

实现思路

使用唯一标识：通过UUID或其他方式生成一个唯一标识符，用于区分不同的客户端/线程。
计数器：在获取锁时，如果已经拥有该锁，则仅增加计数器。
释放锁：当计数器减至零时，才真正释放该锁。

Python 示例代码

import redis
import uuid
import time

class ReentrantLock:
    def __init__(self, redis_client, lock_name, expire=10):
        self.redis = redis_client
        self.lock_name = lock_name
        self.expire = expire
        self.lock_value = str(uuid.uuid4())  # 唯一标识符
        self.counter = 0  # 初始化计数器

    def acquire_lock(self):
        current_value = self.redis.get(self.lock_name)

        if current_value is None:
            # 如果当前没有持有这个lock，可以直接获得并设置它
            if self.redis.set(self.lock_name, self.lock_value, nx=True, ex=self.expire):
                self.counter += 1  # 增加计数器
                return True
            
        elif current_value.decode('utf-8') == self.lock_value:
            # 如果当前值与自己的值相同，说明是可重入的情况
            self.counter += 1  # 增加计数器
            return True
        
        return False  # 未能获得锁

    def release_lock(self):
        if self.counter > 0:  
            current_value = self.redis.get(self.lock_name)
            
            if current_value and current_value.decode('utf-8') == self.lock_value:
                if self.counter == 1:
                    # 如果计数器到达0，删除lock并清空计数器
                    self.redis.delete(self.lock_name)
                else:
                    # 减少计数器，仅减少，不删除lock键
                    self.counter -= 1 

# 示例用法
if __name__ == "__main__":
    client = redis.StrictRedis(host='localhost', port=6379, db=0)

    lock = ReentrantLock(client, 'my_reentrant_lock')

    if lock.acquire_lock():
        try:
            print("Lock acquired! Performing critical section...")
            
            # 模拟嵌套调用来测试可重入性 
            if lock.acquire_lock():
                try:
                    print("Re-acquired the lock!")
                finally:
                    lock.release_lock()
            
        finally:
            lock.release_lock() 
            print("Lock released.")
    else:
        print("Could not acquire the lock.")