分布式概念：分布式锁实现代码

独木人生

于 2024-02-29 11:11:30 发布

阅读量343

点赞数 5

分类专栏：后端文章标签：分布式

本文链接：https://blog.csdn.net/xukris/article/details/136365784

版权

后端专栏收录该内容

14 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Redis作为后端实现一个简单的分布式锁，包括DistributedLock类的构造函数、acquire方法获取锁和release方法释放锁的代码示例。

摘要由CSDN通过智能技术生成

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

import redis

class DistributedLock:
    def __init__(self, redis_client, lock_name, expire_time=10):
        self.redis_client = redis_client
        self.lock_name = lock_name
        self.expire_time = expire_time

    def acquire(self):
        while True:
            result = self.redis_client.set(self.lock_name, "locked", nx=True, ex=self.expire_time)
            if result:
                return True

    def release(self):
        self.redis_client.delete(self.lock_name)

使用示例：

redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
lock = DistributedLock(redis_client, 'mylock')

# 尝试获取锁
if lock.acquire():
    try:
        # 执行加锁操作后的代码
        print("Lock acquired!")
    finally:
        # 释放锁
        lock.release()