使用django_cache怎么实现分布式锁
这篇文章给大家介绍使用django_cache怎么实现分布式锁,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
第一种锁命令INCR
这种加锁的思路是, key 不存在,那么 key 的值会先被初始化为 0 ,然后再执行 INCR 操作进行加一。 然后其它用户在执行 INCR 操作进行加一时,如果返回的数大于 1 ,说明这个锁正在被使用当中。
第二种锁命令SETNX
这种加锁的思路是,如果 key 不存在,将 key 设置为 value 如果 key 已存在,则 SETNX 不做任何动作
第三种锁命令SET
上面两种方法都有一个问题,会发现,都需要设置 key 过期。那么为什么要设置key过期呢?如果请求执行因为某些原因意外退出了,导致创建了锁但是没有删除锁,那么这个锁将一直存在,以至于以后缓存再也得不到更新。于是乎我们需要给锁加一个过期时间以防不测。
在实际编写中,我综合了第二种和第三种方式,即用键名来设置锁,同时设置了过期时间,以防长时间占用。
另外,关于如何使用django-cache去使用数据库缓存,相关的API整理如下:
from django.core.cache import caches # 设置锁和超时时间 cache.set('my_key', 'Initial value', 60) # 获取锁 cache.get('my_key') # 更新锁 cache.add('add_key', 'New value')
代码编写
在经过多次的迭代,并且对比了网上的各路写法后,我结合django-cache的特性,最终总结了一套较为简洁的写法。
首先是一个CacheLock的类,初始化方法里可以传执行超时时间,和拿锁等待的时间。CacheLock类的主要方法有两个,一个是拿锁的方法,一个是释放锁的方法。
拿锁的方法中,键名根据操作的具体对象来定,键值为uuid值,超时时间默认为60s。一旦发现能拿到锁,则返回uuid值。
释放锁的方法中,首先比较键值和uuid值是否一致,一致则释放,避免因超时情况导致把其他的正在操作的锁给释放掉。
class CacheLock(object): def __init__(self, expires=60, wait_timeout=0): self.cache = cache self.expires = expires # 函数执行超时时间 self.wait_timeout = wait_timeout # 拿锁等待超时时间 def get_lock(self, lock_key): # 获取cache锁 wait_timeout = self.wait_timeout identifier = uuid.uuid4() while wait_timeout >= 0: if self.cache.add(lock_key, identifier, self.expires): return identifier wait_timeout -= 1 time.sleep(1) raise LockTimeout({'msg': '当前有其他用户正在编辑该采集配置,请稍后重试'}) def release_lock(self, lock_key, identifier): # 释放cache锁 lock_value = self.cache.get(lock_key) if lock_value == identifier: self.cache.delete(lock_key)
另外,将缓存锁写成一个装饰器,对需要加锁的地方,添加上该装饰器,则可以很轻松地实现锁功能。
def lock(cache_lock): def my_decorator(func): def wrapper(*args, **kwargs): lock_key = 'bk_monitor:lock:xxx' # 具体的lock_key要根据调用时传的参数而定 identifier = cache_lock.get_lock(lock_key) try: return func(*args, **kwargs) finally: cache_lock.release_lock(lock_key, identifier) return wrapper return my_decorator
再举一个实际调用中的例子:
@lock(CacheLock()) def f(): pass
关于使用django_cache怎么实现分布式锁就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。