分布式锁实现方式

import redis
import time
import uuid

class RedisDistributedLock:
    def __init__(self, redis_client, lock_key, expire_time=10):
        self.redis_client = redis_client
        self.lock_key = lock_key
        self.expire_time = expire_time
        self.identifier = str(uuid.uuid4())

    def acquire(self):
        while True:
            if self.redis_client.set(self.lock_key, self.identifier, nx=True, px=self.expire_time):
                return True
            time.sleep(0.01)

    def release(self):
        with self.redis_client.pipeline() as pipe:
            while True:
                try:
                    pipe.watch(self.lock_key)
                    if pipe.get(self.lock_key) == self.identifier:
                        pipe.delete(self.lock_key)
                        return True
                    pipe.unwatch()
                    break
                except redis.WatchError:
                    continue
        return False

# 使用示例
redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
lock = RedisDistributedLock(redis_client, 'my_lock_key')

if lock.acquire():
    try:
        # 执行业务逻辑
        print("Lock acquired, performing business logic...")
    finally:
        lock.release()
        print("Lock released.")