Django分布式锁

启蒙狮

已于 2025-07-03 15:36:17 修改

阅读量835

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Django 文章标签： django 分布式锁

于 2022-09-09 14:12:09 首次发布

本文链接：https://blog.csdn.net/llf_cloud/article/details/126781944

Django 专栏收录该内容

14 篇文章

订阅专栏

该博客介绍了Django中实现缓存锁的类`DjangoCacheLock`，用于在高并发场景下确保资源的安全访问。`acquire_lock`方法通过循环尝试获取锁，如果在`wait_timeout`时间内未获取到，则返回失败。`release_lock`方法则根据标识符释放已获取的锁，确保资源的正确释放。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import time
import uuid

from django.core.cache import cache


class DjangoCacheLock(object):
    def __init__(self, lock_key: str, expires: int = 1 * 60 * 60, wait_timeout: int = 10, retry_interval: float = 0.1):
        self.cache = cache
        self.expires = expires  # 函数执行超时时间
        self.wait_timeout = wait_timeout  # 拿锁等待超时时间
        self.retry_interval = retry_interval
        self.lock_key = lock_key
        self.identifier = f"{uuid.uuid4()}"

    def acquire_lock(self) -> bool:
        # 获取cache锁
        if self.cache.get(self.lock_key) == self.identifier:
       		return True
        while self.wait_timeout >= 0:
            if self.cache.add(self.lock_key, self.identifier, self.expires):
                return True
            time.sleep(self.retry_interval)
            self.wait_timeout -= self.retry_interval
        return False

    def release_lock(self) -> bool:
        # 释放cache锁
        lock_value = self.cache.get(self.lock_key)
        if lock_value == self.identifier:
            self.cache.delete(self.lock_key)
        return True