经验拾忆(纯手工)=> Python版-Redis分布式锁简单实现

系统 1362 0

定义代码如下

          
            import redis
import contextlib
import pickle
import os, socket, threading


class RedisLock:
    def __init__(self, lock_name, host='', port=6379, db=0):
        self.lock_name = lock_name
        self.redis = redis.Redis(connection_pool=redis.ConnectionPool(host=host, port=port, db=db))

    def acquire_lock(self, lock_id, expire=None):
        lock_id = lock_id if lock_id else self.get_lock_id()
        return True if self.redis.set(self.lock_name, pickle.dumps(lock_id), nx=True, ex=expire) else False

        # Above 1 line code can replace with follow codes to debug
        # if self.redis.set(self.lock_name, pickle.dumps(lock_id), nx=True, ex=expire):
        #     print('Lock Succeed')
        #     return True
        # else:
        #     print('Lock Failed')
        #     return False

    def release_lock(self, lock_id=None):
        lock_id = lock_id if lock_id else self.get_lock_id()
        if lock_id == pickle.loads(self.redis.get(self.lock_name)):
            self.redis.delete(self.lock_name)
            # print('Unlock Succeed')
            return True
        else:
            # print('Unlock Failed')
            return False

    @contextlib.contextmanager
    def lock(self, lock_id=None, expire=None):
        if not self.acquire_lock(lock_id, expire):
            exit(0)
        yield self
        self.release_lock(lock_id)

    def get_lock_id(self):
        """ hostname+processID+threadName"""
        return f'{socket.gethostname()}{os.getpid()}{threading.current_thread().name}'
    
          
        

调用代码如下

          
            redis_lock = RedisLock('lockname', host='Your IP') # 第一个匿名参数必传,作为 redis的key
with redis_lock.lock() as lock:
    print('You Can Do Something Here')

          
        

注意说明

          
            1. 注释部分是我写的时候,调试用的代码,最后写完的时候都替换为简洁的语法.

2. 因为锁具有互斥特性, 所以选择 set() 的 nx 参数来实现,
    nx参数:我个人一直这样记(读作 not exist)  ====> 不存在
    理解:
        不存在则添加,存在就不添加了。
        举一反三,没有锁就加个锁。有锁就不加锁了。

3. set方法 的 ex 参数, 可代替expire方法的来设置过期时间

4. redis 有 很多指令变形。 比如 set(nx=, ex=)  可拆分为 setnx setex
   但 "set() 这种指令更优", 能用就尽量用, 理由如下:
       "set指令 好处是 set具有原子性", 避免了解决资源竞争的同时引发自身可能出现的资源竞争
   
5. 我使用装饰器版 的 上下文管理器,对代码做了封装, 所以调用时,用"with语句"即可

6. with redis_lock.lock() as lock ,"lock() 这里可以自己指定2个参数":
    lock_id=None   # 这是区分不同线程的唯一标识符,默认为(主机名+进程ID+线程名),可自传
    expire=None    # 过期时间,秒为单位

7. 需要注意一个点,与redis通信是以二进制形式。 所以我在代码内部对 lock_id 做了"pickle序列化"
   当然如果是字符串用 encode() 与 decode() 来实现也是可以的。

          
        

更多文章、技术交流、商务合作、联系博主

微信扫码或搜索:z360901061

微信扫一扫加我为好友

QQ号联系: 360901061

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描下面二维码支持博主2元、5元、10元、20元等您想捐的金额吧,狠狠点击下面给点支持吧,站长非常感激您!手机微信长按不能支付解决办法:请将微信支付二维码保存到相册,切换到微信,然后点击微信右上角扫一扫功能,选择支付二维码完成支付。

【本文对您有帮助就好】

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描上面二维码支持博主2元、5元、10元、自定义金额等您想捐的金额吧,站长会非常 感谢您的哦!!!

发表我的评论
最新评论 总共0条评论